mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-15 01:05:06 +08:00
46 lines
663 B
Go
46 lines
663 B
Go
|
package websocket
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gorilla/websocket"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
ID string
|
||
|
|
Socket *websocket.Conn
|
||
|
|
Msg chan []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWsClient(ID string, socket *websocket.Conn) *Client {
|
||
|
|
return &Client{
|
||
|
|
ID: ID,
|
||
|
|
Socket: socket,
|
||
|
|
Msg: make(chan []byte, 100),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) Read() {
|
||
|
|
defer func() {
|
||
|
|
close(c.Msg)
|
||
|
|
}()
|
||
|
|
for {
|
||
|
|
_, message, err := c.Socket.ReadMessage()
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
ProcessData(c, message)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) Write() {
|
||
|
|
defer func() {
|
||
|
|
c.Socket.Close()
|
||
|
|
}()
|
||
|
|
for {
|
||
|
|
message, ok := <-c.Msg
|
||
|
|
if !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
_ = c.Socket.WriteMessage(websocket.TextMessage, message)
|
||
|
|
}
|
||
|
|
}
|