feat: trigger reconnect on emit if not connected

source/event_hub.ts

@@ -189,7 +189,7 @@ export class EventHub {
    * @return {Boolean}
    */
   isConnected(): boolean {
-    return this._socketIo?.isConnected() || false;
+    return this._socketIo?.socket.connected || false;
   }
 
   /**

source/simple_socketio.ts

@@ -55,6 +55,7 @@ export default class SimpleSocketIOClient {
   private packetQueue: string[] = [];
   private reconnectionAttempts: number = 0;
   private reconnecting: boolean = false;
+  initializingPromise: Promise<void>;
 
   // Added socket object with connected, open reconnect and transport properties to match current API
   // The old socket-io client uses both a connected and open property that are interchangeable
@@ -129,7 +130,7 @@ export default class SimpleSocketIOClient {
     this.heartbeatTimeoutMs = heartbeatTimeoutMs;
     this.apiUser = apiUser;
     this.apiKey = apiKey;
-    this.initializeWebSocket();
+    this.initializingPromise = this.initializeWebSocket();
   }
   /**
    * Fetches the session ID from the ftrack server.
@@ -235,10 +236,10 @@ export default class SimpleSocketIOClient {
     }
     this.reconnecting = false;
     this.reconnectionAttempts = 0; // Reset reconnection attempts
-    this.handleEvent("connect", {});
-    this.flushPacketQueue();
     // Set connected property to true
     this.socket.connected = true;
+    this.handleEvent("connect", {});
+    this.flushPacketQueue();
   }
   /**
    * Handles WebSocket closing
@@ -313,11 +314,14 @@ export default class SimpleSocketIOClient {
     const dataString = eventData ? `:::${JSON.stringify(payload)}` : "";
     const packet = `${PACKET_TYPES.event}${dataString}`;
 
-    if (this.webSocket?.readyState === WebSocket.OPEN) {
+    if (this.isConnected()) {
       this.webSocket.send(packet);
     } else {
       this.packetQueue.push(packet);
     }
+    if (!this.socket.connected) {
+      this.reconnect();
+    }
   }
   /**
    * Send the queued packets to the server
@@ -404,15 +408,16 @@ export default class SimpleSocketIOClient {
    * @private
    * @param randomizedDelay
    */
-  private attemptReconnect(): void {
+  private async attemptReconnect(): Promise<void> {
+    await this.initializingPromise;
     // Check if already connected or if active reconnection attempt ongoing.
     if (this.socket.connected || this.reconnecting) {
       return;
     }
     this.reconnecting = true;
     this.reconnectionAttempts++;
-    this.initializeWebSocket();
     this.reconnectTimeout = undefined;
+    this.initializingPromise = this.initializeWebSocket();
     this.reconnect();
   }
   /**

test/simple_socketio.test.js

@@ -411,18 +411,18 @@ describe("Tests using SimpleSocketIOClient", () => {
     );
   });
   describe("Reconnection tests", () => {
-    test("attemptReconnect method initialises websocket again", () => {
+    test("attemptReconnect method initialises websocket again", async () => {
       client.initializeWebSocket = vi.fn();
 
-      client.attemptReconnect();
+      await client.attemptReconnect();
 
       expect(client.initializeWebSocket).toHaveBeenCalledTimes(1);
     });
 
-    test("attemptReconnect method increments attempts count", () => {
+    test("attemptReconnect method increments attempts count", async () => {
       const initialAttempts = client.reconnectionAttempts;
 
-      client.attemptReconnect();
+      await client.attemptReconnect();
 
       expect(client.reconnectionAttempts).toBe(initialAttempts + 1);
     });
@@ -456,7 +456,7 @@ describe("Tests using SimpleSocketIOClient", () => {
       // Reconnect should not be called yet
       expect(client.attemptReconnect).toHaveBeenCalledTimes(0);
     });
-    test("reconnect method exponentially increase delay for every attempt, stopping at the max value", () => {
+    test("reconnect method exponentially increase delay for every attempt, stopping at the max value", async () => {
       const originalRandom = Math.random;
       Math.random = vi.fn().mockReturnValue(1);
       vi.useFakeTimers();
@@ -468,6 +468,7 @@ describe("Tests using SimpleSocketIOClient", () => {
         const expectedMaxDelay = expectedMinDelay * 1.5;
         client.reconnecting = false; // Since it never gets to the actual fail state triggered by
         client.reconnect();
+        await client.initializingPromise;
         vi.advanceTimersByTime(expectedMaxDelay + 1);
         expect(client.attemptReconnect).toHaveBeenCalledTimes(i + 1);
       }