diff options
| author |  Vitaly Takmazov | 2018-09-20 17:48:54 +0300 |
|---|---|---|
| committer | Vitaly Takmazov | 2018-09-20 17:48:54 +0300 |
| commit | b1c9ab2d9ccd6d5dcffabe2d5c4bca42563e2d3e (patch) | |
| tree | 51a3b3240d507c2acf20ececffbfd30e283505fe /juick-notifications | |
| parent | ddab0db7c0a44af5921289af7e90f7325f1cdf69 (diff) | |
do not handle messages from invalid websocket sessions
Diffstat (limited to 'juick-notifications')
2 files changed, 1 insertions, 2 deletions
diff --git a/juick-notifications/src/main/java/com/juick/components/Notifications.java b/juick-notifications/src/main/java/com/juick/components/Notifications.java index 15af4a52..79afe8fa 100644 --- a/juick-notifications/src/main/java/com/juick/components/Notifications.java +++ b/juick-notifications/src/main/java/com/juick/components/Notifications.java @@ -321,7 +321,7 @@ public class Notifications extends TextWebSocketHandler implements NotificationC @Override protected void handleTextMessage(WebSocketSession session, TextMessage text) throws Exception { - if (!closeFlag.get()) { + if (!closeFlag.get() && this.session.getDelegate().equals(session)) { com.juick.Message jmsg = jsonMapper.readValue(text.asBytes(), com.juick.Message.class); if (logger.isInfoEnabled()) // prevent writeValueAsString execution if logger disabled diff --git a/juick-notifications/src/main/java/com/juick/components/configuration/JuickServerWebsocketConfiguration.java b/juick-notifications/src/main/java/com/juick/components/configuration/JuickServerWebsocketConfiguration.java index a29566aa..bc7f59fb 100644 --- a/juick-notifications/src/main/java/com/juick/components/configuration/JuickServerWebsocketConfiguration.java +++ b/juick-notifications/src/main/java/com/juick/components/configuration/JuickServerWebsocketConfiguration.java @@ -25,7 +25,6 @@ import javax.inject.Inject; import java.util.Collections; import java.util.List; -@Lazy @Configuration @EnableScheduling public class JuickServerWebsocketConfiguration { |