Added automatic reconnect to WebSocket (#33)

Author: ClemensElflein

src/components/LazyLog/index.tsx

@@ -39,6 +39,15 @@ export interface WebsocketOptions {
      * Callback allback which formats the websocket data stream.
      */
     formatMessage?: ((message: any) => string) | undefined;
+    /**
+     * Set to true, to reconnect the WebSocket automatically.
+     */
+    reconnect?: boolean;
+    /**
+     * Set the time to wait between reconnects in seconds.
+     * Default is 1s
+     */
+    reconnectWait?: number
 }
 
 export interface ErrorStatus extends Error {

src/components/Utils/websocket.ts

@@ -10,6 +10,7 @@ export default (url: string | URL, options: WebsocketOptions) => {
     const emitter = mitt();
     let encodedLog = new Uint8Array();
     let overage: any = null;
+    let aborted: boolean = false;
 
     emitter.on("data", (data) => {
         encodedLog = bufferConcat(
@@ -46,6 +47,10 @@ export default (url: string | URL, options: WebsocketOptions) => {
             });
             socket.addEventListener("close", (e) => {
                 onClose && onClose(e);
+                if(!aborted && options.reconnect) {
+                    const timeout = options.reconnectWait ?? 1;
+                    setTimeout(() => emitter.emit("start"), timeout*1000);
+                }
             });
 
             socket.addEventListener("error", (err) => {
@@ -65,7 +70,10 @@ export default (url: string | URL, options: WebsocketOptions) => {
                 emitter.emit("data", msg);
             });
 
-            emitter.on("abort", () => socket.close());
+            emitter.on("abort", () => {
+                aborted = true;
+                socket.close()
+            });
         } catch (err) {
             emitter.emit("error", err);
         }