From a224f31fe050fe1eeb4435677caa27d523897704 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Wed, 1 Nov 2017 13:53:50 +0300
Subject: api: try reconnect ws component

---
 .../src/main/java/com/juick/api/ApiServer.java     | 22 +++++++++++++++++++++-
 .../java/com/juick/api/TelegramBotManager.java     |  5 +++++
 2 files changed, 26 insertions(+), 1 deletion(-)

(limited to 'juick-api/src/main/java/com/juick')

diff --git a/juick-api/src/main/java/com/juick/api/ApiServer.java b/juick-api/src/main/java/com/juick/api/ApiServer.java
index a5bf5d38..110d8e2f 100644
--- a/juick-api/src/main/java/com/juick/api/ApiServer.java
+++ b/juick-api/src/main/java/com/juick/api/ApiServer.java
@@ -17,10 +17,15 @@
 package com.juick.api;
 
 import com.juick.User;
+import com.juick.server.component.DisconnectedEvent;
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.beans.factory.support.DefaultListableBeanFactory;
+import org.springframework.context.ApplicationListener;
+import org.springframework.web.socket.client.WebSocketConnectionManager;
+import org.springframework.web.socket.client.standard.StandardWebSocketClient;
 import rocks.xmpp.addr.Jid;
 import rocks.xmpp.core.XmppException;
 import rocks.xmpp.core.session.Extension;
@@ -30,14 +35,16 @@ import rocks.xmpp.core.stanza.model.Message;
 import rocks.xmpp.extensions.component.accept.ExternalComponent;
 import rocks.xmpp.extensions.oob.model.x.OobX;
 
+import javax.annotation.Nonnull;
 import javax.annotation.PostConstruct;
+import javax.inject.Inject;
 import java.net.URI;
 import java.net.URISyntaxException;
 
 /**
  * @author Ugnich Anton
  */
-public class ApiServer implements AutoCloseable {
+public class ApiServer implements ApplicationListener<DisconnectedEvent>, AutoCloseable {
     private static Logger logger = LoggerFactory.getLogger(ApiServer.class);
 
     private ExternalComponent xmpp;
@@ -52,6 +59,12 @@ public class ApiServer implements AutoCloseable {
     private int xmppPort;
     @Value("${xmpp_disabled:false}")
     private boolean isXmppDisabled;
+    @Inject
+    private WebSocketConnectionManager webSocketConnectionManager;
+    @Inject
+    private StandardWebSocketClient client;
+    @Inject
+    private DefaultListableBeanFactory beanFactory;
 
     @PostConstruct
     public void init() {
@@ -106,4 +119,11 @@ public class ApiServer implements AutoCloseable {
             logger.warn("attachment error", e1);
         }
     }
+
+    @Override
+    public void onApplicationEvent(@Nonnull DisconnectedEvent event) {
+        logger.info("reconnecting websocket...");
+        beanFactory.destroySingleton("client");
+        beanFactory.destroySingleton("webSocketConnectionManager");
+    }
 }
diff --git a/juick-api/src/main/java/com/juick/api/TelegramBotManager.java b/juick-api/src/main/java/com/juick/api/TelegramBotManager.java
index b9154424..44f0fdc6 100644
--- a/juick-api/src/main/java/com/juick/api/TelegramBotManager.java
+++ b/juick-api/src/main/java/com/juick/api/TelegramBotManager.java
@@ -18,6 +18,7 @@
 package com.juick.api;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.juick.server.component.DisconnectedEvent;
 import com.juick.service.MessagesService;
 import com.juick.service.SubscriptionService;
 import com.juick.service.TelegramService;
@@ -33,6 +34,7 @@ import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 import org.springframework.web.socket.CloseStatus;
@@ -68,6 +70,8 @@ public class TelegramBotManager extends TextWebSocketHandler {
     private SubscriptionService subscriptionService;
     @Inject
     private ObjectMapper jsonMapper;
+    @Inject
+    private ApplicationEventPublisher applicationEventPublisher;
 
     private WebSocketSession session;
 
@@ -99,6 +103,7 @@ public class TelegramBotManager extends TextWebSocketHandler {
     @Override
     public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
         logger.info("WebSocket disconnected with code {}: {}", status.getCode(), status.getReason());
+        applicationEventPublisher.publishEvent(new DisconnectedEvent(this));
     }
 
     @Scheduled(fixedRate = 30000)
-- 
cgit v1.2.3