add retry: 3 times, interval 1.5s

Brainicism · Dec 26, 2024 · 3be502e · 3be502e
1 parent 76f7a08
commit 3be502e
Showing 1 changed file with 32 additions and 24 deletions.
diff --git a/server/src/session_manager.ts b/server/src/session_manager.ts
@@ -165,30 +165,38 @@ export class SessionManager {
 
         const bgConfig: BgConfig = {
             fetch: async (url: any, options: any): Promise<any> => {
-                try {
-                    const response = await axios.post(url, options.body, {
-                        headers: {
-                            ...options.headers,
-                            "User-Agent":
-                                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
-                        },
-                        httpsAgent: dispatcher,
-                    });
-
-                    return {
-                        ok: true,
-                        json: async () => {
-                            return response.data;
-                        },
-                    };
-                } catch (e) {
-                    return {
-                        ok: false,
-                        json: async () => {
-                            return null;
-                        },
-                        status: e.response?.status || e.code,
-                    };
+                const max_retries = 3;
+                for (let attempts = 1; attempts <= max_retries; attempts++) {
+                    try {
+                        const response = await axios.post(url, options.body, {
+                            headers: {
+                                ...options.headers,
+                                "User-Agent":
+                                    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+                            },
+                            httpsAgent: dispatcher,
+                        });
+
+                        return {
+                            ok: true,
+                            json: async () => {
+                                return response.data;
+                            },
+                        };
+                    } catch (e) {
+                        if (attempts >= max_retries) {
+                            return {
+                                ok: false,
+                                json: async () => {
+                                    return null;
+                                },
+                                status: e.response?.status || e.code,
+                            };
+                        }
+                        await new Promise((resolve) =>
+                            setTimeout(resolve, 1500),
+                        );
+                    }
                 }
             },
             globalObj: globalThis,