使用 NestJS 和 class-validator 创建标题自定义验证

Question

我一直在努力使用类验证器和 NestJS 验证以及尝试验证标头内容来验证请求。我的基本接口都在工作，但现在我正在尝试以相同的方式比较一些标题字段数据。

我有这个question about the custom decorator 来尝试处理标题，但该问题的解决方案将返回一个标题。我希望能够处理所有这些，类似于处理所有 body() 数据的方式。

我需要能够创建一个自定义装饰器来提取标题字段，并能够将它们传递给类验证器 DTO。

例如，我想验证三个头字段，例如：

User-Agent = 'Our Client Apps'
Content-Type = 'application/json'
traceabilityId = uuid

还有更多的领域，但如果我能做到这一点，那么我可以推断出其余的领域。我有一个简单的控制器示例：

@Controller(/rest/package)
export class PackageController {

    constructor(
        private PackageData_:PackageService
    )
    { }

    ...

    @Post('inquiry')
    @HttpCode(HttpStatus.OK)        // Not creating data, but need body, so return 200 OK
    async StatusInquiry(
        @RequestHeader() HeaderInfo:HeadersDTO,     // This should be the Headers validation using the decorator from the question above.

我正在尝试验证请求的标头是否包含一些特定数据，并且我正在使用 NestJS。我找到了这个information。虽然这是我想要做的，而且看起来很合适，但 ClassType 引用不存在，我不确定要改用什么。

从例子中，装饰器指的是。

request-header.decorator.ts

export interface iError {
    statusCode:number;
    messages:string[];
    error:string;
}

export const RequestHeader = createParamDecorator(
async (value:  any, ctx: ExecutionContext) => {

    // extract headers
    const headers = ctx.switchToHttp().getRequest().headers;

    // Convert headers to DTO object
    const dto = plainToClass(value, headers, { excludeExtraneousValues: true });

    // Validate
    const errors: ValidationError[] = await validate(dto);

    if (errors.length > 0) {
        let ErrorInfo:IError = {
            statusCode: HttpStatus.BAD_REQUEST,
            error: 'Bad Request',
            message: new Array<string>()
        };
        
        errors.map(obj => { 
            AllErrors = Object.values(obj.constraints);    
            AllErrors.forEach( (OneError) => {
            OneError.forEach( (Key) => {
                ErrorInfo.message.push(Key);
            });
        });

    // Your example, but wanted to return closer to how the body looks, for common error parsing
    //Get the errors and push to custom array
    // let validationErrors = errors.map(obj => Object.values(obj.constraints));
    throw new HttpException(`${ErrorInfo}`, HttpStatus.BAD_REQUEST);
}

// return header dto object
return dto;

},

我在将约束一般映射到字符串数组时遇到问题。

我的标头DTO.ts：

import { Expose } from 'class-transformer';
import { Equals, IsIn, IsString } from 'class-validator';
export class HeadersDTO {

    @IsString()
    @Equals('OurApp')
    @Expose({ name: 'user-agent' })
    public readonly 'user-agent':string;

    @IsString() 
    @IsIn(['PRODUCTION', 'TEST'])
    public readonly operationMode:string;
}

通过 Postman 发送请求的标头：

Content-Type:application/json
operationMode:PRODUCTION
Accept-Language:en

Answer 1

我刚刚测试了下面的代码，这是有效的。我认为您在这里缺少合适的类型，

async StatusInquiry(
    @RequestHeader() HeaderInfo:HeadersDTO,

您应该将 HeadersDTO 作为参数传入 RequestHeader 装饰器 this @RequestHeader(HeadersDTO) HeaderInfo:HeadersDTO,

然后我就这样创建了customDecorator.ts，

export const RequestHeader = createParamDecorator(
//Removed ClassType<unknown>,, I don't think you need this here
async (value:  any, ctx: ExecutionContext) => {

    // extract headers
    const headers = ctx.switchToHttp().getRequest().headers;

    // Convert headers to DTO object
    const dto = plainToClass(value, headers, { excludeExtraneousValues: true });

    // Validate
    const errors: ValidationError[] = await validate(dto);
    
    if (errors.length > 0) {
        //Get the errors and push to custom array
        let validationErrors = errors.map(obj => Object.values(obj.constraints));
        throw new HttpException(`Validation failed with following Errors: ${validationErrors}`, HttpStatus.BAD_REQUEST);
    }

    // return header dto object
    return dto;
},

);

我的 HeadersDTO.ts 文件

export class HeadersDTO 
{
  @IsDefined()
  @Expose({ name: 'custom-header' })
  "custom-header": string; // note the param here is in double quotes
}

我在查看编译后的 TS 文件时发现了这个原因，它看起来像这样，

class HeadersDTO {
}
tslib_1.__decorate([
    class_validator_1.IsDefined(),
    class_transformer_1.Expose({ name: 'custom-header' }),
    tslib_1.__metadata("design:type", String)
], HeadersDTO.prototype, "custom-header", void 0);
exports.HeadersDTO = HeadersDTO;

当我没有通过标头时出现以下错误，

[
  ValidationError {
    target: HeadersDTO { 'custom-header': undefined },
    value: undefined,
    property: 'custom-header',
    children: [],
    constraints: { isDefined: 'custom-header should not be null or undefined' }
  }
]

Answer 2

正如issue you referenced 中提到的，您需要创建一个DTO 类并将其传递给RequestHeader 装饰器。

例如

export class MyHeaderDTO {
    @IsString()
    @IsDefined()
    @Expose({ name: 'myheader1' })        // required as headers are case insensitive
    myHeader1: string;
}

...

@Get('/hello')
getHello(@RequestHeader(MyHeaderDTO) headers: MyHeaderDTO) {
    console.log(headers);
}