feat(microservice): register modules on init call nestjs#2350

Liuer · Jul 1, 2019 · 2487ab8 · 2487ab8
1 parent e8a66ea
commit 2487ab8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/packages/core/nest-application-context.ts b/packages/core/nest-application-context.ts
@@ -71,9 +71,9 @@ export class NestApplicationContext implements INestApplicationContext {
    * @returns {Promise<this>} The NestApplicationContext instance as Promise
    */
   public async init(): Promise<this> {
-    // Ignore if is already initialized
-    if (this.isInitialized) return;
-
+    if (this.isInitialized) {
+      return this;
+    }
     await this.callInitHook();
     await this.callBootstrapHook();
 

diff --git a/packages/microservices/nest-microservice.ts b/packages/microservices/nest-microservice.ts
@@ -103,6 +103,15 @@ export class NestMicroservice extends NestApplicationContext
     return this;
   }
 
+  public async init(): Promise<this> {
+    if (this.isInitialized) {
+      return this;
+    }
+    await super.init();
+    await this.registerModules();
+    return this;
+  }
+
   public listen(callback: () => void) {
     !this.isInitialized && this.registerModules();