gopkg/recws
2
0
Fork 0
Code Activity

simplify CloseError testing

Browse Source
This commit is contained in:
Stephan Knauer
2021-12-11 13:54:13 +01:00
parent 1e95e4be4b
commit d31f6143c1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
recws.go
+4 -4
View File
@@ -116,7 +116,7 @@ func (rc *RecConn) ReadMessage() (messageType int, message []byte, err error) {
rc.Close() rc.Close()
return messageType, message, nil return messageType, message, nil
} }
if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) { if err != nil {
rc.CloseAndReconnect() rc.CloseAndReconnect()
} }
} }
@@ -138,7 +138,7 @@ func (rc *RecConn) WriteMessage(messageType int, data []byte) error {
rc.Close() rc.Close()
return nil return nil
} }
if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) { if err != nil {
rc.CloseAndReconnect() rc.CloseAndReconnect()
} }
} }
@@ -162,7 +162,7 @@ func (rc *RecConn) WriteJSON(v interface{}) error {
rc.Close() rc.Close()
return nil return nil
} }
if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) { if err != nil {
rc.CloseAndReconnect() rc.CloseAndReconnect()
} }
} }
@@ -185,7 +185,7 @@ func (rc *RecConn) ReadJSON(v interface{}) error {
rc.Close() rc.Close()
return nil return nil
} }
if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) { if err != nil {
rc.CloseAndReconnect() rc.CloseAndReconnect()
} }
} }