From e63832fd592c99c848ee52d92535fc8d63db5e45 Mon Sep 17 00:00:00 2001
From: ITOH <to@itoh.at>
Date: Thu, 27 May 2021 12:54:10 +0200
Subject: [PATCH] fix: endless testing

---
 src/ws/identify.ts | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/ws/identify.ts b/src/ws/identify.ts
index 3501efdf4..2ee3a3877 100644
--- a/src/ws/identify.ts
+++ b/src/ws/identify.ts
@@ -51,14 +51,17 @@ export function identify(shardId: number, maxShards: number) {
   };
 
   return new Promise((resolve, reject) => {
-    ws.loadingShards.set(shardId, {
-      shardId,
-      resolve,
-      startedAt: Date.now(),
-    });
-
-    setTimeout(() => {
+    const timeout = setTimeout(() => {
       reject(`[Identify Failure] Shard ${shardId} has not received READY event in over a minute.`);
     }, 600000);
+
+    ws.loadingShards.set(shardId, {
+      shardId,
+      resolve: (args) => {
+        clearTimeout(timeout);
+        resolve(args);
+      },
+      startedAt: Date.now(),
+    });
   });
 }