Only remove first time handshake failures

mrniko · Jul 26, 2024 · 17ed3a0 · 17ed3a0
1 parent 08359a3
commit 17ed3a0
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);
     }