server: ConcurrentWebSocketSessionDecorator

1 files changed, 3 insertions, 3 deletions

diff --git a/juick-server/src/main/java/com/juick/server/WebsocketManager.java b/juick-server/src/main/java/com/juick/server/WebsocketManager.java
index c43e191c..fdee5065 100644
--- a/juick-server/src/main/java/com/juick/server/WebsocketManager.java
+++ b/juick-server/src/main/java/com/juick/server/WebsocketManager.java
@@ -25,7 +25,6 @@ import com.juick.server.util.HttpBadRequestException;
 import com.juick.server.util.HttpForbiddenException;
 import com.juick.server.util.HttpNotFoundException;
 import com.juick.service.MessagesService;
-import com.juick.service.SubscriptionService;
 import com.juick.service.UserService;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.commons.lang3.math.NumberUtils;
@@ -39,6 +38,7 @@ import org.springframework.web.socket.CloseStatus;
 import org.springframework.web.socket.PingMessage;
 import org.springframework.web.socket.TextMessage;
 import org.springframework.web.socket.WebSocketSession;
+import org.springframework.web.socket.handler.ConcurrentWebSocketSessionDecorator;
 import org.springframework.web.socket.handler.TextWebSocketHandler;
 import org.springframework.web.util.UriComponents;
 import org.springframework.web.util.UriComponentsBuilder;
@@ -177,7 +177,7 @@ public class WebsocketManager extends TextWebSocketHandler {
     }
 
     class SocketSubscribed {
-        WebSocketSession session;
+        ConcurrentWebSocketSessionDecorator session;
         String clientName;
         User visitor;
         int MID;
@@ -188,7 +188,7 @@ public class WebsocketManager extends TextWebSocketHandler {
         boolean legacy;
 
         public SocketSubscribed(WebSocketSession session, String clientName, User visitor, boolean legacy) {
-            this.session = session;
+            this.session = new ConcurrentWebSocketSessionDecorator(session, 60000, 65536);
             this.clientName = clientName;
             this.visitor = visitor;
             tsConnected = tsLastData = System.currentTimeMillis();