fix: bug

KU-Taverse · Oct 26, 2024 · f849800 · f849800
1 parent 538e384
commit f849800
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/kutaverse/game/websocket/chat/ChatWebSocketHandler.java b/src/main/java/kutaverse/game/websocket/chat/ChatWebSocketHandler.java
@@ -53,10 +53,13 @@ public Mono<Void> handle(WebSocketSession session) {
                     });
                 })
                 .flatMap(data->{
-                    return Flux.fromIterable(sessions.values())
-                            .flatMap(wsSession -> {
+                    return Flux.fromIterable(sessions.entrySet())
+                            .flatMap(entry -> {
+                                WebSocketSession wsSession = entry.getValue();
+                                String key = entry.getKey();
                                 if(!wsSession.isOpen()) {
                                     wsSession.close().subscribe();
+                                    sessions.remove(key);
                                     return Mono.never();
                                 }
                                 else