Update read deadline before reading.

This commit is contained in:
Joachim Bauch 2020-12-11 16:01:49 +01:00
parent d2b7fafde5
commit 9ad4a0c021
Failed to extract signature

View file

@ -236,7 +236,6 @@ func (c *Client) ReadPump() {
} }
conn.SetReadLimit(maxMessageSize) conn.SetReadLimit(maxMessageSize)
conn.SetReadDeadline(time.Now().Add(pongWait))
conn.SetPongHandler(func(msg string) error { conn.SetPongHandler(func(msg string) error {
now := time.Now() now := time.Now()
conn.SetReadDeadline(now.Add(pongWait)) conn.SetReadDeadline(now.Add(pongWait))
@ -259,6 +258,7 @@ func (c *Client) ReadPump() {
decodeBuffer := bufferPool.Get().(*bytes.Buffer) decodeBuffer := bufferPool.Get().(*bytes.Buffer)
defer bufferPool.Put(decodeBuffer) defer bufferPool.Put(decodeBuffer)
for { for {
conn.SetReadDeadline(time.Now().Add(pongWait))
messageType, reader, err := conn.NextReader() messageType, reader, err := conn.NextReader()
if err != nil { if err != nil {
if websocket.IsUnexpectedCloseError(err, if websocket.IsUnexpectedCloseError(err,