From 1707d9ea3b57f08c85efc8718ab7879bf569e49e Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 2 Feb 2024 17:09:23 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20ws=20=E5=A4=84=E7=90=86=20Panic=20?=
 =?UTF-8?q?=E5=8D=8F=E7=A8=8B=E5=BC=82=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/ws/service/ws.impl.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/modules/ws/service/ws.impl.go b/src/modules/ws/service/ws.impl.go
index 6dfb236b..d210d4bf 100644
--- a/src/modules/ws/service/ws.impl.go
+++ b/src/modules/ws/service/ws.impl.go
@@ -100,6 +100,12 @@ func (s *WSImpl) NewClient(uid string, groupIDs []string, conn *websocket.Conn)
 		}
 	}
 
+	defer func() {
+		// 在这里处理 Panic 协程异常
+		if err := recover(); err != nil {
+			logger.Errorf("ws Panic Error: %s => %v", clientID, err)
+		}
+	}()
 	go s.clientRead(wsClient)
 	go s.clientWrite(wsClient)