自定义路由参数装饰器

Nest 是基于装饰器这种语言特性而创建的。它是许多常用编程语言中众所周知的概念,但在 JavaScript 世界中,这个概念仍然相对较新。所以为了更好地理解装饰器是如何工作的,你应该看看 这篇 文章。下面给出一个简单的定义:

ES2016 的装饰器是一个可以将目标对象,名称和属性描述符作为参数的返回函数的表达式。你可以通过装饰器前缀 @ 来使用它,并将其放置在您想要装饰的最顶端。装饰器可以被定义为一个类或是属性。

参数装饰器

Nest 提供了一组有用的参数装饰器,可以和 HTTP 路由处理器(route handlers)一起使用。下面是一组装饰器和普通表达式对象的对照。

@Request() req
@Response() res
@Next() next
@Session() req.session
@Param(param?: string) req.params / req.params[param]
@Body(param?: string) req.body / req.body[param]
@Query(param?: string) req.query / req.query[param]
@Headers(param?: string)           req.headers / req.headers[param]       

另外,你还可以创建自定义装饰器。为什么它很有用呢?

Node.js 的世界中,把属性值附加到 request 对象中是一种很常见的做法。然后你可以在任何时候在路由处理程器(route handlers)中手动取到它们,例如,使用下面这个构造:

  1. const user = req.user;

为了使其更具可读性和透明性,我们可以创建 @User() 装饰器并且在所有控制器中重复利用它。

user.decorator.ts

  1. import { createParamDecorator } from '@nestjs/common';
  2. export const User = createParamDecorator((data, req) => {
  3. return req.user;
  4. });

现在你可以在任何你想要的地方很方便地使用它。

  1. @Get()
  2. async findOne(@User() user: UserEntity) {
  3. console.log(user);
  4. }

传递数据(Passing data)

当装饰器的行为取决于某些条件时,可以使用 data 参数将参数传递给装饰器的工厂函数。 一个用例是自定义装饰器,它通过键从请求对象中提取属性。 例如,假设我们的身份验证层验证请求并将用户实体附加到请求对象。 经过身份验证的请求的用户实体可能类似于:

  1. {
  2. "id": 101,
  3. "firstName": "Alan",
  4. "lastName": "Turing",
  5. "email": "alan@email.com",
  6. "roles": ["admin"]
  7. }

让我们定义一个将属性名作为键的装饰器,如果存在则返回关联的值(如果不存在则返回未定义的值,或者如果尚未创建 user 对象,则返回未定义的值)。

user.decorator.ts

  1. import { createParamDecorator } from '@nestjs/common';
  2. export const User = createParamDecorator((data: string, req) => {
  3. return data ? req.user && req.user[data] : req.user;
  4. });

然后,您可以通过控制器中的 @User() 装饰器访问以下特定属性:

  1. @Get()
  2. async findOne(@User('firstName') firstName: string) {
  3. console.log(`Hello ${firstName}`);
  4. }

您可以使用具有不同键的相同装饰器来访问不同的属性。如果用户对象复杂,这可以使请求处理程序实现更容易、更可读。

使用管道

Nest 对待自定义的路由参数装饰器和这些内置的装饰器(@Body()@Param()@Query())一样。这意味着管道也会因为自定义注释参数(在本例中为 user 参数)而被执行。此外,你还可以直接将管道应用到自定义装饰器上:

  1. @Get()
  2. async findOne(@User(new ValidationPipe()) user: UserEntity) {
  3. console.log(user);
  4. }

译者署名

用户名 头像 职能 签名
@zuohuadong 自定义装饰器 - 图1 翻译 专注于 caddy 和 nest,@zuohuadong at Github
@Drixn 自定义装饰器 - 图2 翻译 专注于 nginx 和 C++,@Drixn
@Armor 自定义装饰器 - 图3 翻译 专注于 Java 和 Nest,@Armor
@franken133 自定义装饰器 - 图4 翻译 专注于 java 和 nest,@franken133