From d31f6143c1f72605c4a11c975361f58eb1e65fda Mon Sep 17 00:00:00 2001
From: Stephan Knauer <sknr@users.noreply.github.com>
Date: Sat, 11 Dec 2021 13:54:13 +0100
Subject: [PATCH] simplify CloseError testing

---
 recws.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/recws.go b/recws.go
index 1e14eb4..d572630 100644
--- a/recws.go
+++ b/recws.go
@@ -116,7 +116,7 @@ func (rc *RecConn) ReadMessage() (messageType int, message []byte, err error) {
 			rc.Close()
 			return messageType, message, nil
 		}
-		if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+		if err != nil {
 			rc.CloseAndReconnect()
 		}
 	}
@@ -138,7 +138,7 @@ func (rc *RecConn) WriteMessage(messageType int, data []byte) error {
 			rc.Close()
 			return nil
 		}
-		if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+		if err != nil {
 			rc.CloseAndReconnect()
 		}
 	}
@@ -162,7 +162,7 @@ func (rc *RecConn) WriteJSON(v interface{}) error {
 			rc.Close()
 			return nil
 		}
-		if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+		if err != nil {
 			rc.CloseAndReconnect()
 		}
 	}
@@ -185,7 +185,7 @@ func (rc *RecConn) ReadJSON(v interface{}) error {
 			rc.Close()
 			return nil
 		}
-		if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+		if err != nil {
 			rc.CloseAndReconnect()
 		}
 	}