From 284e89d4108bc2c2bfe349fd49a0f61405658d7a Mon Sep 17 00:00:00 2001
From: Gordon <1432970085@qq.com>
Date: Thu, 23 Feb 2023 12:14:58 +0800
Subject: [PATCH] websocket update

---
 internal/msg_gateway/gate/ws_server.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/internal/msg_gateway/gate/ws_server.go b/internal/msg_gateway/gate/ws_server.go
index 2e1a33fe5..7323c3fc2 100644
--- a/internal/msg_gateway/gate/ws_server.go
+++ b/internal/msg_gateway/gate/ws_server.go
@@ -106,6 +106,12 @@ func (ws *WServer) readMsg(conn *UserConn) {
 			ws.delUserConn(conn)
 			return
 		}
+		if messageType == websocket.CloseMessage {
+			log.NewWarn("", "WS receive error ", " userIP", conn.RemoteAddr().String(), "userUid", "platform", "error", string(msg))
+			userCount--
+			ws.delUserConn(conn)
+			return
+		}
 		log.NewDebug("", "size", utils.ByteSize(uint64(len(msg))))
 		if conn.IsCompress {
 			buff := bytes.NewBuffer(msg)