Websocket reconnect reload page data. Fixes #504

1 files changed, 11 insertions, 6 deletions

diff --git a/ui/src/services/WebSocketService.ts b/ui/src/services/WebSocketService.ts
index c5e45e79..6d951618 100644
--- a/ui/src/services/WebSocketService.ts
+++ b/ui/src/services/WebSocketService.ts
@@ -40,6 +40,7 @@ import {
   GetPrivateMessagesForm,
   UserJoinForm,
   MessageType,
+  WebSocketJsonResponse,
 } from '../interfaces';
 import { UserService } from './';
 import { i18n } from '../i18next';
@@ -59,17 +60,21 @@ export class WebSocketService {
 
   private constructor() {
     this.ws = new ReconnectingWebSocket(wsUri);
-    this.ws.onopen = () => {
-      console.log(`Connected to ${wsUri}`);
-      if (UserService.Instance.user) {
-        this.userJoin();
-      }
-    };
 
     this.subject = Observable.create((obs: any) => {
       this.ws.onmessage = e => {
         obs.next(JSON.parse(e.data));
       };
+      this.ws.onopen = () => {
+        console.log(`Connected to ${wsUri}`);
+        if (UserService.Instance.user) {
+          this.userJoin();
+        }
+        let res: WebSocketJsonResponse = {
+          reconnect: true,
+        };
+        obs.next(res);
+      };
     }).pipe(share());
   }