feat: interceptor on individual modules

DIY0R · Jul 12, 2024 · 0eb6802 · 0eb6802
1 parent 4342d24
commit 0eb6802
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/lib/constants.ts b/lib/constants.ts
@@ -7,6 +7,7 @@ export const TARGET_MODULE = 'TARGET_MODULE';
 export const SER_DAS_KEY = 'SER_DAS_KEY';
 export const INTERCEPTOR_KEY = 'INTERCEPTOR_KEY';
 export const SERDES = 'SERDES';
+export const INTERCEPTORS = 'INTERCEPTORS';
 
 export const INITIALIZATION_STEP_DELAY = 400;
 export const DEFAULT_TIMEOUT = 40000;

diff --git a/lib/interfaces/rmq-options.interface.ts b/lib/interfaces/rmq-options.interface.ts
@@ -2,6 +2,7 @@ import { Options } from 'amqplib';
 import { LoggerService, ModuleMetadata } from '@nestjs/common';
 import { RMQIntercepterClass, RMQPipeClass } from '../common';
 import { ISerDes } from './serdes.interface';
+import { TypeRmqInterceptor } from './interceptor.interface';
 
 export interface IQueue {
   queue: string;
@@ -34,6 +35,7 @@ export interface IMessageBroker {
   replyTo?: IQueue;
   queue?: IQueue;
   serDes?: ISerDes;
+  interceptor?: TypeRmqInterceptor[];
   messageTimeout?: number;
   serviceName?: string;
 }

diff --git a/lib/rmq.module.ts b/lib/rmq.module.ts
@@ -6,7 +6,12 @@ import {
   IRabbitMQConfig,
   IGlobalOptions,
 } from './interfaces';
-import { MODULE_TOKEN, RMQ_BROKER_OPTIONS, SERDES } from './constants';
+import {
+  INTERCEPTORS,
+  MODULE_TOKEN,
+  RMQ_BROKER_OPTIONS,
+  SERDES,
+} from './constants';
 import { DiscoveryModule } from '@nestjs/core';
 import { MetaTegsScannerService, getUniqId } from './common';
 import { RmqNestjsCoreModule } from './rmq-core.module';
@@ -39,6 +44,7 @@ export class RmqNestjsModule {
       providers: [
         { provide: RMQ_BROKER_OPTIONS, useValue: options },
         { provide: SERDES, useValue: options.serDes ?? serDes },
+        { provide: INTERCEPTORS, useValue: options.interceptor ?? [] },
         { provide: MODULE_TOKEN, useFactory: getUniqId },
         RmqService,
         MetaTegsScannerService,

diff --git a/lib/rmq.service.ts b/lib/rmq.service.ts
@@ -13,6 +13,7 @@ import {
   ISerDes,
   TypeChanel,
   TypeQueue,
+  TypeRmqInterceptor,
 } from './interfaces';
 import {
   IConsumFunction,
@@ -26,6 +27,7 @@ import {
   INDICATE_REPLY_QUEUE,
   INITIALIZATION_STEP_DELAY,
   INOF_NOT_FULL_OPTIONS,
+  INTERCEPTORS,
   MODULE_TOKEN,
   NACKED,
   NON_ROUTE,
@@ -62,6 +64,7 @@ export class RmqService implements OnModuleInit, OnModuleDestroy {
     @Inject(RMQ_BROKER_OPTIONS) private readonly options: IMessageBroker,
     @Inject(RMQ_APP_OPTIONS) private readonly globalOptions: IGlobalOptions,
     @Inject(SERDES) private readonly serDes: ISerDes,
+    @Inject(INTERCEPTORS) private readonly interceptors: TypeRmqInterceptor[],
     @Inject(MODULE_TOKEN) private readonly moduleToken: string,
   ) {
     this.logger = globalOptions.appOptions?.logger
@@ -238,7 +241,9 @@ export class RmqService implements OnModuleInit, OnModuleDestroy {
       : this.serDes.deserialize(content);
   }
   private getInterceptors(consumer: MetaTegEnpoint) {
-    return consumer.interceptors.map((interceptor: any) => new interceptor());
+    return this.interceptors
+      .concat(consumer.interceptors)
+      .map((interceptor: any) => new interceptor());
   }
   private async handleMessage(
     handler: IConsumFunction,