fix(): multi-connection lock blocks reconnect

tiagosiebler · Aug 6, 2024 · 7e8e7d0 · 7e8e7d0
1 parent 567ff0a
commit 7e8e7d0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/lib/BaseWSClient.ts b/src/lib/BaseWSClient.ts
@@ -370,7 +370,9 @@ export abstract class BaseWebsocketClient<
         return { wsKey };
       }
 
-      if (this.wsStore.isConnectionAttemptInProgress(wsKey)) {
+      if (
+        this.wsStore.isConnectionState(wsKey, WsConnectionStateEnum.CONNECTING)
+      ) {
         this.logger.error(
           'Refused to connect to ws, connection attempt already active',
           { ...WS_LOGGER_CATEGORY, wsKey },
@@ -468,6 +470,7 @@ export abstract class BaseWebsocketClient<
 
   private reconnectWithDelay(wsKey: TWSKey, connectionDelayMs: number) {
     this.clearTimers(wsKey);
+
     if (!this.wsStore.isConnectionAttemptInProgress(wsKey)) {
       this.setWsState(wsKey, WsConnectionStateEnum.RECONNECTING);
     }

diff --git a/src/lib/websocket/WsStore.types.ts b/src/lib/websocket/WsStore.types.ts
@@ -6,6 +6,7 @@ export enum WsConnectionStateEnum {
   CONNECTED = 2,
   CLOSING = 3,
   RECONNECTING = 4,
+  // ERROR_RECONNECTING = 5,
   // ERROR = 5,
 }