fix: ws 处理 Panic 协程异常

2024-02-02 17:09:23 +08:00
parent fb7a2dfa3c
commit 1707d9ea3b
1 changed files with 6 additions and 0 deletions
--- 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)