Nest.js 从零到壹系列（五）：使用管道、DTO 验证入参，摆脱 if-else 的恐惧

前言

上一篇介绍了如何使用中间件、拦截器、过滤器打造日志系统，接下来将介绍后端永远绕不过去的痛：参数验证。
你是否曾经为了验证参数，写了一大堆 if - else ？然后还要判断各种参数类型？相似的结构在不同的方法里判断，却又要复制一遍代码？
使用 DTO 可以清晰的了解对象的结构，使用 Pipes（管道）配合 class-validator 还可以对参数类型进行判断，还可以在验证失败的时候抛出错误信息。

一、什么是 DTO？

数据传输对象（DTO)(Data Transfer Object)，是一种设计模式之间传输数据的软件应用系统。数据传输目标往往是数据访问对象从数据库中检索数据。数据传输对象与数据交互对象或数据访问对象之间的差异是一个以不具有任何行为除了存储和检索的数据（访问和存取器）。
根据定义，我们需要在代码中约定一下 DTO，还是以注册接口为例，先创建 user.dto.ts 简单定义一下：

// src/logical/user
export class RegisterInfoDTO {
  readonly accountName: string | number;
  readonly realName: string;
  readonly password: string;
  readonly repassword: string;
  readonly mobile: number;
}

其实就是输出了一个类似于声明接口的 class，表明了参数名和类型，并且是只读的。
当然，Nest 支持使用 Interface（接口）来定义 DTO，具体语法可以浏览 TypeScript 官方文档，不过 Nest 建议使用 Class 来做 DTO（就踩坑经验而言， Class 确实比 Interface 方便多了），所以 Interface 在这里就不多介绍了。
定义好 DTO 后，接下来将演示怎么和管道配合来验证参数。

二、管道

1. 概念

管道和拦截器有点像，都是在数据传输过程中的 “关卡”，只不过各司其职。
管道有两个类型:

转换：管道将输入数据转换为所需的数据输出；
验证：对输入数据进行验证，如果验证成功继续传递，验证失败则抛出异常；

ValidationPipe 是 Nest.js 自带的三个开箱即用的管道之一（另外两个是 ParseIntPipe 和 ParseUUIDPipe，现在还用不到）。

ValidationPipe 只接受一个值并立即返回相同的值，其行为类似于一个标识函数，标准代码如下：

import { PipeTransform, Injectable, ArgumentMetadata } from '@nestjs/common';
@Injectable()
export class ValidationPipe implements PipeTransform {
  transform(value: any, metadata: ArgumentMetadata) {
    return value;
  }
}

每个管道必须提供 transform() 方法。这个方法有两个参数：

value
metadata

value 是当前处理的参数，而 metadata 是其元数据。

2. 创建管道

简单介绍完一些概念后，开始实战，先创建 pipe 文件：

nest g pipe validation pipe

这里我们还需要安装两个依赖包：

yarn add class-validator class-transformer -S

然后在 validation.pipe.ts 中编写验证逻辑：

// src/pipe/validation.pipe.ts
import {
  ArgumentMetadata,
  Injectable,
  PipeTransform,
  BadRequestException,
} from '@nestjs/common';
import { validate } from 'class-validator';
import { plainToClass } from 'class-transformer';
import { Logger } from '../utils/log4js';
@Injectable()
export class ValidationPipe implements PipeTransform {
  async transform(value: any, { metatype }: ArgumentMetadata) {
    console.log(`value:`, value, 'metatype: ', metatype);
    if (!metatype || !this.toValidate(metatype)) {
      // 如果没有传入验证规则，则不验证，直接返回数据
      return value;
    }
    // 将对象转换为 Class 来验证
    const object = plainToClass(metatype, value);
    const errors = await validate(object);
    if (errors.length > 0) {
      const msg = Object.values(errors[0].constraints)[0]; // 只需要取第一个错误信息并返回即可
      Logger.error(`Validation failed: ${msg}`);
      throw new BadRequestException(`Validation failed: ${msg}`);
    }
    return value;
  }
  private toValidate(metatype: any): boolean {
    const types: any[] = [String, Boolean, Number, Array, Object];
    return !types.includes(metatype);
  }
}

3. 绑定管道

绑定管道非常简单，就和之前使用 Guards 那样，直接用修饰符绑定在 Controller 上，然后将 body 的类型指定 DTO 即可：

// src/logical/user/user.controller.ts
import { Controller, Post, Body, UseGuards, UsePipes } from '@nestjs/common';
import { AuthGuard } from '@nestjs/passport';
import { AuthService } from '../auth/auth.service';
import { UserService } from './user.service';
import { ValidationPipe } from '../../pipe/validation.pipe';
import { RegisterInfoDTO } from './user.dto'; // 引入 DTO
@Controller('user')
export class UserController {
 constructor(private readonly authService: AuthService, private readonly usersService: UserService) {}
 // JWT验证 - Step 1: 用户请求登录
 @Post('login')
 async login(@Body() loginParmas: any) {
   ...
 }
 @UseGuards(AuthGuard('jwt'))
 @UsePipes(new ValidationPipe()) // 使用管道验证
 @Post('register')
 async register(@Body() body: RegisterInfoDTO) { // 指定 DTO类型
   return await this.usersService.register(body);
 }
}

4. 完善错误提示

光有这些还不行，我们应该增加错误提示：

// src/logical/user/user.dto.ts
import { IsNotEmpty } from "class-validator";


export class RegisterInfoDTO {
  @IsNotEmpty({ message: "用户名不能为空" })
  readonly account: string | number;
  @IsNotEmpty({ message: "密码不能为空" })
  readonly password: string;
}

上面简单编写了一些常用的验证手段，class-validator 里面有非常多的验证方法，有兴趣的读者可以访问官方文档去学习：GitHub: class-validator
接下来我们测试一下，先测试为空的情况：

再看错误日志：

至此，入参验证功能已基本完成，有了这些，我们就可以摆脱各种 if - else 来验证入参了（当然，特殊的，逻辑比较复杂的还是需要的）。

总结

本篇介绍了如何定义 DTO，如何使用 Pipes 管道，以及如何配合 class-validator 进行入参验证。
定义 DTO 有人可能会觉得好麻烦，直接 any 一把梭不就好了，然后 TypeScript 就逐渐变成了 AnyScript 了。。。。

但如果不拥抱 TypeScript 的特性，那还不如直接用 JavaScript 来写，这样还更快（如 Koa、Egg 等），定义 DTO 还有一个好处，那就是可以配合 Swagger 自动生成文档，并且是可请求的，极大方便了前端阅读文档，以后的教程会说明如何操作。

至此已经结束

最后再次感谢大佬图雀社区