fix(websocket): replace websocket error handling with helper function

Author: 0xJacky

api/cluster/websocket.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/0xJacky/Nginx-UI/internal/analytic"
 	"github.com/0xJacky/Nginx-UI/internal/kernel"
+	"github.com/0xJacky/Nginx-UI/internal/helper"
 	"github.com/0xJacky/Nginx-UI/model"
 	"github.com/gin-gonic/gin"
 	"github.com/gorilla/websocket"
@@ -286,10 +287,10 @@ func (c *Client) readPump() {
 		for {
 			_, _, err := c.conn.ReadMessage()
 			if err != nil {
-				if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
+				if helper.IsUnexpectedWebsocketError(err) {
 					logger.Error("Websocket error:", err)
 				}
-				break
+				return 
 			}
 		}
 	}()

api/nginx_log/websocket.go

@@ -146,7 +146,7 @@ func handleLogControl(ws *websocket.Conn, controlChan chan controlStruct, errCha
 
 	for {
 		msgType, payload, err := ws.ReadMessage()
-		if err != nil && websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+		if err != nil && helper.IsUnexpectedWebsocketError(err) {
 			errChan <- errors.Wrap(err, "error handleLogControl read message")
 			return
 		}