Fix zero writeTimeout cause write failure

Signed-off-by: Yilun <[email protected]>
nknorg · Feb 17, 2023 · 3b183c6 · 3b183c6
1 parent e903dc0
commit 3b183c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -785,7 +785,11 @@ func (c *Client) handleReconnect() {
 
 func (c *Client) writeMessage(buf []byte, writeTimeout time.Duration) error {
 	c.lock.Lock()
-	c.conn.SetWriteDeadline(time.Now().Add(writeTimeout))
+	if writeTimeout == 0 {
+		c.conn.SetWriteDeadline(zeroTime)
+	} else {
+		c.conn.SetWriteDeadline(time.Now().Add(writeTimeout))
+	}
 	err := c.conn.WriteMessage(websocket.BinaryMessage, buf)
 	c.lock.Unlock()
 	if err != nil {