aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/main/java/com/juick/ws/WebsocketComponent.java7
1 files changed, 3 insertions, 4 deletions
diff --git a/src/main/java/com/juick/ws/WebsocketComponent.java b/src/main/java/com/juick/ws/WebsocketComponent.java
index 83e811a6..7399c298 100644
--- a/src/main/java/com/juick/ws/WebsocketComponent.java
+++ b/src/main/java/com/juick/ws/WebsocketComponent.java
@@ -97,19 +97,18 @@ public class WebsocketComponent extends TextWebSocketHandler {
}
}
if (sockSubscr != null) {
- synchronized (clients) {
- clients.add(sockSubscr);
- }
+ clients.add(sockSubscr);
}
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
synchronized (clients) {
- clients.stream().filter(c -> c.session.equals(session)).forEach(c -> {
+ clients.stream().filter(c -> c.session.getId().equals(session.getId())).forEach(c -> {
logger.info(String.format("session %s closed with status %s", c.clientName, status.getCode()));
clients.remove(c);
});
}
+
}
class SocketSubscribed {