feat: add delay timer when try to reconncet (#23)

* feat: add delay timer when try to reconncet * update tests
HuolalaTech · Dec 14, 2023 · b229dd6 · b229dd6
1 parent 2c84649
commit b229dd6
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 12 deletions.
diff --git a/src/utils/socket.ts b/src/utils/socket.ts
@@ -44,6 +44,8 @@ export class SocketStore {
 
   private timer: number | null = null;
 
+  private retryTimer: number | null = null;
+
   // messages store
   private messages: (SpySocket.BroadcastEvent | SpySocket.UnicastEvent)[] = [];
 
@@ -142,7 +144,14 @@ export class SocketStore {
       return;
     }
 
-    this.tryReconnect();
+    if (this.retryTimer) {
+      clearTimeout(this.retryTimer);
+    }
+
+    this.retryTimer = window.setTimeout(() => {
+      this.retryTimer = null;
+      this.tryReconnect();
+    }, 2000);
   }
 
   tryReconnect() {

diff --git a/tests/utils/socket.test.ts b/tests/utils/socket.test.ts
@@ -29,22 +29,26 @@ describe('Socket store', () => {
     expect(client.connectionStatus).toBe(true);
     client.getSocket()?.close();
 
-    await sleep();
+    await sleep(2000 + 500);
     expect(reconnect).toHaveBeenCalledTimes(1);
     expect(client.connectionStatus).toBe(true);
   });
 
-  it('Connect failed if reconnect over 3 times', async () => {
-    // @ts-ignore
-    expect(client.reconnectTimes).toBe(3);
-    server.close();
+  it(
+    'Connect failed if reconnect over 3 times',
+    async () => {
+      // @ts-ignore
+      expect(client.reconnectTimes).toBe(3);
+      server.close();
 
-    await sleep(300);
-    // @ts-ignore
-    expect(client.reconnectTimes).toBe(0);
-    // @ts-ignore
-    expect(client.reconnectable).toBe(false);
-  });
+      await sleep(2000 * 3);
+      // @ts-ignore
+      expect(client.reconnectTimes).toBe(0);
+      // @ts-ignore
+      expect(client.reconnectable).toBe(false);
+    },
+    2000 * 4,
+  );
 
   it('Stop', async () => {
     expect(client.connectionStatus).toBe(true);