从C#到TypeScript - 装饰器

在C#里面如果想要不直接修改类或方法,但给类或方法添加一些额外的信息或功能,可以想到用Attribute,这是一个十分方便的功能装饰器。
用TypeScript同样也可以利用装饰器来给类、函数、属性以及参数添加附加功能,装饰器是ES7的一个提案,在TypeScript里已经有实现可用,不过需要在tsconfig.json里启用experimentalDecorators

"compilerOptions": {
    ..., // other options
    "experimentalDecorators": true}

装饰器介绍

TypeScript中装饰器可以应用到类、方法、属性及函数参数上,而且可以同时应用多个。
装饰器的写法是@name()()可以不要,也可以在里面写一些参数。

@Testable
@Log('controller')class Controller{

    @GET
    getContent(@QueryParam arg: string): string{        return '';
    }
}

装饰器的实现

装饰器根据实现可以分两种:
一种是不带括号,和属性一样,如@Testable

function Testable(target: Function) { // 类、方法、属性、方法参数的参数各不相同
    //这里可以记录一些信息到target,或者针对target做一些处理,如seal}

另外一种是带括号的,和函数一样,如@Log('controller'),实现函数里的参数就是括号里的参数,而且需要返回一个function

function Log(name: string) { // name就是传进来的参数'controller'
    return function(target: Function) { // 类、方法、属性、方法参数的参数各不相同
        // 这里可以根据name和target来做一些处理
    }
}

类装饰器

上面的(target: Function)其实就是类的装饰器参数,指向的是类的构造函数,如果想给类加一个简单的seal功能,可以这样做:

function sealed(target: Function) {    Object.seal(target);    Object.seal(target.prototype);
}

@sealedclass Test{
    
}

Test.prototype.test = ''; // 运行时出错,不能添加

上面的sealed就是类的装饰器,target指构造函数,类装饰器就这么一个参数。

方法装饰器

方法装饰器的使用方法和类装饰器类似,只是参数不一样,方法装饰器有三个参数:

  1. 如果装饰的是静态方法,则是类的构造函数,如果是实例方法则是类的原型。

  2. 方法的名字。

  3. 方法的PropertyDescriptor
    PropertyDescriptor即属性描述符,有
    configurable 是否可以配置,如动态添加删除函数属性之类
    writable 是否可写,可以用来设置只读属性
    enumerable 是否可枚举,即是否能在for...in中能枚举到
    value 对象或属性的值

有了这些参数就可以很好的给方法添加一些功能,比如下面实现类型WebApi里的Get的路由:

const Router = Symbol(); // 唯一key,用来存装饰器的信息function GET(path?: string) { // GET带了个可选参数
    return (target: any, name: string) => setMethodDecorator(target, name, 'GET', path);
}//把method和path存起来,路由查找的时候就可以用了function setMethodDecorator(target: any, name: string, method: string, path?: string){
    target[Router] = target[Router] || {};
    target[Router][name] = target[Router][name] || {};
    target[Router][name].method = method;
    target[Router][name].path = path;
}// 通过PropertyDescriptor来设置enumerablefunction Enumerable(enumerable: boolean) { 
    return (target: any, name: string, descriptor: PropertyDescriptor) => {
        descriptor.enumerable = enumerable;
    };
}class Controller{

    @GET
    @Enumerable(true)
    getContent(arg: string): string{        return '';
    }
}

参数装饰器

方法参数同样可以有装饰器,同样有三个参数,前两个参数和方法的一致,最后一个参数是所装饰的参数的位置。
能过参数装饰器可以给方法动态的检查或设置参数值,下面是检查参数是否为空,为空则抛出异常。

const CheckNullKey = Symbol();const Router = Symbol();// 把CheckNull装饰的参数存起来function CheckNull(target: any, name: string, index: number) {
    target[Router] = target[Router] || {};
    target[Router][name] = target[Router][name] || {};
    target[Router][name].params = target[Router][name].params || [];
    target[Router][name].params[index] = CheckNullKey;
}// 找出CheckNull的参数,并检查参数值,为空则抛异常,否则继续执行方法function Check(target: any, name: string, descriptor: PropertyDescriptor) {    let method = descriptor.value;
    descriptor.value = function () {        let params = target[Router][name].params;        if (params) {            for (let index = 0; index < params.length; index++) {                if (params[index] == CheckNullKey &&  // 找到CheckNull的参数并抛异常
                    (arguments[index] === undefined || arguments[index] === null)) {                    throw new Error("Missing required argument.");
                }
            }
        }        return method.apply(this, arguments);
    }
}class Controller{

    @Check
    getContent(@CheckNull id: string): string{        console.info(id);        return id;
    }
}new Controller().getContent(null); // error : Missing required argument.

属性装饰器

用法同上,参数只有两个,和类装饰器的前两个一样,常用来标识属性的特性。

function Column(target: any, name: string) {    //把name存起来,这个column仅仅是标识出来对应数据库中的列,常用在ORM框架中}class Table{

    @Column  
    name: string;
}

另外还有属性访问器的装饰器,和方法基本一样,同样的三个参数,不过同个属性的getset只能有一个有,而且必须是先声明的那个。

class User {    private _name: string;

    @Enumerable(true)    get name(){        return this._name;
    }    set name(value: string) {        this._name = value;
    }
}

多个装饰器的执行顺序

一个声明可以添加多个装饰器,所以会有个执行先后顺序。
首先从上到下执行装饰器函数,然后再从下往上应用带括号的装饰器返回的函数。

function Test1(){    console.info('eval test1');    return function(target: any, name: string, descriptor: PropertyDescriptor){        console.info('apply test1');
    }
}function Test2(){    console.info('eval test2');    return function(target: any, name: string, descriptor: PropertyDescriptor){        console.info('apply test2');
    }
}class User1{

    @test1()
    @Test2()
    getName(){

    }
}

结果是:

eval test1eval test2
apply test2
apply test1

总之,装饰器等于引入了天然的装饰模式,给类,方法等添加额外功能。不过装饰器目前还不算太稳定,但是由于确实方便,已经有成熟项目在使用了。