fix(fe): connect to websocket immediately after app created

semaphoreui · Nov 22, 2020 · 90d1639 · 90d1639
1 parent 043b885
commit 90d1639
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 17 deletions.
diff --git a/web2/src/lib/Socket.js b/web2/src/lib/Socket.js
@@ -29,20 +29,4 @@ export default class Socket extends Listenable {
     this.ws.close();
     delete this.ws;
   }
-
-  addListener(callback) {
-    const isFirstListener = !this.hasListeners();
-    const listenerId = super.addListener(callback);
-    if (isFirstListener) {
-      this.start();
-    }
-    return listenerId;
-  }
-
-  removeListener(id) {
-    super.removeListener(id);
-    if (!this.hasListeners()) {
-      this.stop();
-    }
-  }
 }
diff --git a/web2/src/socket.js b/web2/src/socket.js
@@ -1,6 +1,10 @@
 import Socket from '@/lib/Socket';
 
-export default new Socket(() => {
+const socket = new Socket(() => {
   const protocol = document.location.protocol === 'https:' ? 'wss' : 'ws';
   return new WebSocket(`${protocol}://${document.location.host}/api/ws`);
 });
+
+socket.start();
+
+export default socket;