feat: added onconnectioncreate module option

nestjs · Mar 20, 2024 · 388edd2 · 388edd2
1 parent 399af69
commit 388edd2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
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, MongooseError } from 'mongoose';
+import { ConnectOptions, MongooseError, Connection } from 'mongoose';
 
 export interface MongooseModuleOptions extends ConnectOptions {
   uri?: string;
@@ -9,6 +9,7 @@ export interface MongooseModuleOptions extends ConnectOptions {
   connectionFactory?: (connection: any, name: string) => any;
   connectionErrorFactory?: (error: MongooseError) => MongooseError;
   lazyConnection?: boolean;
+  onConnectionCreate?: (connection: Connection) => void;
 }
 
 export interface MongooseOptionsFactory {

diff --git a/lib/mongoose-core.module.ts b/lib/mongoose-core.module.ts
@@ -43,6 +43,7 @@ export class MongooseCoreModule implements OnApplicationShutdown {
       connectionFactory,
       connectionErrorFactory,
       lazyConnection,
+      onConnectionCreate,
       ...mongooseOptions
     } = options;
 
@@ -69,6 +70,7 @@ export class MongooseCoreModule implements OnApplicationShutdown {
                 uri,
                 mongooseOptions,
                 lazyConnection,
+                onConnectionCreate,
               ),
               mongooseConnectionName,
             ),
@@ -107,6 +109,7 @@ export class MongooseCoreModule implements OnApplicationShutdown {
           connectionFactory,
           connectionErrorFactory,
           lazyConnection,
+          onConnectionCreate,
           ...mongooseOptions
         } = mongooseModuleOptions;
 
@@ -123,6 +126,7 @@ export class MongooseCoreModule implements OnApplicationShutdown {
                 uri as string,
                 mongooseOptions,
                 lazyConnection,
+                onConnectionCreate,
               ),
               mongooseConnectionName,
             ),
@@ -191,13 +195,19 @@ export class MongooseCoreModule implements OnApplicationShutdown {
     uri: string,
     mongooseOptions: ConnectOptions,
     lazyConnection?: boolean,
+    onConnectionCreate?: MongooseModuleOptions['onConnectionCreate'],
   ): Promise<Connection> {
     const connection = mongoose.createConnection(uri, mongooseOptions);
 
     if (lazyConnection) {
       return connection;
     }
 
+    const mongooseOnConnectionCreate = onConnectionCreate || (() => {});
+    if (mongooseOnConnectionCreate) {
+      mongooseOnConnectionCreate(connection);
+    }
+
     return connection.asPromise();
   }