装饰器了解了吗？一起来看看 Nest 实现自定义装饰器吧！

2022-12-19 13:17:22

自定义装饰器：赋予 Nest.js 代码更多力量

在 Nest.js 中，自定义装饰器为我们提供了增强类行为和代码灵活性的一条强大途径。本文将深入探讨如何实现方法、类和参数装饰器，以及如何使用它们来提升 Nest.js 应用程序的性能和可维护性。

方法装饰器：增强方法行为

方法装饰器允许我们在方法执行前后注入额外的逻辑。在 Nest.js 中，我们可以使用 @MethodDecorator() 装饰器工厂函数来创建方法装饰器。

示例：实现方法执行时间统计装饰器

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

export const TrackTime = createParamDecorator((data, ctx) => {
  const start = Date.now();

  const next = ctx.switchToHttp().getNext();
  next()
    .then(() => {
      const end = Date.now();
      console.log(`Method ${ctx.getHandler().name} took ${end - start}ms`);
    })
    .catch((err) => {
      const end = Date.now();
      console.log(`Method ${ctx.getHandler().name} took ${end - start}ms`);
      throw err;
    });
});

使用此装饰器，我们可以记录方法的执行时间，这对于性能分析和优化至关重要。

类装饰器：提升类级别功能

类装饰器允许我们在类级别应用逻辑，例如添加元数据或修改类的行为。在 Nest.js 中，我们可以使用 @ClassDecorator() 装饰器工厂函数来创建类装饰器。

示例：实现类级别缓存装饰器

import { Injectable, SetMetadata } from '@nestjs/common';

export const Cacheable = () => Injectable();

export const CACHE_KEY_METADATA = 'cache_key';

export const Cache = (cacheKey: string) => (target: object) => {
  SetMetadata(CACHE_KEY_METADATA, cacheKey, target);
};

此装饰器启用对类的缓存，通过指定缓存键来控制缓存行为。

参数装饰器：处理方法参数

参数装饰器使我们能够在方法级别处理参数。在 Nest.js 中，我们可以使用 @ParamDecorator() 装饰器工厂函数来创建参数装饰器。

示例：实现参数校验装饰器

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

export const IsString = createParamDecorator((data, ctx) => {
  const value = ctx.args[0];
  if (typeof value !== 'string') {
    throw new Error('Parameter must be a string');
  }
});