feat(ws-manual-ack): add explore ack param

Author: kim-sung-jee

packages/common/enums/route-paramtypes.enum.ts

@@ -12,4 +12,5 @@ export enum RouteParamtypes {
   HOST = 10,
   IP = 11,
   RAW_BODY = 12,
+  ACK = 13,
 }

packages/websockets/context/ws-metadata-constants.ts

@@ -1,6 +1,7 @@
 import { WsParamtype } from '../enums/ws-paramtype.enum';
 
 export const DEFAULT_CALLBACK_METADATA = {
+  [`${WsParamtype.ACK}:2`]: { index: 2, data: undefined, pipes: [] },
   [`${WsParamtype.PAYLOAD}:1`]: { index: 1, data: undefined, pipes: [] },
   [`${WsParamtype.SOCKET}:0`]: { index: 0, data: undefined, pipes: [] },
 };

packages/websockets/enums/ws-paramtype.enum.ts

@@ -3,4 +3,5 @@ import { RouteParamtypes } from '@nestjs/common/enums/route-paramtypes.enum';
 export enum WsParamtype {
   SOCKET = RouteParamtypes.REQUEST,
   PAYLOAD = RouteParamtypes.BODY,
+  ACK = RouteParamtypes.ACK,
 }

packages/websockets/gateway-metadata-explorer.ts

@@ -5,13 +5,17 @@ import {
   GATEWAY_SERVER_METADATA,
   MESSAGE_MAPPING_METADATA,
   MESSAGE_METADATA,
+  PARAM_ARGS_METADATA,
 } from './constants';
 import { NestGateway } from './interfaces/nest-gateway.interface';
+import { ParamsMetadata } from '@nestjs/core/helpers/interfaces';
+import { WsParamtype } from './enums/ws-paramtype.enum';
 
 export interface MessageMappingProperties {
   message: any;
   methodName: string;
   callback: (...args: any[]) => Observable<any> | Promise<any>;
+  isAckHandledManually: boolean;
 }
 
 export class GatewayMetadataExplorer {
@@ -38,13 +42,37 @@ export class GatewayMetadataExplorer {
       return null;
     }
     const message = Reflect.getMetadata(MESSAGE_METADATA, callback);
+    const isAckHandledManually = this.hasAckDecorator(
+      instancePrototype,
+      methodName,
+    );
+
     return {
       callback,
       message,
       methodName,
+      isAckHandledManually,
     };
   }
 
+  private hasAckDecorator(
+    instancePrototype: object,
+    methodName: string,
+  ): boolean {
+    const paramsMetadata: ParamsMetadata = Reflect.getMetadata(
+      PARAM_ARGS_METADATA,
+      instancePrototype.constructor,
+      methodName,
+    );
+
+    if (!paramsMetadata) {
+      return false;
+    }
+
+    const params = Object.values(paramsMetadata);
+    return params.some((param: any) => param.type === WsParamtype.ACK);
+  }
+
   public *scanForServerHooks(instance: NestGateway): IterableIterator<string> {
     for (const propertyKey in instance) {
       if (isFunction(propertyKey)) {