From cbf80b0d32f8f37ebc3818f7c88a7458e30146f7 Mon Sep 17 00:00:00 2001
From: Pranav C <pranavxc@gmail.com>
Date: Fri, 22 Nov 2024 12:19:53 +0000
Subject: [PATCH] fix: add retry mechanism on failure

---
 .../nc-gui/composables/useApi/interceptors.ts | 68 ++++++++++---------
 1 file changed, 37 insertions(+), 31 deletions(-)

diff --git a/packages/nc-gui/composables/useApi/interceptors.ts b/packages/nc-gui/composables/useApi/interceptors.ts
index 4f546984af..bdb365f640 100644
--- a/packages/nc-gui/composables/useApi/interceptors.ts
+++ b/packages/nc-gui/composables/useApi/interceptors.ts
@@ -1,6 +1,8 @@
 import type { Api } from 'nocodb-sdk'
 const DbNotFoundMsg = 'Database config not found'
 
+const TIMEOUT_RETRY_COUNT = 1
+
 export function addAxiosInterceptors(api: Api<any>) {
   const state = useGlobal()
   const router = useRouter()
@@ -61,40 +63,44 @@ export function addAxiosInterceptors(api: Api<any>) {
         return Promise.reject(error)
       }
 
-      try {
-        const token = await state.refreshToken({
-          axiosInstance,
-          skipLogout: true,
-        })
-
-        if (!token) {
-          await state.signOut({
-            redirectToSignin: !isSharedPage,
-            skipApiCall: true,
+      let retry = 0
+      do {
+        try {
+          const token = await state.refreshToken({
+            axiosInstance,
+            skipLogout: true,
           })
-          return Promise.reject(error)
-        }
-
-        const config = error.config
-        config.headers['xc-auth'] = token
 
-        const response = await axiosInstance.request(config)
-        return response
-      } catch (refreshTokenError) {
-        if ((refreshTokenError as any)?.code === 'ERR_CANCELED') {
-          return Promise.reject(refreshTokenError)
+          if (!token) {
+            await state.signOut({
+              redirectToSignin: !isSharedPage,
+              skipApiCall: true,
+            })
+            return Promise.reject(error)
+          }
+
+          const config = error.config
+          config.headers['xc-auth'] = token
+
+          const response = await axiosInstance.request(config)
+          return response
+        } catch (refreshTokenError) {
+          if ((refreshTokenError as any)?.code === 'ERR_CANCELED') {
+            return Promise.reject(refreshTokenError)
+          }
+
+          // if shared execution error, don't sign out
+          if (!(refreshTokenError instanceof SharedExecutionError)) {
+            await state.signOut({
+              redirectToSignin: !isSharedPage,
+              skipApiCall: true,
+            })
+            return Promise.reject(error)
+          }
+
+          if (retry >= TIMEOUT_RETRY_COUNT) return Promise.reject(error)
         }
-
-        // if shared execution error, don't sign out
-        if (!(refreshTokenError instanceof SharedExecutionError)) {
-          await state.signOut({
-            redirectToSignin: !isSharedPage,
-            skipApiCall: true,
-          })
-        }
-
-        return Promise.reject(error)
-      }
+      } while (retry++ < TIMEOUT_RETRY_COUNT)
     },
   )