diff --git a/v1/client/client.go b/v1/client/client.go index 5d7a75b..91b5272 100644 --- a/v1/client/client.go +++ b/v1/client/client.go @@ -324,19 +324,19 @@ func (wsClient *SafeWebsocketClient) connect() error { }() conn.SetPingHandler(func(pingData string) error { - wsClient.writeChan <- Message{ - MessageType: MessageTypePong, - Data: []byte(pingData), - } - // if err := conn.WriteMessage(websocket.PongMessage, []byte(pingData)); err != nil { - // if err == websocket.ErrCloseSent { - // return nil - // } - // if netErr, ok := err.(interface{ Timeout() bool }); ok && netErr.Timeout() { - // return nil - // } - // return err + // wsClient.writeChan <- Message{ + // MessageType: MessageTypePong, + // Data: []byte(pingData), // } + if err := conn.WriteMessage(websocket.PongMessage, []byte(pingData)); err != nil { + if err == websocket.ErrCloseSent { + return nil + } + if netErr, ok := err.(interface{ Timeout() bool }); ok && netErr.Timeout() { + return nil + } + return err + } return nil })