feature(@nestjs) close method, fixes, init (#20, #52, #40)

example/modules/client/client.controller.ts

@@ -1,19 +1,20 @@
 import { Controller } from '../../../src/common/utils/decorators/controller.decorator';
 import { Client } from '../../../src/microservices/utils/client.decorator';
-import { RequestMapping } from '../../../src/common/utils/decorators/request-mapping.decorator';
+import { Get } from '../../../src/common/utils/decorators/request-mapping.decorator';
 import { ClientProxy } from '../../../src/microservices/client/client-proxy';
 import { Observable } from 'rxjs';
 import { Transport } from '../../../src/microservices/enums/transport.enum';
 import 'rxjs/add/operator/catch';
+import { MessagePattern } from '../../../src/microservices/index';
 
 const MicroserviceClient = { transport: Transport.TCP, port: 5667 };
 
 @Controller()
 export class ClientController {
     @Client(MicroserviceClient)
-    public client: ClientProxy;
+    private client: ClientProxy;
 
-    @RequestMapping({ path: 'client' })
+    @Get('client')
     public sendMessage(req, res) {
         const pattern = { command: 'add' };
         const data = [ 1, 2, 3, 4, 5 ];
@@ -22,4 +23,10 @@ export class ClientController {
             .catch((err) => Observable.empty())
             .subscribe((result) => res.status(200).json({ result }));
     }
+
+    @MessagePattern({ command: 'add' })
+    public add(data, respond) {
+        const numbers = data || [];
+        respond(null, numbers.reduce((a, b) => a + b));
+    }
 }

example/server.ts

@@ -1,12 +1,15 @@
-import { NestFactory } from './../src/';
+import { NestFactory } from './../src/core';
 import { ApplicationModule } from './modules/app.module';
+import { Transport } from '../src/microservices/index';
 
 const port = 3001;
 const app = NestFactory.create(ApplicationModule);
-
-app.init();
-app.listen(port, (server) => {
+const microservice = app.connectMicroservice({
+    transport: Transport.TCP,
+    port: 5667,
+});
+microservice.listen(() => ({}));
+app.listen(port, () => {
     console.log('Application listen on port:', port);
-    server.close();
-    process.exit();
+    //app.close();
 });

src/common/interfaces/nest-application.interface.ts

@@ -1,5 +1,12 @@
+import { MicroserviceConfiguration } from '@nestjs/microservices';
+import { INestMicroservice } from './index';
+
 export interface INestApplication {
     init(): void;
-    listen(port: number, callback?: (server?) => void): void;
+    listen(port: number, callback?: () => void);
     setGlobalPrefix(prefix: string): void;
+    connectMicroservice(config: MicroserviceConfiguration): INestMicroservice;
+    getMicroservices(): INestMicroservice[];
+    startAllMicroservices(callback: () => void): void;
+    close(): void;
 }

src/common/interfaces/nest-microservice.interface.ts

@@ -1,3 +1,4 @@
 export interface INestMicroservice {
     listen(callback: () => void);
+    close(): void;
 }

src/common/test/utils/shared.utils.spec.ts

@@ -0,0 +1,71 @@
+import { expect } from 'chai';
+import { isUndefined, isFunction, isObject, isString, isConstructor, validatePath, isNil, isEmpty } from '../../utils/shared.utils';
+
+describe('Shared utils', () => {
+    describe('isUndefined', () => {
+        it('should returns true when obj is undefined', () => {
+            expect(isUndefined(undefined)).to.be.true;
+        });
+        it('should returns false when object is not undefined', () => {
+            expect(isUndefined({})).to.be.false;
+        });
+    });
+    describe('isFunction', () => {
+        it('should returns true when obj is function', () => {
+            expect(isFunction(() => ({}))).to.be.true;
+        });
+        it('should returns false when object is not function', () => {
+            expect(isFunction(null)).to.be.false;
+        });
+    });
+    describe('isObject', () => {
+        it('should returns true when obj is object', () => {
+            expect(isObject({})).to.be.true;
+        });
+        it('should returns false when object is not object', () => {
+            expect(isObject(3)).to.be.false;
+        });
+    });
+    describe('isString', () => {
+        it('should returns true when obj is string', () => {
+            expect(isString('true')).to.be.true;
+        });
+        it('should returns false when object is not string', () => {
+            expect(isString(false)).to.be.false;
+        });
+    });
+    describe('isConstructor', () => {
+        it('should returns true when string is equal constructor', () => {
+            expect(isConstructor('constructor')).to.be.true;
+        });
+        it('should returns false when string is not equal constructor', () => {
+            expect(isConstructor('nope')).to.be.false;
+        });
+    });
+    describe('validatePath', () => {
+        it('should returns validated path ("add / if not exists")', () => {
+            expect(validatePath('nope')).to.be.eql('/nope');
+        });
+        it('should returns same path', () => {
+            expect(validatePath('/nope')).to.be.eql('/nope');
+        });
+    });
+    describe('isNil', () => {
+        it('should returns true when obj is undefined or null', () => {
+            expect(isNil(undefined)).to.be.true;
+            expect(isNil(null)).to.be.true;
+        });
+        it('should returns false when object is not undefined and null', () => {
+            expect(isNil('3')).to.be.false;
+        });
+    });
+    describe('isEmpty', () => {
+        it('should returns true when array is empty or not exists', () => {
+            expect(isEmpty([])).to.be.true;
+            expect(isEmpty(null)).to.be.true;
+        });
+        it('should returns false when array is not empty', () => {
+            expect(isEmpty([1, 2])).to.be.false;
+        });
+    });
+});

src/core/constants.ts

@@ -1,5 +1,6 @@
 export const messages = {
     APPLICATION_START: `Starting Nest application...`,
     APPLICATION_READY: `Nest application is ready!`,
+    MICROSERVICE_READY: `Nest microservice is ready!`,
     UNKNOWN_EXCEPTION_MESSAGE: 'Unknown exception',
 };

src/core/nest-application.ts

@@ -7,14 +7,19 @@ import { Logger } from '@nestjs/common/services/logger.service';
 import { messages } from './constants';
 import { MicroservicesModule } from '@nestjs/microservices/microservices-module';
 import { Resolver } from './router/interfaces/resolver.interface';
-import { INestApplication } from '@nestjs/common';
+import { INestApplication, INestMicroservice } from '@nestjs/common';
 import { ApplicationConfig } from './application-config';
 import { validatePath } from '@nestjs/common/utils/shared.utils';
+import { MicroserviceConfiguration } from '@nestjs/microservices';
+import { NestMicroservice } from './index';
 
 export class NestApplication implements INestApplication {
     private readonly config = new ApplicationConfig();
     private readonly logger = new Logger(NestApplication.name);
-    private readonly routesResolver: Resolver;
+    private readonly routesResolver: Resolver = null;
+    private readonly microservices = [];
+    private isInitialized = false;
+    private server = null;
 
     constructor(
         private readonly container: NestContainer,
@@ -38,14 +43,36 @@ export class NestApplication implements INestApplication {
             validatePath(this.config.getGlobalPrefix()),
             router,
         );
-
         this.logger.log(messages.APPLICATION_READY);
     }
 
+    public connectMicroservice(config: MicroserviceConfiguration): INestMicroservice {
+        const instance = new NestMicroservice(this.container, config);
+        instance.setupListeners();
+
+        this.microservices.push(instance);
+        return instance;
+    }
+
+    public getMicroservices(): INestMicroservice[] {
+        return this.microservices;
+    }
+
+    public startAllMicroservices(callback: () => void) {
+        Promise.all(this.microservices.map(this.listenToPromise)).then(callback);
+    }
+
     public listen(port: number, callback?: () => void) {
-        const server = this.express.listen(port, () => {
-            callback && callback.call(callback, server);
-        });
+        (!this.isInitialized) && this.init();
+
+        this.server = this.express.listen(port, callback);
+        return this.server;
+    }
+
+    public close() {
+        SocketModule.close();
+        this.server && this.server.close();
+        this.microservices.forEach((microservice) => microservice.close());
     }
 
     public setGlobalPrefix(prefix: string) {
@@ -59,4 +86,10 @@ export class NestApplication implements INestApplication {
     private setupRoutes(instance) {
         this.routesResolver.resolve(instance);
     }
+
+    private listenToPromise(microservice: INestMicroservice) {
+        return new Promise((resolve, reject) => {
+            microservice.listen(resolve);
+        });
+    }
 }

src/core/nest-factory.ts

@@ -63,9 +63,13 @@ export class NestFactory {
                 return;
 
             if (isFunction(receiver[prop])) {
-                return (...args) => ExceptionsZone.run(() => {
-                    receiver[prop](...args);
-                });
+                return (...args) => {
+                    let result;
+                    ExceptionsZone.run(() => {
+                        result = receiver[prop](...args);
+                    });
+                    return result;
+                };
             }
             return receiver[prop];
         };

src/core/nest-microservice.ts

@@ -26,11 +26,19 @@ export class NestMicroservice implements INestMicroservice {
 
     public setupModules() {
         MicroservicesModule.setupClients(this.container);
+        this.setupListeners();
+    }
+
+    public setupListeners() {
         MicroservicesModule.setupListeners(this.container, this.server);
     }
 
     public listen(callback: () => void) {
-        this.logger.log(messages.APPLICATION_READY);
+        this.logger.log(messages.MICROSERVICE_READY);
         this.server.listen(callback);
     }
+
+    public close() {
+        this.server.close();
+    }
 }

src/core/test/application-config.spec.ts

@@ -0,0 +1,21 @@
+import { expect } from 'chai';
+import { ApplicationConfig } from '../application-config';
+
+describe('ApplicationConfig', () => {
+    let appConfig: ApplicationConfig;
+
+    beforeEach(() => {
+        appConfig = new ApplicationConfig();
+    });
+    describe('globalPath', () => {
+        it('should set global path', () => {
+            const path = 'test';
+            appConfig.setGlobalPrefix(path);
+
+            expect(appConfig.getGlobalPrefix()).to.be.eql(path);
+        });
+        it('should has empty string as a global path by default', () => {
+            expect(appConfig.getGlobalPrefix()).to.be.eql('');
+        });
+    });
+});

src/microservices/server/server-redis.ts

@@ -3,20 +3,28 @@ import { Server } from './server';
 import { NO_PATTERN_MESSAGE } from '../constants';
 import { MicroserviceConfiguration } from '../interfaces/microservice-configuration.interface';
 
+const DEFAULT_URL = 'redis://localhost:6379';
+
 export class ServerRedis extends Server {
     private readonly url: string;
-    private readonly DEFAULT_URL = 'redis://localhost:6379';
+    private sub = null;
+    private pub = null;
 
     constructor(config: MicroserviceConfiguration) {
         super();
-        this.url = config.url || this.DEFAULT_URL;
+        this.url = config.url || DEFAULT_URL;
     }
 
     public listen(callback?: () => void) {
-        const sub = this.createRedisClient();
-        const pub = this.createRedisClient();
+        this.sub = this.createRedisClient();
+        this.pub = this.createRedisClient();
+
+        this.sub.on('connect', () => this.handleConnection(callback, this.sub, this.pub));
+    }
 
-        sub.on('connect', () => this.handleConnection(callback, sub, pub));
+    public close() {
+        this.pub && this.pub.quit();
+        this.sub && this.sub.quit();
     }
 
     public createRedisClient() {
@@ -69,7 +77,7 @@ export class ServerRedis extends Server {
         };
     }
 
-    public  tryParse(content) {
+    public tryParse(content) {
         try {
             return JSON.parse(content);
         }

src/microservices/server/server-tcp.ts

@@ -19,6 +19,10 @@ export class ServerTCP extends Server {
         this.server.listen(this.port, callback);
     }
 
+    public close() {
+        this.server.close();
+    }
+
     public bindHandler(socket) {
         const sock = this.getSocketInstance(socket);
         sock.on('message', (msg) => this.handleMessage(sock, msg));

src/microservices/server/server.ts

@@ -5,6 +5,7 @@ export abstract class Server {
     protected readonly msgHandlers = {};
 
     public abstract listen(callback: () => void);
+    public abstract close(): void;
 
     public getHandlers() {
         return this.msgHandlers;

src/microservices/test/server/server-redis.spec.ts

@@ -25,6 +25,20 @@ describe('ServerRedis', () => {
             expect(onSpy.getCall(0).args[0]).to.be.equal('connect');
         });
     });
+    describe('close', () => {
+        const pub = { quit: sinon.spy() };
+        const sub = { quit: sinon.spy() };
+        beforeEach(() => {
+            (server as any).pub = pub;
+            (server as any).sub = sub;
+        });
+        it('should close pub & sub server', () => {
+            server.close();
+
+            expect(pub.quit.called).to.be.true;
+            expect(sub.quit.called).to.be.true;
+        });
+    });
     describe('handleConnection', () => {
         let onSpy: sinon.SinonSpy,
             subscribeSpy: sinon.SinonSpy,

src/microservices/test/server/server-tcp.spec.ts

@@ -20,6 +20,16 @@ describe('ServerTCP', () => {
             expect(socket.on.called).to.be.true;
         });
     });
+    describe('close', () => {
+        const tcpServer = { close: sinon.spy() };
+        beforeEach(() => {
+            (server as any).server = tcpServer;
+        });
+        it('should close server', () => {
+            server.close();
+            expect(tcpServer.close.called).to.be.true;
+        });
+    });
     describe('listen', () => {
         const serverMock = { listen: sinon.spy() };
         beforeEach(() => {