2019-08-15 16:09:56 +02:00
|
|
|
// Package socket provides a pseudo socket
|
|
|
|
package socket
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
|
2021-10-12 13:55:53 +02:00
|
|
|
"go-micro.dev/v4/transport"
|
2019-08-15 16:09:56 +02:00
|
|
|
)
|
|
|
|
|
2022-09-30 16:27:07 +02:00
|
|
|
// Socket is our pseudo socket for transport.Socket.
|
2019-08-15 21:08:49 +02:00
|
|
|
type Socket struct {
|
2019-08-15 16:09:56 +02:00
|
|
|
// closed
|
|
|
|
closed chan bool
|
|
|
|
// send chan
|
|
|
|
send chan *transport.Message
|
|
|
|
// recv chan
|
|
|
|
recv chan *transport.Message
|
2023-04-26 02:16:34 +02:00
|
|
|
id string
|
|
|
|
// remote addr
|
|
|
|
remote string
|
|
|
|
// local addr
|
|
|
|
local string
|
2019-08-15 16:09:56 +02:00
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) SetLocal(l string) {
|
2019-08-15 16:09:56 +02:00
|
|
|
s.local = l
|
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) SetRemote(r string) {
|
2019-08-15 16:09:56 +02:00
|
|
|
s.remote = r
|
|
|
|
}
|
|
|
|
|
2022-09-30 16:27:07 +02:00
|
|
|
// Accept passes a message to the socket which will be processed by the call to Recv.
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Accept(m *transport.Message) error {
|
2019-08-15 16:09:56 +02:00
|
|
|
select {
|
|
|
|
case s.recv <- m:
|
|
|
|
return nil
|
2019-08-24 21:12:04 +02:00
|
|
|
case <-s.closed:
|
|
|
|
return io.EOF
|
2019-08-15 16:09:56 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-09-30 16:27:07 +02:00
|
|
|
// Process takes the next message off the send queue created by a call to Send.
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Process(m *transport.Message) error {
|
2019-08-15 16:09:56 +02:00
|
|
|
select {
|
|
|
|
case msg := <-s.send:
|
|
|
|
*m = *msg
|
2019-08-24 21:12:04 +02:00
|
|
|
case <-s.closed:
|
|
|
|
// see if we need to drain
|
|
|
|
select {
|
|
|
|
case msg := <-s.send:
|
|
|
|
*m = *msg
|
|
|
|
return nil
|
|
|
|
default:
|
|
|
|
return io.EOF
|
|
|
|
}
|
2019-08-15 16:09:56 +02:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Remote() string {
|
2019-08-15 16:09:56 +02:00
|
|
|
return s.remote
|
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Local() string {
|
2019-08-15 16:09:56 +02:00
|
|
|
return s.local
|
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Send(m *transport.Message) error {
|
2019-08-15 16:09:56 +02:00
|
|
|
// send a message
|
|
|
|
select {
|
2020-04-26 13:44:59 +02:00
|
|
|
case s.send <- m:
|
2019-08-15 16:09:56 +02:00
|
|
|
case <-s.closed:
|
|
|
|
return io.EOF
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Recv(m *transport.Message) error {
|
2019-08-15 16:09:56 +02:00
|
|
|
// receive a message
|
|
|
|
select {
|
|
|
|
case msg := <-s.recv:
|
|
|
|
// set message
|
|
|
|
*m = *msg
|
|
|
|
case <-s.closed:
|
|
|
|
return io.EOF
|
|
|
|
}
|
|
|
|
|
|
|
|
// return nil
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-09-30 16:27:07 +02:00
|
|
|
// Close closes the socket.
|
2019-08-15 21:08:49 +02:00
|
|
|
func (s *Socket) Close() error {
|
2019-08-15 16:09:56 +02:00
|
|
|
select {
|
|
|
|
case <-s.closed:
|
|
|
|
// no op
|
|
|
|
default:
|
|
|
|
close(s.closed)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// New returns a new pseudo socket which can be used in the place of a transport socket.
|
|
|
|
// Messages are sent to the socket via Accept and receives from the socket via Process.
|
|
|
|
// SetLocal/SetRemote should be called before using the socket.
|
2019-11-27 19:12:07 +02:00
|
|
|
func New(id string) *Socket {
|
2019-08-15 21:08:49 +02:00
|
|
|
return &Socket{
|
2019-11-27 19:12:07 +02:00
|
|
|
id: id,
|
2019-08-15 16:09:56 +02:00
|
|
|
closed: make(chan bool),
|
|
|
|
local: "local",
|
|
|
|
remote: "remote",
|
|
|
|
send: make(chan *transport.Message, 128),
|
|
|
|
recv: make(chan *transport.Message, 128),
|
|
|
|
}
|
|
|
|
}
|