请解释 NestJS 的装饰器 (Decorators) 的用法。如何创建自定义装饰器？

推荐答案

在 NestJS 中，装饰器是一种特殊类型的声明，它可以附加到类声明、方法、属性或参数上，以修改它们的行为。装饰器使用 @expression 的形式，其中 expression 必须是一个函数，它会在运行时被调用，并接收被装饰的目标作为参数。

创建自定义装饰器

要创建自定义装饰器，可以使用 @nestjs/common 包中的 createParamDecorator 函数。以下是一个简单的自定义装饰器示例：

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

export const User = createParamDecorator(
  (data: string, ctx: ExecutionContext) => {
    const request = ctx.switchToHttp().getRequest();
    return data ? request.user[data] : request.user;
  },
);

在这个例子中，User 装饰器可以从请求对象中提取用户信息。如果提供了 data 参数，它将返回用户对象中的特定属性；否则，返回整个用户对象。

本题详细解读

装饰器的基本概念

装饰器是 TypeScript 和 JavaScript 中的一种语法特性，允许你在不修改类定义的情况下，动态地扩展类的功能。在 NestJS 中，装饰器被广泛用于定义路由、中间件、守卫、拦截器等。

装饰器的类型

NestJS 中常见的装饰器类型包括：

• 类装饰器：用于修饰类，例如 @Controller、@Injectable。
• 方法装饰器：用于修饰类的方法，例如 @Get、@Post。
• 参数装饰器：用于修饰方法的参数，例如 @Body、@Query。
• 属性装饰器：用于修饰类的属性，例如 @Inject。

自定义装饰器的实现

自定义装饰器通常用于提取请求中的特定数据或执行某些逻辑。createParamDecorator 函数允许你定义一个参数装饰器，它接收一个回调函数，该函数会在每次请求时被调用。

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

export const User = createParamDecorator(
  (data: string, ctx: ExecutionContext) => {
    const request = ctx.switchToHttp().getRequest();
    return data ? request.user[data] : request.user;
  },
);

在这个例子中，User 装饰器可以从请求对象中提取用户信息。ExecutionContext 提供了当前请求的上下文信息，允许你访问请求和响应对象。

使用自定义装饰器

你可以在控制器的方法中使用自定义装饰器：

@Get('profile')
async getProfile(@User() user) {
  return user;
}

在这个例子中，@User() 装饰器会自动将请求中的用户信息注入到 user 参数中。

总结

通过自定义装饰器，你可以简化代码并提高可读性，同时保持代码的灵活性和可维护性。NestJS 提供了强大的工具来创建和使用自定义装饰器，使得处理请求和响应变得更加方便。