iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

muxed socket.send fixes #27

Browse Source
This commit is contained in:
Craig 2020-02-03 14:46:00 +00:00
parent 1cf3eee3e6
commit 0d86073f7e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/internal/websocket/socket.go
View File

@ -2,6 +2,7 @@ package websocket
import ( import (
"strings" "strings"
"sync"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
) )
@ -9,6 +10,7 @@ import (
type WebSocket struct { type WebSocket struct {
id string id string
connection *websocket.Conn connection *websocket.Conn
mu sync.Mutex
} }
func (socket *WebSocket) Address() *string { func (socket *WebSocket) Address() *string {
@ -21,6 +23,8 @@ func (socket *WebSocket) Address() *string {
} }
func (socket *WebSocket) Send(v interface{}) error { func (socket *WebSocket) Send(v interface{}) error {
socket.mu.Lock()
defer socket.mu.Unlock()
if socket.connection == nil { if socket.connection == nil {
return nil return nil
} }