auth work

ws/ws.go

@@ -52,7 +52,11 @@ func WsHandler(w http.ResponseWriter, r *http.Request) {
 	for {
 		_, msg, err := conn.ReadMessage()
 		if err != nil {
-		}
+			if websocket.IsCloseError(err, websocket.CloseNormalClosure, websocket.CloseGoingAway, websocket.CloseNoStatusReceived) {
+				config.Logger.Info("WS closed")
+				conn.Close()
+				break
+			}
 			config.Logger.Error(fmt.Sprintf("Error reading websocket message: %v", err))
 			continue
 		}