Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit f094de2

Browse files
committed
加上写超时
1 parent 0b8f250 commit f094de2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/http/controller/websocket.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,13 @@ func (this *WebsocketController) Ws(wsConn *websocket.Conn) {
5353
for {
5454
select {
5555
case message := <-userData.MessageQueue(serverId):
56+
wsConn.SetWriteDeadline(time.Now().Add(5 * 10e9))
5657
if err := websocket.JSON.Send(wsConn, message); err != nil {
5758
clientClosed = true
5859
}
5960
// 心跳
6061
case <-time.After(30e9):
62+
wsConn.SetWriteDeadline(time.Now().Add(5 * 10e9))
6163
if err := websocket.JSON.Send(wsConn, ""); err != nil {
6264
clientClosed = true
6365
}

0 commit comments

Comments
 (0)