如何复用CreateDTO实现PATCH请求实体更新且避免重复定义？

针对你遇到的复用CreateDTO做PATCH更新、部分字段需设为可选的问题，这里提供几个实用的方案，不需要重复定义UpdateDTO：

方案1：利用class-validator的校验组（Groups）动态切换规则 #

这是最简洁的方案，完全基于class-validator和NestJS的原生能力，无需额外依赖：

步骤1：给DTO的校验规则添加分组 #

给CreateDTO的每个字段标记create和update分组，在update分组下设置@IsOptional：

import { IsString, IsNumber, IsOptional } from 'class-validator';

export class CreateEntityDto {
  // 创建时必填，更新时可选
  @IsString({ groups: ['create'] })
  @IsOptional({ groups: ['update'] })
  name: string;

  // 创建时必填，更新时可选
  @IsNumber({ groups: ['create'] })
  @IsOptional({ groups: ['update'] })
  age: number;

  // 始终可选
  @IsOptional({ groups: ['create', 'update'] })
  @IsString({ groups: ['create', 'update'] })
  description?: string;
}

步骤2：自定义管道动态切换校验组 #

写一个继承自ValidationPipe的自定义管道，根据请求方法自动切换校验分组：

import { ValidationPipe, ExecutionContext, ArgumentMetadata } from '@nestjs/common';

export class GroupValidationPipe extends ValidationPipe {
  async transform(value: any, metadata: ArgumentMetadata, context: ExecutionContext) {
    const req = context.switchToHttp().getRequest();
    // 根据请求方法设置校验组
    this.options.groups = req.method === 'POST' ? ['create'] : ['update'];
    return super.transform(value, metadata, context);
  }
}

步骤3：在控制器中使用管道 #

在POST和PATCH接口的@Body()装饰器中传入这个自定义管道：

import { Controller, Post, Patch, Body, Param } from '@nestjs/common';
import { CreateEntityDto } from './dto/create-entity.dto';
import { GroupValidationPipe } from './pipes/group-validation.pipe';

@Controller('entities')
export class EntitiesController {
  @Post()
  create(@Body(new GroupValidationPipe()) dto: CreateEntityDto) {
    // 处理创建逻辑
  }

  @Patch(':id')
  update(@Param('id') id: string, @Body(new GroupValidationPipe()) dto: CreateEntityDto) {
    // 处理更新逻辑
  }
}

这个方案的优点是完全复用同一个DTO，无需重复定义，规则清晰，符合框架原生生态。

方案2：结合AsyncLocalStorage和@ValidateIf动态判断请求方法 #

如果不想用校验组，可以通过NestJS的AsyncLocalStorage获取当前请求上下文，在@ValidateIf中判断请求方法：

步骤1：初始化AsyncLocalStorage #

在main.ts中设置请求上下文的存储：

import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import { AsyncLocalStorage } from 'async_hooks';

export const asyncLocalStorage = new AsyncLocalStorage();

async function bootstrap() {
  const app = await NestFactory.create(AppModule);
  
  // 用中间件把请求对象存入AsyncLocalStorage
  app.use((req, res, next) => {
    asyncLocalStorage.run(req, next);
  });

  await app.listen(3000);
}
bootstrap();

步骤2：在DTO中使用@ValidateIf判断请求方法 #

在需要设为可选的字段上，用@ValidateIf判断如果是PATCH请求则允许为空：

import { IsString, IsNumber, IsOptional, ValidateIf } from 'class-validator';
import { asyncLocalStorage } from '../main';
import { Request } from 'express';

export class CreateEntityDto {
  @IsString()
  @ValidateIf(() => {
    const req = asyncLocalStorage.getStore() as Request;
    // PATCH请求时该字段可选，其他请求（如POST）必填
    return req.method !== 'PATCH';
  })
  name: string;

  @IsNumber()
  @ValidateIf(() => {
    const req = asyncLocalStorage.getStore() as Request;
    return req.method !== 'PATCH';
  })
  age: number;

  @IsOptional()
  @IsString()
  description?: string;
}

步骤3：使用默认ValidationPipe #

控制器中直接使用默认的ValidationPipe即可，无需额外配置：

@Controller('entities')
export class EntitiesController {
  @Post()
  create(@Body() dto: CreateEntityDto) {
    // 处理创建逻辑
  }

  @Patch(':id')
  update(@Param('id') id: string, @Body() dto: CreateEntityDto) {
    // 处理更新逻辑
  }
}

这个方案适合不想用校验组的场景，但需要额外配置AsyncLocalStorage，稍微复杂一点。

方案对比 #

推荐优先使用校验组方案，因为它更符合class-validator的设计，代码更简洁易维护。

内容的提问来源于stack exchange，提问作者monkeyUser