Merge pull request #1550 from jiayisheji/master

feat: Add connection error handle
nestjs · Oct 31, 2022 · 4fae301 · 4fae301
2 parents cd646d5 + 2effb51
commit 4fae301
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/lib/interfaces/mongoose-options.interface.ts b/lib/interfaces/mongoose-options.interface.ts
@@ -1,5 +1,5 @@
 import { ModuleMetadata, Type } from '@nestjs/common';
-import { ConnectOptions } from 'mongoose';
+import { ConnectOptions, MongooseError } from 'mongoose';
 
 export interface MongooseModuleOptions
   extends ConnectOptions,
@@ -9,6 +9,7 @@ export interface MongooseModuleOptions
   retryDelay?: number;
   connectionName?: string;
   connectionFactory?: (connection: any, name: string) => any;
+  connectionErrorFactory?: (error: MongooseError) => MongooseError;
 }
 
 export interface MongooseOptionsFactory {

diff --git a/lib/mongoose-core.module.ts b/lib/mongoose-core.module.ts
@@ -10,6 +10,7 @@ import {
 import { ModuleRef } from '@nestjs/core';
 import * as mongoose from 'mongoose';
 import { defer, lastValueFrom } from 'rxjs';
+import { catchError } from 'rxjs/operators';
 import { getConnectionToken, handleRetry } from './common/mongoose.utils';
 import {
   MongooseModuleAsyncOptions,
@@ -39,11 +40,16 @@ export class MongooseCoreModule implements OnApplicationShutdown {
       retryDelay,
       connectionName,
       connectionFactory,
+      connectionErrorFactory,
       ...mongooseOptions
     } = options;
 
     const mongooseConnectionFactory =
       connectionFactory || ((connection) => connection);
+
+    const mongooseConnectionError =
+      connectionErrorFactory || ((error) => error);   
+
     const mongooseConnectionName = getConnectionToken(connectionName);
 
     const mongooseConnectionNameProvider = {
@@ -59,7 +65,12 @@ export class MongooseCoreModule implements OnApplicationShutdown {
               await mongoose.createConnection(uri, mongooseOptions).asPromise(),
               mongooseConnectionName,
             ),
-          ).pipe(handleRetry(retryAttempts, retryDelay)),
+          ).pipe(
+            handleRetry(retryAttempts, retryDelay),
+            catchError((error) => {
+              throw mongooseConnectionError(error);
+            }),
+          ),
         ),
     };
     return {
@@ -87,12 +98,16 @@ export class MongooseCoreModule implements OnApplicationShutdown {
           retryDelay,
           uri,
           connectionFactory,
+          connectionErrorFactory,
           ...mongooseOptions
         } = mongooseModuleOptions;
 
         const mongooseConnectionFactory =
           connectionFactory || ((connection) => connection);
 
+        const mongooseConnectionError =
+        connectionErrorFactory || ((error) => error);
+
         return await lastValueFrom(
           defer(async () =>
             mongooseConnectionFactory(
@@ -101,7 +116,12 @@ export class MongooseCoreModule implements OnApplicationShutdown {
                 .asPromise(),
               mongooseConnectionName,
             ),
-          ).pipe(handleRetry(retryAttempts, retryDelay)),
+          ).pipe(
+            handleRetry(retryAttempts, retryDelay),
+            catchError((error) => {
+              throw mongooseConnectionError(error);
+            }),
+          ),
         );
       },
       inject: [MONGOOSE_MODULE_OPTIONS],