retryOnError refactoring

trosck · Aug 11, 2022 · 54bc778 · 54bc778
1 parent 79ac672
commit 54bc778
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/core/retryOnError.ts b/core/retryOnError.ts
@@ -11,16 +11,13 @@ export default async (
   tryCount: number,
   onError: (error: any) => void = () => {}
 ) => {
-  let currentTry = -1
-  while(++currentTry < tryCount) {
+  let currentTry = 0
+  while(currentTry++ < tryCount) {
     try {
-      const isEnd = await func()
-      if (isEnd) return
-
-      currentTry = -1
+      const result = await func()
+      return result
     } catch(e) {
-      // last try
-      if (currentTry + 1 === tryCount) {
+      if (currentTry === tryCount) {
         return await onError(e)
       }
     }