feature(microservices): add RequestContext to request scoped

Author: kamilmysliwiec

packages/microservices/index.ts

@@ -14,3 +14,4 @@ export * from './interfaces';
 export * from './module';
 export * from './nest-microservice';
 export * from './server';
+export * from './tokens';

packages/microservices/interfaces/index.ts

@@ -6,3 +6,4 @@ export * from './message-handler.interface';
 export * from './microservice-configuration.interface';
 export * from './packet.interface';
 export * from './pattern-metadata.interface';
+export * from './request-context.interface';

packages/microservices/interfaces/request-context.interface.ts

@@ -0,0 +1,4 @@
+export interface RequestContext<T = any> {
+  pattern: string | Record<string, any>;
+  data: T;
+}

packages/microservices/listeners-controller.ts

@@ -2,12 +2,16 @@ import { Controller } from '@nestjs/common/interfaces/controllers/controller.int
 import { createContextId } from '@nestjs/core/helpers/context-id-factory';
 import { NestContainer } from '@nestjs/core/injector/container';
 import { Injector } from '@nestjs/core/injector/injector';
-import { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper';
+import {
+  ContextId,
+  InstanceWrapper,
+} from '@nestjs/core/injector/instance-wrapper';
 import { MetadataScanner } from '@nestjs/core/metadata-scanner';
+import { REQUEST } from '@nestjs/core/router/request/request-constants';
 import { IClientProxyFactory } from './client/client-proxy-factory';
 import { ClientsContainer } from './container';
 import { RpcContextCreator } from './context/rpc-context-creator';
-import { CustomTransportStrategy } from './interfaces';
+import { CustomTransportStrategy, RequestContext } from './interfaces';
 import { ListenerMetadataExplorer } from './listener-metadata-explorer';
 import { Server } from './server/server';
 
@@ -48,8 +52,10 @@ export class ListenersController {
         }
         server.addHandler(
           pattern,
-          async data => {
+          async (data: unknown) => {
             const contextId = createContextId();
+            this.registerRequestProvider({ pattern, data }, contextId);
+
             const contextInstance = await this.injector.loadPerContext(
               instance,
               module,
@@ -88,4 +94,17 @@ export class ListenersController {
   ) {
     Reflect.set(instance, property, client);
   }
+
+  private registerRequestProvider(
+    request: RequestContext,
+    contextId: ContextId,
+  ) {
+    const coreModuleRef = this.container.getInternalCoreModuleRef();
+    const wrapper = coreModuleRef.getProviderByKey(REQUEST);
+
+    wrapper.setInstanceByContextId(contextId, {
+      instance: request,
+      isResolved: true,
+    });
+  }
 }

packages/microservices/tokens.ts

@@ -0,0 +1,3 @@
+import { REQUEST } from '@nestjs/core';
+
+export const CONTEXT = REQUEST;

sample/03-microservices/src/math/math.controller.ts

@@ -1,11 +1,19 @@
 import { Controller, Get, Inject } from '@nestjs/common';
-import { ClientProxy, MessagePattern } from '@nestjs/microservices';
+import {
+  ClientProxy,
+  CONTEXT,
+  MessagePattern,
+  RequestContext,
+} from '@nestjs/microservices';
 import { Observable } from 'rxjs';
 import { MATH_SERVICE } from './math.constants';
 
 @Controller()
 export class MathController {
-  constructor(@Inject(MATH_SERVICE) private readonly client: ClientProxy) {}
+  constructor(
+    @Inject(MATH_SERVICE) private readonly client: ClientProxy,
+    @Inject(CONTEXT) private readonly ctx: RequestContext<number[]>,
+  ) {}
 
   @Get()
   execute(): Observable<number> {
@@ -16,6 +24,7 @@ export class MathController {
 
   @MessagePattern({ cmd: 'sum' })
   sum(data: number[]): number {
+    console.log(this.ctx);
     return (data || []).reduce((a, b) => a + b);
   }
 }