装饰器

写在前面：本章只介绍 TypeScript 5.0+ 的装饰器用法，对于 5.0 以下的版本，请参考

什么是装饰器

首先，什么是装饰器呢？是这么说的：

In , the decorator pattern is a that allows behavior to be added to an individual , dynamically, without affecting the behavior of other instances of the same .

本人的蹩足翻译：在 OOP (面向对象编程)中，装饰器模式是一种允许动态地往一个对象上添加自定义行为，而又不影响该对象所属的类的其他实例的一种设计模式。

什么是 OOP 和类？做过介绍。

这句话未免过于拗口了，我们不妨换个角度去切入。

装饰器的使用场景

要知道，一切设计模式的诞生，都是为了解决某个问题。在 JavaScript 的世界中，装饰器通常出现于以下场景：

提供一种易读且容易实现的方式，修改类或者类的方法，避免出现大量重复的代码。
下面以修改类的方法为例。
首先，假设我们有一个 Animal 类：
```
class Animal {
  type: string
  constructor(type: string) {
	  this.type = type
  }
  
  greet() {
    console.log(`Hello, I'm a(n) ${this.type}!`)
  }
}

const xcat = new Animal('cat')
xcat.greet() // Hello, I'm a(n) cat!
```
该类有一个 greet 方法，和调用方打招呼。
假如说，我还希望根据不同的 type，往 console 打印不同动物的叫声呢？
聪明的你或许想到了，这不就是类的继承吗！在子类的 greet() 方法中，实现不同的逻辑，再调用 super.greet() 即可。
```
class Xcat extends Animal {
  constructor() {
    super('cat')
  }
  
  greet() {
    console.log('meow~ meow~')
    super.greet()
  }
}

const xcat = new Xcat()
xcat.greet() // meow~ meow~
             // Hello, I'm a(n) cat!
```
用装饰器实现，也不妨为一种思路，比如在 Animal 类中，为 greet() 方法添加「打印不同动物叫声的」行为:
```
class Animal {
  type: string
  constructor(type: string) {
	  this.type = type
  }

  @yelling
  greet() {
    console.log(`Hello, I'm a(n) ${this.type}!`)
  }
}

const typeToYellingMap = {
  cat: 'meow~ meow~'
}

function yelling(originalMethod: any, context: ClassMethodDecoratorContext) {
  return function(...args: any[]) {
    console.log(typeToYellingMap[this.type])
    originalMethod.call(this, ...args)
  }
}

const xcat = new Animal('cat')
xcat.greet() // meow~ meow~
             // Hello, I'm a(n) cat!
```
在 Animal.greet() 方法上出现的 @yelling ，就是 TypeScript 中装饰器的写法，即 @ + 函数名的组合。
上述示例对装饰器的应用属于方法装饰器，此类装饰器本身接收两个参数，一是被装饰的方法，二是方法装饰器的上下文。方法装饰器应返回一个函数，此函数在运行时真正被执行。在上述例子中，我们在装饰器返回的函数中做了两件事情：
1. 打印相应类别的动物的叫声。
2. 调用 originalMethod.call(this, …args) ，确保原方法（即装饰器所装饰的方法）能够正确地被执行。
结合「依赖注入」这一设计模式，优化模块与 class 的依赖关系。
什么是依赖注入呢？引用同事的原话：
依赖注入其实是将一个模块所依赖的部分作为参数传入，而不是由模块自己去构造。
可见，依赖注入解决了实际工程项目中，类、模块间依赖关系层级复杂的问题，将构造单例的行为交由实现依赖注入的框架去处理。
举个例子：
```
@injectable
class Dog implements IAnimal {
  sayHi() {
    console.log('woof woof woof')
  }
}

@injectable
class Cat implements IAnimal {
  sayHi() {
    console.log('meow meow meow')
  }
}

class AnimalService {
  constructor(
    @inject dog: Dog
    @inject cat: Cat
  ) {
    this._dog = dog
    this._cat = cat
  }
  
  sayHiByDog() {
    this._dog.sayHi()
  }
  
