使用自定义拦截器进行响应

Question

我正在使用全局拦截器来获取如下响应：

{
  "data": "",
  "statusCode": int
  "message": "string"
}

所以我创建了拦截器文件

import { CallHandler, ExecutionContext, Injectable, NestInterceptor } from "@nestjs/common";
import { map, Observable } from "rxjs";

export interface Response<T> {
    data: T;
}

@Injectable()
export class TransformationInterceptor<T> implements NestInterceptor<T, Response<T>> {
    intercept(context: ExecutionContext, next: CallHandler): Observable<Response<T>> {
        return next.handle().pipe(map(data => ({ 
            data: data,
            statusCode: context.switchToHttp().getResponse().statusCode,
            message: data.message
        })));
    }
}

并将其放入我的 main.ts

在我的控制器中：

  @Patch('/:userId')
  @HttpCode(201)
  public async updateUser(    
    @Param('userId') userId: string,
    @Body() userUpdate: UpdateUserDto): Promise<any> {      
    return await this.usersService.update(userId, userUpdate);    
  }

结果是：

{
  "data": {
    "_id": "621d07d9ea0cdc600fae0f02",    
    "username": "foo",
    "name": "stringwwww",
    "__v": 0
  },
  "statusCode": 201
}

如果我想添加我的自定义消息，我需要返回一个对象，例如：

@Patch('/:userId')
  @HttpCode(201)
  public async updateUser(    
    @Param('userId') userId: string,
    @Body() userUpdate: UpdateUserDto): Promise<any> {      
    const result = await this.usersService.update(userId, userUpdate);    
    return { message: 'User updated', result };    
  }

但在这种情况下，我有两次消息并且结构不正确：

{
  "data": {
    "message": "User updated",
    "result": {
      "_id": "621d07d9ea0cdc600fae0f02",
      "username": "foo",
      "name": "stringwwww",
      "__v": 0
    }
  },
  "statusCode": 201,
  "message": "User updated"
}

如何设置自定义（可选）消息？

I can modify my interceptors like:
@Injectable()
export class TransformationInterceptor<T> implements NestInterceptor<T, Response<T>> {
    intercept(context: ExecutionContext, next: CallHandler): Observable<Response<T>> {
        return next.handle().pipe(map(data => ({ 
            data: data.res,
            statusCode: context.switchToHttp().getResponse().statusCode,
            message: data.message
        })));
    }
}

我的控制器就像：

@Patch('/:userId')
  @HttpCode(201)
  public async updateUser(    
    @Param('userId') userId: string,
    @Body() userUpdate: UpdateUserDto): Promise<any> {      
    const result = await this.usersService.update(userId, userUpdate);    
    return { message: 'User updated', res: result };    
  }

我会得到正确的形式，但我不想

return { message: 'User updated', res: result };    

为每个控制器添加

Answer 1

实现此目的的一种方法如下，但每个控制器都将绑定到固定消息

创建响应装饰器 (response.decorator.ts)

    import { SetMetadata } from '@nestjs/common'
    
    export const ResponseMessageKey = 'ResponseMessageKey'
    export const ResponseMessage = (message: string) => SetMetadata(ResponseMessageKey, message)

为响应创建常量文件 (response.constants.ts)

    export const USER_INSERTED = 'User Inserted'
    export const USER_UPDATED = 'User Updated'
    export const USER_DELETED = 'User Deleted'

将装饰器添加到您的响应中控制器设置响应消息元数据

    @Patch('/:userId')
    @HttpCode(201)
    @ResponseMessage(USER_UPDATED)
    public async updateUser(    
      @Param('userId') userId: string,
      @Body() userUpdate: UpdateUserDto
    ): Promise<any> {      
      const result = await this.usersService.update(userId, userUpdate);    
      return result;    
    }

更新您的拦截器以从控制器上设置的元数据读取响应消息并将其添加到响应中

    import { Reflector } from '@nestjs/core'

    @Injectable()
    export class TransformationInterceptor<T>
      implements NestInterceptor<T, Response<T>>
    {
      constructor(private reflector: Reflector) {}

      intercept(
        context: ExecutionContext,
        next: CallHandler
      ): Observable<Response<T>> {
        const responseMessage = this.reflector.get<string>(
          ResponseMessageKey,
          context.getHandler()
        ) ?? ''

        return next.handle().pipe(
          map((data) => ({
            data,
            statusCode: context.switchToHttp().getResponse().statusCode,
            message: responseMessage
          }))
        )
      }
    }

您可以扩展此方法以将字符串/对象列表设置为可能的响应（元数据），并基于拦截器中的响应代码，发送特定消息作为 response.message