1 files changed, 9 insertions, 4 deletions
diff --git a/juick-api/src/main/java/com/juick/api/TGBot.java b/juick-api/src/main/java/com/juick/api/TGBot.java
index 7456c8e7..d665a197 100644
--- a/juick-api/src/main/java/com/juick/api/TGBot.java
+++ b/juick-api/src/main/java/com/juick/api/TGBot.java
@@ -20,6 +20,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.client.WebSocketConnectionManager;
 import org.springframework.web.socket.handler.TextWebSocketHandler;
 
 import javax.inject.Inject;
@@ -42,6 +43,8 @@ public class TGBot extends TextWebSocketHandler {
     TelegramService telegramService;
     @Inject
     MessagesService messagesService;
+    @Inject
+    WebSocketConnectionManager connectionManager;
 
     WebSocketSession session;
 
@@ -76,14 +79,16 @@ public class TGBot extends TextWebSocketHandler {
 
     @Override
     public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
-        logger.info("WebSocket disconnected with code {}: {}", status.getCode(), status.getReason());
-        this.session = null;
+        logger.info("WebSocket disconnected with code {}: {}, trying to reconnect", status.getCode(), status.getReason());
+        connectionManager.start();
     }
 
     @Scheduled(fixedRate = 30000)
     public void ping() throws IOException {
-        logger.debug("Sending WebSocket ping");
-        session.sendMessage(new PingMessage());
+        if (session != null) {
+            logger.debug("Sending WebSocket ping");
+            session.sendMessage(new PingMessage());
+        }
     }
 
     @Override