Improves error handling on reading from WS connections

ksysoev · Apr 13, 2024 · 5a76f13 · 5a76f13
1 parent f0a4bd6
commit 5a76f13
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/channel/connection.go b/channel/connection.go
@@ -69,14 +69,18 @@ func (c *Conn) HandleRequests() {
 		msgType, reader, err := c.ws.Reader(c.ctx)
 
 		if err != nil {
-			slog.Warn("Error reading message: " + err.Error())
-
 			return
 		}
 
 		data, err := io.ReadAll(reader)
-
 		if err != nil {
+			switch {
+			case errors.Is(err, io.EOF):
+				return
+			case errors.Is(err, context.Canceled):
+				return
+			}
+
 			slog.Warn("Error reading message: " + err.Error())
 
 			return