ws recycle conn bug fix

2 years ago · 4a7026622e
parent 09fe19d97b
commit 4a7026622e
1 changed files with 6 additions and 2 deletions
--- a/internal/msg_gateway/gate/ws_server.go
+++ b/internal/msg_gateway/gate/ws_server.go
@ -220,8 +220,12 @@ func (ws *WServer) delUserConn(conn *UserConn) {
 			platform = k
 			uid = v
 		}
-		if oldConnMap, ok := ws.wsUserToConn[uid]; ok {
-			delete(oldConnMap, platform)
+		if oldConnMap, ok := ws.wsUserToConn[uid]; ok { // only recycle self conn
+			if oldconn, okMap := oldConnMap[platform]; okMap {
+				if oldconn == conn {
+					delete(oldConnMap, platform)
+				}
+			}
 			ws.wsUserToConn[uid] = oldConnMap
 			if len(oldConnMap) == 0 {
 				delete(ws.wsUserToConn, uid)