Merge branch 'message_pattern_string-key' of https://github.com/RivaCode/nest into RivaCode-message_pattern_string-key

Author: Kamil

packages/microservices/server/server.ts

@@ -1,7 +1,13 @@
 import { Logger } from '@nestjs/common/services/logger.service';
 import { loadPackage } from '@nestjs/common/utils/load-package.util';
 import { isFunction } from '@nestjs/common/utils/shared.utils';
-import { EMPTY as empty, Observable, Subscription, from as fromPromise, of } from 'rxjs';
+import {
+  EMPTY as empty,
+  Observable,
+  Subscription,
+  from as fromPromise,
+  of,
+} from 'rxjs';
 import { catchError, finalize } from 'rxjs/operators';
 import { MessageHandlers } from '../interfaces/message-handlers.interface';
 import { MicroserviceOptions, WritePacket } from './../interfaces';
@@ -24,7 +30,8 @@ export abstract class Server {
     pattern: any,
     callback: (data) => Promise<Observable<any>>,
   ) {
-    this.messageHandlers[JSON.stringify(pattern)] = callback;
+    const key = typeof pattern === 'string' ? pattern : JSON.stringify(pattern);
+    this.messageHandlers[key] = callback;
   }
 
   public send(

packages/microservices/test/server/server.spec.ts

@@ -11,7 +11,7 @@ class TestServer extends Server {
 describe('Server', () => {
   const server = new TestServer();
   const callback = () => {},
-    pattern = { test: 'test' };
+    pattern = { test: 'test pattern' };
 
   describe('add', () => {
     it(`should add handler as a stringified pattern key`, () => {
@@ -20,6 +20,12 @@ describe('Server', () => {
       const handlers = server.getHandlers();
       expect(handlers[JSON.stringify(pattern)]).to.equal(callback);
     });
+
+    it(`should add handler as string pattern key`, () => {
+      server.addHandler(pattern.test, callback as any);
+      const handlers = server.getHandlers();
+      expect(handlers[pattern.test]).to.equal(callback);
+    });
   });
   describe('send', () => {
     let stream$: Observable<string>;
@@ -73,9 +79,7 @@ describe('Server', () => {
         it('should returns Observable', async () => {
           const value = 100;
           expect(
-            await server
-              .transformToObservable(of(value))
-              .toPromise(),
+            await server.transformToObservable(of(value)).toPromise(),
           ).to.be.eq(100);
         });
       });