如何在NestJS中处理TypeORM实体字段唯一验证错误及自定义验证器注入问题？

我之前也碰到过这个问题，核心原因是class-validator的默认验证器实例不在Nest的DI容器管理范围内，所以没法直接注入Repository这类服务。下面给你一套完整的解决方案，既能实现自定义邮箱唯一验证，又能优雅处理异常，避免直接抛出MySQL错误导致应用崩溃：

1. 实现带依赖注入的自定义验证器 #

首先创建一个受Nest DI容器管理的验证器，这样就能注入UserRepository来查询数据库：

// src/user/custom-email.validator.ts
import { ValidatorConstraint, ValidatorConstraintInterface, ValidationArguments } from 'class-validator';
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { User } from './user.entity';

// 标记为ValidatorConstraint，同时用@Injectable让Nest管理实例
@ValidatorConstraint({ name: 'isEmailUnique', async: true })
@Injectable()
export class IsEmailUniqueConstraint implements ValidatorConstraintInterface {
  constructor(
    @InjectRepository(User)
    private readonly userRepository: Repository<User>,
  ) {}

  // 验证逻辑：支持更新时排除当前用户的邮箱
  async validate(email: string, args: ValidationArguments) {
    const userId = args.object['id']; // 从实体对象获取当前用户ID（更新场景用）
    const existingUser = await this.userRepository.findOne({ where: { email } });
    
    if (userId) {
      // 更新时：如果邮箱存在，且不属于当前用户才判定重复
      return !existingUser || existingUser.id === userId;
    }
    // 创建时：邮箱不存在则通过
    return !existingUser;
  }

  // 验证失败的默认提示语
  defaultMessage(args: ValidationArguments) {
    return `邮箱 ${args.value} 已被注册`;
  }
}

2. 创建自定义装饰器 #

用registerDecorator把验证器封装成可直接用在实体字段上的装饰器：

// 继续在custom-email.validator.ts中添加
import { registerDecorator, ValidationOptions } from 'class-validator';

export function IsEmailUnique(validationOptions?: ValidationOptions) {
  return function (object: Object, propertyName: string) {
    registerDecorator({
      target: object.constructor,
      propertyName: propertyName,
      options: validationOptions,
      constraints: [],
      validator: IsEmailUniqueConstraint, // 绑定上面的验证器
    });
  };
}

3. 在实体中使用自定义装饰器 #

修改你的user.entity.ts，给email字段添加自定义验证：

// src/user/user.entity.ts
import { Entity, Column, PrimaryGeneratedColumn } from 'typeorm';
import { IsEmail } from 'class-validator';
import { IsEmailUnique } from './custom-email.validator';

@Entity()
export class User {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ unique: true }) // 可选：保留数据库层面的唯一约束，做并发场景的兜底
  @IsEmail({ message: '请输入合法的邮箱格式' })
  @IsEmailUnique({ message: '该邮箱已被使用，请更换其他邮箱' })
  email: string;

  // 其他字段...
}

4. 在模块中注册验证器 #

必须把验证器加入模块的providers数组，让Nest的DI容器能识别并注入依赖：

// src/user/user.module.ts
import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { User } from './user.entity';
import { IsEmailUniqueConstraint } from './custom-email.validator';
import { UserService } from './user.service';

@Module({
  imports: [TypeOrmModule.forFeature([User])],
  providers: [IsEmailUniqueConstraint, UserService], // 注册验证器
  exports: [TypeOrmModule],
})
export class UserModule {}

5. 在服务层触发验证并处理异常 #

在创建/更新用户时，手动触发class-validator的验证，提前捕获错误，避免触发MySQL的唯一约束异常：

// src/user/user.service.ts
import { Injectable, BadRequestException } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository, QueryFailedError } from 'typeorm';
import { User } from './user.entity';
import { validate } from 'class-validator';

@Injectable()
export class UserService {
  constructor(
    @InjectRepository(User)
    private readonly userRepository: Repository<User>,
  ) {}

  async createUser(userData: Partial<User>) {
    const user = this.userRepository.create(userData);
    
    // 触发验证
    const validationErrors = await validate(user);
    if (validationErrors.length > 0) {
      throw new BadRequestException({
        message: '数据验证失败',
        details: validationErrors.map(err => ({
          field: err.property,
          message: Object.values(err.constraints).join(', '),
        })),
      });
    }

    // 兜底处理并发场景的数据库唯一约束错误
    try {
      return await this.userRepository.save(user);
    } catch (error) {
      if (error instanceof QueryFailedError && error.driverError.code === 'ER_DUP_ENTRY') {
        throw new BadRequestException('邮箱已被注册');
      }
      throw error; // 其他错误正常抛出
    }
  }
}

为什么不直接用Column的unique？ #

你提到的问题确实存在：数据库层面的unique约束触发时会抛出MySQL原生错误，Nest的ExceptionsHandler会直接捕获并返回500错误，没法优雅返回自定义提示。上面的方案先在应用层做验证，同时保留数据库unique约束做并发兜底，再通过捕获QueryFailedError转换成友好的业务提示，完美解决这个问题。

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