Merge pull request #992 from berinhardt/websocket-leak-fix

Only remove first time handshake failures
mrniko · Aug 8, 2024 · c4ca71e · c4ca71e
2 parents 6981e0c + 9675e98
commit c4ca71e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/com/corundumstudio/socketio/transport/WebSocketTransport.java b/src/main/java/com/corundumstudio/socketio/transport/WebSocketTransport.java
@@ -183,8 +183,11 @@ private void closeClient(UUID sessionId, Channel channel) {
         } catch (Throwable t) {
             log.warn("Can't close channel for sessionId: {}", sessionId, t);
         }
-        ClientHead clientHead = clientsBox.removeClient(sessionId);
-        clientHead.disconnect();
+        ClientHead clientHead = clientsBox.get(sessionId);
+        if (clientHead != null && clientHead.getNamespaces().isEmpty()) {
+        	clientsBox.removeClient(sessionId);
+        	clientHead.disconnect();
+    	}
         log.info("Client with sessionId: {} was disconnected", sessionId);
     }