perf(asyncThrottle): Improve performance (TanStack#7224)

Co-authored-by: Dominik Dorfmeister <[email protected]>

Author: webNeat

packages/query-async-storage-persister/src/asyncThrottle.ts

@@ -11,46 +11,33 @@ export function asyncThrottle<TArgs extends ReadonlyArray<unknown>>(
 ) {
   if (typeof func !== 'function') throw new Error('argument is not function.')
 
-  let running = false
-  let lastTime = 0
-  let timeout: ReturnType<typeof setTimeout>
-  let currentArgs: TArgs | null = null
+  let nextExecutionTime = 0
+  let lastArgs = null
+  let isExecuting = false
+  let isScheduled = false
 
-  const execFunc = async () => {
-    if (currentArgs) {
-      const args = currentArgs
-      currentArgs = null
+  return async (...args: TArgs) => {
+    lastArgs = args
+    if (isScheduled) return
+    isScheduled = true
+    while (isExecuting) {
+      await new Promise((done) => setTimeout(done, interval))
+    }
+    while (Date.now() < nextExecutionTime) {
+      await new Promise((done) =>
+        setTimeout(done, nextExecutionTime - Date.now()),
+      )
+    }
+    isScheduled = false
+    isExecuting = true
+    try {
+      await func(...lastArgs)
+    } catch (error) {
       try {
-        running = true
-        await func(...args)
-      } catch (error) {
         onError(error)
-      } finally {
-        lastTime = Date.now() // this line must after 'func' executed to avoid two 'func' running in concurrent.
-        running = false
-      }
-    }
-  }
-
-  const delayFunc = async () => {
-    clearTimeout(timeout)
-    timeout = setTimeout(() => {
-      if (running) {
-        delayFunc() // Will come here when 'func' execution time is greater than the interval.
-      } else {
-        execFunc()
-      }
-    }, interval)
-  }
-
-  return (...args: TArgs) => {
-    currentArgs = args
-
-    const tooSoon = Date.now() - lastTime < interval
-    if (running || tooSoon) {
-      delayFunc()
-    } else {
-      execFunc()
+      } catch {}
     }
+    nextExecutionTime = Date.now() + interval
+    isExecuting = false
   }
 }