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
1 changed files with 1 additions and 1 deletions

View File

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