lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add mutex lock to WebSocketClient for thread-safe message sending

Browse Source
This commit is contained in:
lixiangwuxian 2025-01-09 22:21:43 +08:00
parent ffc73605e8
commit fa635ad9f1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ws_client/client.go
View File

@ -4,6 +4,7 @@ import (
"encoding/json" "encoding/json"
"log" "log"
"net/url" "net/url"
"sync"
"git.lxtend.com/qqbot/action" "git.lxtend.com/qqbot/action"
"git.lxtend.com/qqbot/constants" "git.lxtend.com/qqbot/constants"
@ -16,6 +17,7 @@ type WebSocketClient struct {
conn *websocket.Conn conn *websocket.Conn
closed bool closed bool
Done chan struct{} Done chan struct{}
lock sync.Mutex
} }
func NewWebSocketClient(scheme, host, path string) (*WebSocketClient, error) { func NewWebSocketClient(scheme, host, path string) (*WebSocketClient, error) {
@ -88,6 +90,8 @@ func (c *WebSocketClient) receiveMessages() {
} }
func (c *WebSocketClient) SendMessage(messageType int, message []byte) error { func (c *WebSocketClient) SendMessage(messageType int, message []byte) error {
c.lock.Lock()
defer c.lock.Unlock()
return c.conn.WriteMessage(messageType, message) return c.conn.WriteMessage(messageType, message)
} }