  sayHiByCat() {
    this._cat.sayHi()
  }
}
```
在上述代码中，@injectable 将一个类标记为「可被注入的」，在面向业务的类（即 AnimalService）中，使用 @inject 注入此类的单例，实现了「依赖倒置」。注意到这里的 implements IAnimal 用法，也是实战中依赖注入运用的精妙之处 —— 关心接口，而非具体实现。

实现「AOP」，即 Aspect-oriented programming，面向切面编程。

所谓的「切面」，可以理解成，在复杂的各个业务维度中，只关注一个维度的事务。

例如，使用装饰器，实现对类的某个方法的执行时间记录：

class MyService {
  @recordExecution
  myFn() {
    // do something...
  }
}

function recordExecution(originalMethod: any, context: ClassMethodDecoratorContext) {
  return function(...args: any[]) {
    console.time('mark execution')
    originalMethod.call(this, ...args)
    console.timeEnd('mark execution')
  }
}

装饰器的类别

装饰器是一种函数，写成@ + 函数名，可以用来装饰四种类型的值。
类
类的属性
类的方法
属性存取器（accessor, getter, setter）

装饰器的执行步骤如下。
计算各个装饰器的值，按照从左到右，从上到下的顺序。
调用方法装饰器。
调用类装饰器。

不管是哪种类型的装饰器，它们的函数签名都可以认为是一致的，即均接收 value, context 两个参数，前者指被装饰的对象，后者指一个存储了上下文信息的对象。

context 与 metadata 二三讲

四种装饰器的 context，均包含以下信息：

kind
描述被装饰的 value 的类型，可取 class, method, field, getter, setter, accessor 这些值。
name
描述被装饰的 value 的名字。
addInitializer
一个方法，接收一个回调函数，使得开发者可以侵入 value 的初始化过程作修改。
对 class 来说，这个回调函数会在类定义最终确认后调用，即相当于在初始化过程的最后一步。
对其他的 value 来说，如果是被 static 所修饰的，则会在类定义期间被调用，且早于其他静态属性的赋值过程；否则，会在类初始化期间被调用，且早于 value 自身的初始化。
以下是 @bound 类方法装饰器的例子，该装饰器自动为方法绑定 this：
```
const bound = (value, context: ClassMemberDecoratorContext) {
  if (context.private) throw new TypeError("Not supported on private methods.");
  context.addInitializer(function () {
      this[context.name] = this[context.name].bind(this);
  });
}
```
metadata
举个例子：
```
function meta(key, value) {
  return (_, context) => {
    context.metadata[key] = value;
  };
}

@meta('a', 'x')
class C {
  @meta('b', 'y')
  m() {}
}

C[Symbol.metadata].a; // 'x'
C[Symbol.metadata].b; // 'y'
```
在上述程序中，我们通过访问类的 Symbol.metadata ，读取到了 meta 装饰器所写入的元数据。对元数据的访问，有且仅有这一种形式。
注意一点，metadata 是作用在类上的，即使它的位置在类方法上。想实现细粒度的元数据存储，可以考虑手动维护若干 WeakMap。

除了类装饰器以外，其他3种装饰器的 context 还拥有以下 3 个字段：

static
布尔值，描述 value 是否为 static 所修饰。
private
布尔值，描述 value 是否为 private 所修饰。

access

一个对象，可在运行时访问 value 相关数据。

以类方法装饰器为例，用 access.get 可在运行时读取方法值，access.has 可在运行时查询对象上是否有某方法，举个例子：

const typeToYellingMap = {
  cat: 'meow~ meow~',
}

let yellingMethodContext: ClassMethodDecoratorContext

class Animal {
  type: string
  constructor(type: string) {
    this.type = type
  }

  @yelling
  greet() {
    console.log(`Hello, I'm a(n) ${this.type}!`)
  }

  accessor y = 1
}

function yelling(originalMethod: any, context: ClassMethodDecoratorContext) {
  yellingMethodContext = context
  return function (this: any, ...args: any[]) {
    console.log(typeToYellingMap[this.type as keyof typeof typeToYellingMap])
    originalMethod.call(this, ...args)
  }
}

const xcat = new Animal('cat')
xcat.greet() // meow~ meow~
// Hello, I'm a(n) cat!
yellingMethodContext.access.get(xcat).call(xcat) // meow~ meow~
// Hello, I'm a(n) cat!
console.log(yellingMethodContext.access.has(xcat)) // true

getter 类别的装饰器，其 context.access 同样拥有 has, get 两个方法。

对于 setter 类别的装饰器，则是 has 与 set 方法。

filed 与 accessor 类别的装饰器，拥有 has, get, set 全部三个方法。

Previous声明合并 Next枚举

Last updated 8 months ago

Was this helpful?