shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 03:23:49 +08:00
Code Issues Projects Releases Wiki Activity

feat: add WriteDeadline

Browse Source
This commit is contained in:
wdvxdr 2021-12-26 13:09:08 +08:00
parent 94ec3ccbcd
commit ef2223828f
No known key found for this signature in database
GPG Key ID: 703F8C071DE7A1B6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/websocket.go
View File

@ -60,6 +60,7 @@ type wsConn struct {
func (c *wsConn) WriteText(b []byte) error { func (c *wsConn) WriteText(b []byte) error {
c.mu.Lock() c.mu.Lock()
defer c.mu.Unlock() defer c.mu.Unlock()
_ = c.conn.SetWriteDeadline(time.Now().Add(time.Second * 15))
return c.conn.WriteMessage(websocket.TextMessage, b) return c.conn.WriteMessage(websocket.TextMessage, b)
} }
@ -460,6 +461,7 @@ func (c *wsConn) handleRequest(_ *coolq.CQBot, payload []byte) {
c.mu.Lock() c.mu.Lock()
defer c.mu.Unlock() defer c.mu.Unlock()
_ = c.conn.SetWriteDeadline(time.Now().Add(time.Second * 15))
writer, _ := c.conn.NextWriter(websocket.TextMessage) writer, _ := c.conn.NextWriter(websocket.TextMessage)
_ = json.NewEncoder(writer).Encode(ret) _ = json.NewEncoder(writer).Encode(ret)
_ = writer.Close() _ = writer.Close()