update v2.1.0; fix disconnected sometimes

mimamch · Apr 27, 2023 · a574d98 · a574d98
1 parent 43e5845
commit a574d98
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "wa-multi-session",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "Multi Session Whatsapp Library",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

diff --git a/src/Socket/index.ts b/src/Socket/index.ts
@@ -64,19 +64,24 @@ export const startSession = async (
           if (connection === "close") {
             const code = (lastDisconnect?.error as Boom)?.output?.statusCode;
             let retryAttempt = retryCount.get(sessionId) ?? 0;
-            const shouldRetry =
-              code != DisconnectReason.loggedOut &&
-              (code == DisconnectReason.restartRequired || retryAttempt < 10);
-            code != DisconnectReason.restartRequired && retryAttempt++;
+            let shouldRetry;
+            if (code != DisconnectReason.loggedOut && retryAttempt < 10) {
+              shouldRetry = true;
+            }
+            if (shouldRetry) {
+              retryAttempt++;
+            }
             if (shouldRetry) {
               retryCount.set(sessionId, retryAttempt);
               startSocket();
             } else {
+              retryCount.delete(sessionId);
               deleteSession(sessionId);
               callback.get(CALLBACK_KEY.ON_DISCONNECTED)?.(sessionId);
             }
           }
           if (connection == "open") {
+            retryCount.delete(sessionId);
             callback.get(CALLBACK_KEY.ON_CONNECTED)?.(sessionId);
           }
         }