fix(): use to retry option

nestjs · Sep 1, 2020 · 986ee91 · 986ee91
1 parent 687fb66
commit 986ee91
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/common/typeorm.utils.ts b/lib/common/typeorm.utils.ts
@@ -121,12 +121,16 @@ export function handleRetry(
   retryDelay = 3000,
   connectionName = DEFAULT_CONNECTION_NAME,
   verboseRetryLog = false,
+  toRetry?: (err: any) => boolean,
 ): <T>(source: Observable<T>) => Observable<T> {
   return <T>(source: Observable<T>) =>
     source.pipe(
       retryWhen((e) =>
         e.pipe(
           scan((errorCount, error: Error) => {
+            if (toRetry && !toRetry(error)) {
+              throw error;
+            }
             const connectionInfo =
               connectionName === DEFAULT_CONNECTION_NAME
                 ? ''

diff --git a/lib/typeorm-core.module.ts b/lib/typeorm-core.module.ts
@@ -203,6 +203,7 @@ export class TypeOrmCoreModule implements OnApplicationShutdown {
           options.retryDelay,
           connectionToken,
           options.verboseRetryLog,
+          options.toRetry,
         ),
       )
       .toPromise();