Nestjs 如何从会话 jwt req.user 在正文请求中注入值

Question

我创建了一个“自定义装饰器”，并使用它在主体请求中注入来自 jwt 会话用户的公司代码 (ZON_EMPRESA)。

// DECORATOR
import { createParamDecorator } from '@nestjs/common';
export const Zona = createParamDecorator((data, req) => {
    req.body.ZON_EMPRESA = req.user.USU_EMPRESA;
    return req.body;
});

验证在控制器中不起作用。如果我使用标准装饰器 @Body () 验证正确。 但我无法从 jwt 会话的 req.user 中注入值为 ZON_EMPRESA 的正文

// CONTROLLER
@Put('')
async update( @Zona() body: IZona ) {
    return await this.zonaonaService.update( body );
}

// VALIDATION
import { IsString, Length, IsInt, Min } from 'class-validator';
export class IZona {        
    readonly ZON_CODIGO?: number;

    @IsInt() 
    @Min(1)
    readonly ZON_EMPRESA: number;

    @IsString() 
    @Length(5, 50)
    readonly ZON_NOME: string;

    @IsInt() 
    @Min(1)
    readonly ZON_VENDEDOR: number;
}

我该如何解决这个问题。 谢谢

Answer 1

请求装饰器不适合您想要完成的任务，因为它们旨在从请求对象中检索信息，而不是用于改变它。您应该使用 interceptor 代替，因为它将在其余装饰器之前运行，允许您正确更新请求正文，以便 @Body 和 ValidationPipe 自动拾取它（我假设您正在使用其他方式不会发生验证）。

import { ExecutionContext, Injectable, NestInterceptor } from '@nestjs/common';
import { Observable } from 'rxjs';

@Injectable()
export class CompanyCodeInterceptor implements NestInterceptor {
  intercept(
    context: ExecutionContext,
    call$: Observable<any>,
  ): Observable<any> {
    const req = context.switchToHttp().getRequest();

    req.body.ZON_EMPRESA = req.user.USU_EMPRESA;

    return call$;
  }
}

拦截器可以应用于方法、控制器或全局范围。例如，将其应用于控制器内的单个 HTTP 处理程序：

@Post()
@UseInterceptors(CompanyCodeInterceptor)
exampleHandler(@Body() model: IZona) {
   // model is properly validated here, do whatever
   return model;
}

Answer 2

代替

// DECORATOR
import { createParamDecorator } from '@nestjs/common';
export const Zona = createParamDecorator((data, req) => {
    req.body.ZON_EMPRESA = req.user.USU_EMPRESA;
    return req.body;
});

在控制器中使用User参数装饰器，并从用户那里获取此信息，并将其传递给服务。

// DECORATOR
import { createParamDecorator } from '@nestjs/common';
export const User = createParamDecorator((data, req) => {
    return req.user;
});

// CONTROLLER
@Put('')
async update( @User() user: IUser ) {
    return await this.zonaonaService.update( user.USU_EMPRESA );
}