如何验证参数字符串是否为Nestjs中的MongoId

Question

我的控制器中有请求，@Param是MongoId的字符串版本。如果我使用无效的字符串格式（与MongoId格式不匹配）调用此请求，则该请求将一直执行，直到MongoDB调用抛出内部服务器错误。

我如何验证，例如"aaa"或"ANWPINREBAFSOFASD"未被验证，并在我的请求中尽早停止

电流控制器endpoint：

@Get(':id')
  @ApiOperation({ summary: 'Get nice information' })
  findOne(
    @Param('id') id: string) {
    return this.niceService.findOne(id);
  }

该服务称为：

async findOne(id: string): Promise<NiceDocument> {

    const niceResult: NiceDocument = await this.NiceSchema.findById(id)

    if (!niceResult) {
      throw new NotFoundException()
    }
    return table
  }

Answer 1

答案是使用自定义验证管道：

创建管道并将其导出：

import { ArgumentMetadata, BadRequestException, Injectable, PipeTransform } from "@nestjs/common";
import {ObjectId} from 'mongodb'

@Injectable()
export class ValidateMongoId implements PipeTransform<string> {
  transform(value: string, metadata: ArgumentMetadata): string{
      if(ObjectId.isValid(value)){
          if((String)(new ObjectId(value)) === value)
              return value;        
          throw new BadRequestException
      }
      throw new BadRequestException
  
  };
}

使用控制器中的管道验证字符串

@Get(':id')
  @ApiOperation({ summary: 'Get nice information' })
  findOne(
    @Param('id', ValidateMongoId) id: string) {
    return this.niceService.findOne(id);
  }

或者，如果您使用mongoDB而不是猫鼬，您可以将管道中的返回类型从string更改为ObjectId，猫鼬支持字符串格式的带有id的请求