Automatically reconnect WebSocket

Signed-off-by: Michiel <[email protected]>
openwallet-foundation-labs · Sep 4, 2024 · 576b6a8 · 576b6a8
1 parent f6d1d4a
commit 576b6a8
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/examples/script.js b/examples/script.js
@@ -106,20 +106,23 @@ const updateIdentities = async () => {
 
   // listen for messages on websocket
   if (!ws) {
-    const proto = window.location.protocol === 'http:' ? 'ws' : 'wss';
-    ws = new WebSocket(`${proto}://${window.location.host}/receive-messages`);
-    ws.addEventListener('message', (event) => {
-      const message = JSON.parse(event.data);
-      const messagesContainer =
-        document.querySelector('.received-messages');
-      messagesContainer.appendChild(renderMessage(message));
-    });
-    ws.addEventListener('open', () => {
-      Object.values(identities).forEach((identity) => {
-        ws.send(JSON.stringify(identity));
-        registered.push(identity.id);
+    function connectWS() {
+      const proto = window.location.protocol === 'http:' ? 'ws' : 'wss';
+      ws = new WebSocket(`${proto}://${window.location.host}/receive-messages`);
+      ws.addEventListener('message', (event) => {
+        const message = JSON.parse(event.data);
+        const messagesContainer = document.querySelector('.received-messages');
+        messagesContainer.appendChild(renderMessage(message));
       });
-    });
+      ws.addEventListener('open', () => {
+        Object.values(identities).forEach((identity) => {
+          ws.send(JSON.stringify(identity));
+          registered.push(identity.id);
+        });
+      });
+      ws.addEventListener('close', connectWS);
+    }
+    connectWS();
   }
 };