39 lines
656 B
Go
39 lines
656 B
Go
package connection
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type Connection struct {
|
|
Identificator string
|
|
Socket *websocket.Conn
|
|
|
|
writeMu sync.Mutex
|
|
closeMu sync.Mutex
|
|
closed bool
|
|
}
|
|
|
|
func (c *Connection) WriteBinary(payload []byte) error {
|
|
c.writeMu.Lock()
|
|
defer c.writeMu.Unlock()
|
|
return c.Socket.WriteMessage(websocket.BinaryMessage, payload)
|
|
}
|
|
|
|
func (c *Connection) WriteJSON(v any) error {
|
|
c.writeMu.Lock()
|
|
defer c.writeMu.Unlock()
|
|
return c.Socket.WriteJSON(v)
|
|
}
|
|
|
|
func (c *Connection) Close() error {
|
|
c.closeMu.Lock()
|
|
defer c.closeMu.Unlock()
|
|
if c.closed {
|
|
return nil
|
|
}
|
|
c.closed = true
|
|
return c.Socket.Close()
|
|
}
|