Typescript 装饰器

我想绝大多数开发人员都见识过 java 里的 annotation,经典的 @ 图标, 如 @Override:

class Derived extends Base {
  @Override
  public void run() {...}
}

Java @annotation 常常运用于面向切片编程(AOP)——利用一些附加属性监听或修改类中元素。那 Typescript 作为一种新兴语言,它有没有自己的 “annotation” 语法呢?答案是,有;并且它有一个特定的名字叫 Decorator(修饰器)

Decorator 简介

TS 很早就在语法里加入了 Decorator 的概念,和 Java 类似,也是一个 @expression 标签。只不过 Decorator 在tc39里迟迟没有离开过 Stage 2,TS至今不敢轻举妄动。(TS 声称兼容 ECMAScript,早些年过早推出实验性功能,结果被 ECMAScript 坑了)。所以想尝鲜的话,还得在 tsconfig.json 里添加如下内容:

{
  "compilerOptions": {
      "target": "ES6",
      "experimentalDecorators": true
  }
}

我们看一下修饰器在 TS 中的使用方式:

@classDecorator
class Persion {
    @propertyDecorator
    private name: string;
    constructor(name: string) {
        this.name = name;
    }

    @methodDecorator
    public greet(@parameterDecorator message: string) {}

    @accessorDecorator
    get name() {}
}

有如下五种装饰器,分别是添加在类、属性、方法、参数和访问者头上的@修饰符;主要功能就是在运行时,观察或是修改它们的装饰对象。

  • Class Decorator
  • Property Decorator
  • Method Decorator
  • Parameter Decorator
  • Accessor Decorator

Class Decorator

先从类修饰器开始讲。类修饰器作用的对象是 class 的构造器——constructor。

老规矩,我们先定义一个 Onion 类,它只包含一个叫 name 的公共属性;我们为它的构造器定义一个 type —— ConstructorTpye 留作后用。

class Onion {
  constructor(public readonly name: string) {}
}

type ConstructorTpye = new(name: string) => Onion;

接着,我们再看看类修饰器在 TS 里的定义:是一个参数为函数、返回为函数(或void)的函数。(有点绕!)

declare type ClassDecorator = <T extends Function>(target: T) => T | void

实际使用中,上述的参数和返回类型事实上就是class构造器的类型,在我们的案例里就是刚才准备了的 ConstructorTpye。看一下具体实现,大家体会一下:

function overrideName(constructor: ConstructorTpye) {
  return class extends constructor {
    public name: string = 'override';
  };
}

使用方法如下:

@overrideName
class Onion {
  public name: string;
  constructor(name: string) {
    this.name = name;
  }
}

const onion: Onion = new Onion('Onion');
console.log(onion.name); // override

嗯,onion.name的输出是 override,不是 new 里的Onion;说明构造器在运行时被我们自定义的 Decorator 给替换掉了。事实上,类修饰器的原理就是将原始的构造函数以参数形式传入,并以新的形式传出。我们甚至可以给类添加多个修饰器,如下:

@f @g
class Onion{
  ...
}

@f@g 的作用机理相当于数学上的函数组合——f(g(constructor))

当然更多的时候,你会看到类修饰器是这么用的:

@colorDecorator('red')
class Onion {
  ...
}

修饰器里会传一个参数,这就是所谓的修饰器工厂。了解工厂方法的朋友应该很快能想到他的实现了:

function colorDecorator(color: string) {
  return (constructor: ConstructorTpye ) =>
    (class extends constructor {
      public color: string = color;
    });
}

在TS里我们的实现就是一个高阶函数——colorDecorator('red') 返回的就是一个 ClassDecorator 函数。上述代码,我们通过修饰器工厂给类添加了一个新的公共属性—— color,并对其赋值。

@colorDecorator('red')
class Onion {
  public name: string;
  constructor(name: string) {
    this.name = name;
  }
}

const onion: Onion = new Onion('Onion');
console.log(onion); // Onion { name: 'Onion', color: 'red' }

打印结果如上所示,onion 被添加了新的属性 color;它变成了一个红色的洋葱。

Methond Decorators

有了类修饰器的基础,方法修饰器也应该不难理解了——就是在运行时监测、修改方法调用。用法也很简单:在方法前加一个@expression就行了。

class Persion {
  constructor( private firstName: string, private lastName: string) {}

  @log
  public getFullName(): string {
      return `${this.firstName} ${this.lastName}`;
  }
}

我们看一看方法修饰器的类型定义:

declare type MethodDecorator = <T>(target: object, propertyKey: string | symbol, descriptor: PropertyDescriptor) => PropertyDescriptor<T> | void;

相比类修饰器,方法修饰器的参数列表稍长一点——共三个:

  1. target:当前对象的原型链,以 Persion 类为例,它就是 Person.prototype
  2. propertyKey:方法的名字:如 getFullName
  3. descriptor:方法的描述,也就是Object.getOwnPropertyDescriptor(Person.prototype, propertyKey)

如代码示例,我们试着给 getFullName 方法添加 @log,并希望在 getFullName 被调用时打印返回结果。

export function log(
  target: object,
  propertyName: string,
  descriptor: PropertyDescriptor): void {

    const getFullName: Function = descriptor.value;

    descriptor.value = function() {
      // invoke getFullName() and get its return value
      const result: string = getFullName.call(this);
      console.log(result);
      return result;
  };
}

实现上很简单,从 descriptor 里取出被修饰方法,然后重写 descriptor.value ——即在新 function 里打印出原始方法(getFullName)的结果。看一下执行结果:

const emp: Persion = new Persion('Onion', 'Garlic');
const fullName = emp.getFullName(); // Onion Garlic

这样,即便我们没有使用console.log(fullName),也成功打印出了全名。

Others

剩下还有三种Decorator,它们比较简单,我就快速过一下了:

  • Accessor Decorator:

    它就是作用在gettersetter上的修饰器,类型和 Method Decorator 一模一样,一般就是修改一下对象的PropertyDescriptor

  • Property Decorator

    属性修饰器的类型如下,targetpropertyKey 与 Method Decorator 前两个参数表意一样,一般就是利用Object.defineProperty修改 target 的描述。

    declare type PropertyDecorator = <T>(target: object, propertyKey: string | symbol) => void;
    
  • Parameter Decorator

    参数修饰符的类型如下,targetpropertyKey 不提了,parameterIndex 是所在函数参数列表的位置,因此常对该参数进行定制化处理,比如检验一下该参是否为空。

    declare type ParameterDecorator = (target: object, propertyKey: string | symbol, parameterIndex: number) => void;
    

小结

这期介绍了TS 修饰器的使用方式。Decorator 与 java 的 @annotion 非常相似,就是利用内省机制(introspection),在运行时观察、修改被修饰对象的一种 function。它目前还处于实验阶段,我看了一下tc39提案,它的语法就与TS大相径庭——有一个叫 decorator 的语法申明;因此现阶段的实现很可能在之后有巨大的转向。不过,修饰器在各类TS框架中已被广泛应用,如vue3、nestjs等等,大家也应该紧跟时代潮流尽早接触这类新兴的技术实现。

// tc39 demo
decorator @log {
  @initialize((instance, key, value) => {
    instance[key] = value;
  })
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,133评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,682评论 3 390
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,784评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,508评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,603评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,607评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,604评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,359评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,805评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,121评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,280评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,959评论 5 339
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,588评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,206评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,442评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,193评论 2 367
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,144评论 2 352

推荐阅读更多精彩内容