You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
864 B
Go
60 lines
864 B
Go
package websocket
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
const MaxMessageQuenue = 32
|
|
|
|
type Client struct {
|
|
ID string
|
|
Socket *websocket.Conn
|
|
Msg chan []byte
|
|
closed atomic.Bool
|
|
}
|
|
|
|
func NewWsClient(ID string, socket *websocket.Conn) *Client {
|
|
return &Client{
|
|
ID: ID,
|
|
Socket: socket,
|
|
Msg: make(chan []byte, MaxMessageQuenue),
|
|
}
|
|
}
|
|
|
|
func (c *Client) Read() {
|
|
defer func() {
|
|
c.closed.Store(true)
|
|
close(c.Msg)
|
|
}()
|
|
for {
|
|
_, message, err := c.Socket.ReadMessage()
|
|
if err != nil {
|
|
return
|
|
}
|
|
ProcessData(c, message)
|
|
}
|
|
}
|
|
|
|
func (c *Client) Write() {
|
|
defer c.Socket.Close()
|
|
for {
|
|
message, ok := <-c.Msg
|
|
if !ok {
|
|
return
|
|
}
|
|
_ = c.Socket.WriteMessage(websocket.TextMessage, message)
|
|
}
|
|
}
|
|
|
|
func (c *Client) Send(res []byte) {
|
|
if c.closed.Load() {
|
|
return
|
|
}
|
|
select {
|
|
case c.Msg <- res:
|
|
default:
|
|
}
|
|
}
|