跳到主要内容
版本：2.0.0

高级装饰器 API

Midway 内部有一套标准的装饰器管理 API，用来将装饰器对接依赖注入容器，实现扫描和扩展，这些 API 方法我们都从 @midwayjs/decorator  包进行导出。

通过装饰器高级 API，我们可以自定义装饰器，并且将元数据附加其中，内部的各种装饰器都是通过该能力实现的。

常见的扩展 API 有:

装饰器

  • saveModule  用于保存某个类到某个装饰器
  • listModule  获取所有绑定到某类型装饰器的 class

元信息 （对应 reflect-metadata)

  • saveClassMetadata  保存元信息到 class
  • attachClassMetadata  附加元信息到 class
  • getClassMetadata  从 class 获取元信息
  • savePropertyDataToClass   保存属性的元信息到 class
  • attachPropertyDataToClass  附加属性的元信息到 class
  • getPropertyDataFromClass  从 class 获取属性元信息
  • listPropertyDataFromClass  列出 class 上保存的所有的属性的元信息
  • savePropertyMetadata  保存属性元信息到属性本身
  • attachPropertyMetadata  附加属性元信息到属性本身
  • getPropertyMetadata   从属性上获取保存的元信息

快捷操作

  • getProviderId  获取 class 上 provide 出来的 id
  • getObjectDefinition  获取对象定义（ObjectDefiniton)
  • getParamNames  获取一个函数的所有参数名
  • clearAllModule  清理装饰器对应的 class map（对应 saveModule）

自定义装饰器​

我们举一个例子，如果我们需要定义一个类装饰器 @Model ，被修饰的 class 则表明是一个 Model class，然后进一步操作。

创建一个装饰器文件，比如 src/decorator/model 。

import { Scope, ScopeEnum, saveClassMetadata, saveModule } from '@midwayjs/decorator';

const MODEL_KEY = 'decorator:model';

export function Model(): ClassDecorator {
  return (target: any) => {
    // 将装饰的类，绑定到该装饰器，用于后续能获取到 class
    saveModule(MODEL_KEY, target);
    // 保存一些元数据信息，任意你希望存的东西
    saveClassMetadata(
      MODEL_KEY,
      {
        test: 'abc',
      },
      target
    );
    // 指定 IoC 容器创建实例的作用域，这里注册为请求作用域，这样能取到 ctx
    Scope(ScopeEnum.Request)(target);
  };
}

上面只是定了了这个装饰器，我们还要实现相应的功能，这个功能文件必须在一开始就加载，midway v2 开始有生命周期的概念，可以在 configuration  中的生命周期中执行。midway1 可以在 app.ts  中执行。

// 实现 Model 装饰器功能
import { listModule } from '@midwayjs/decorator';

const MODEL_KEY = 'decorator:model';
// 可以获取到所有装饰了 @Model 装饰器的 class
const modules = listModule(MODEL_KEY);
for (let mod of modules) {
  // 实现自定义能力
  // 从 mod 上拿元数据，做不同的处理
  // 提前初始化等 app.applicationContext.getAsync(getProviderId(mod));
}

最后，我们要使用这个装饰器。

import { Model } from '../decorator/model';

// Provide 的作用是暴露出一个 IoC id，能被 IoC 扫描到
@Provide()
// Model 的作用是我们自己的逻辑能被执行（保存的元数据）
@Model()
export class UserModel {}

清理信息​

由于这些装饰器保存的元数据是全局唯一的，方法都为静态方法，所以在单测或者多 IoC 容器的场景下，需要对其进行清理。

import { clearAllModule } from '@midwayjs/decorator';

// 执行即可
clearAllModule();

方法装饰器获取上下文​

在请求链路上，如果自定义了装饰器要获取上下文往往比较困难，如果代码没有显示的注入上下文，装饰器中获取会非常困难。

在 Midway 的依赖注入的请求作用域中，我们将上下文绑定到了每个实例上，从实例的特定属性 REQUEST_OBJ_CTX_KEY  上即可获取当前的上下文，从而进一步对请求做操作。

import { REQUEST_OBJ_CTX_KEY } from '@midwayjs/core';

export function MyCustomDecorator(): MethodDecorator {
  return (target: object, propertyKey: string, descriptor: PropertyDescriptor) => {
    const method = descriptor.value;
    descriptor.value = function (...args) {
      // 指向当前上层框架的上下文对象，上层框架的上下文对象请参考各上层框架文档。
      console.log(this[REQUEST_OBJ_CTX_KEY]);

      return method.apply(this, [...args]);
    };
    return descriptor;
  };
}