前言

本文主要讲解如何使用TypeScript装饰器定义Express路由。文中出现的代码经过简化不能直接运行，完整代码的请戳：https://github.com/WinfredWang/express-decorator

1 为什么使用装饰器

当我们在使用Express时，经常要暴露RESTful服务，代码如下：

var express = require('express');var app = express();
app.get('/users', function(req, res) {
  res.send([{name:'xx'}]);
});// 路由模块化写法var router = express.Router();
app.get('/users', function(req, res) {
  res.send([{name:'xx'}]);
});

熟悉Java WEB童鞋知道

使用这种方式声明的服务非常简洁方便，免去了写一坨重复代码之苦，而且看起来更加清晰，那我们看看在Node.js中如何做。

2 需求

参照jax-rs规范，我们列出如下需求：

• 使用@Path声明RESTful服务路由

• 使用@GET/@POST/@DELETE/@PUT声明子路由

• 使用@PathParam，@QueryParam，@HeaderParam，@CookieParam，@FormParam,来接受服务参数

3 实现思路

在ES6和TypeScript中有新特性:装饰器(Decorator)，正好我们可以借助它实现我们的需求。至于装饰器用法，可以参考我的上一篇文章。

20180107195916

上图中左边是Java中定义RESTful代码，右边是Express代码，其实他们本质上是一一对应的。我们只要在装饰器的定义中实现Express 路由即可。

继续思考，我们Express 路由到底是放到那个注解中实现呢？
我们知道不同装饰器(类/方法/参数)执行顺序不同：

参数装饰器先执行，然后方法最后类装饰器

根据这个特性我们应该将核心实现放到类装饰器Path中执行是不是就可以了呢？

其实不是，我们看如下代码，我们在user-service.ts中定义了UserService服务。

@Path("/user") class UserService {    @GET("/{id}")    public getUsers(@PathParam("id") id： string) {       // TODO
    }
 }

我们定义好了服务，然后想让Node.js模块加载，我们必须在工程入口模块(main.ts)中导入上述文件
main.ts代码：

import { HelloService } from './hello-service'// TODO

上述服务代码会执行吗？也就是说
如果仅仅导入模块，而没有使用该模块的话，Node.js是否会加载这个模块呢，换句话说这个模块会执行吗？答案是NO。
为啥呀？因为Node.js对其做了优化，只有一个模块被真正用到才会加载。

上有政策，下有对策。我们就在模块引用一下。

import { HelloService } from './hello-service'HelloService; // 就是为了让Node加载它

这样好吗，当然不好。谁知道这是干嘛的。

所以我们应该换了思路，将Express 注册路由代码拿到装饰器外部，额外提供注册服务的入口，通过该注册服务入口，用户可以显式看到有哪些服务。

import { HelloService } from './hello-service';import {RegisterService } from 'xxx';

RegisterService([HelloService]);//注册服务

4 装饰器核心代码

基于上面的思考，我们在装饰器的实现中只是单纯地存储RESTful url以及参数即可，剩下服务注册工作交给RegisterService去做。

Path装饰器实现

 function Path(baseUrl: string) {    return function (target) {
        target.prototype.$Meta = {
            baseUrl: baseUrl
        }
    }
}

这里我们将RESTful路由存储到类的原型中，以便服务实例化时能获取到。

GET/POST/DELETE/PUT

function GET (url: string) => {    return (target, methodName: string, descriptor: PropertyDescriptor) => {        let meta = getMethod(target, methodName);
        meta.subUrl = url;
        meta.httpMethod = httpMehod;
    }
}

QueryParam/PathParam等实现

function PahtParam(paramType: string) {    return function (target, methodName: string, paramIndex: number) {        let meta = getMethod(target, methodName);
        meta.params.push({            name: paramName ? paramName : paramType,            index: paramIndex,            type: paramType
        });
    }
}

上述就装饰自身代码，本质上就是讲路由、http请求方法和参数存储到类的原型对象中，以便后续可以去到。

5 注册服务核心代码

路由实现

经过上面的分析，我们可知注册服务主要将Express中注册路由交由我们框架处理,核心代码如下：

function RegisterService(app, service) {    let router = Router();    // 1. 获取存储在原型对象中的http请求信息()
    let meta = getClazz(service.prototype);    // 2. 实例化服务类
    let serviceInstance = new service();    let routes = meta.routes;    for (const methodName in routes) {        let methodMeta = routes[methodName];        let httpMethod = methodMeta.httpMethod;        // 3. 回调函数
        let fn = (req, res, next) => {            let result = service.prototype[methodName].apply(serviceInstance, params);
            res.send(result);
        };        // 4. 注册路由
        router[httpMethod].apply(router, methodMeta.subUrl);
    }    // 5. 路由中间件
    app.use.apply(app, [meta.baseUrl]);
}

image 6

http请求参数处理

 @GET('/:id', [ testMidware1 ])
 list( @PathParam('id') id: string, @QueryParam('name') name: string) {    return {name:"tom", age: 10}
 }

用户编码时我们期望回调函数中的参数框架自动注入，而不是让用户自己从request中取，所以在注册服务代码中第3处，框架需要出更加参数装饰器中信息，从request中取值后注入回调函数中

// 3. 回调函数let params = extractParameters(req, res, methodMeta['params']);let fn = (req, res, next) => {    let result = service.prototype[methodName].apply(serviceInstance, params);
    res.send(result);
};// 根据参数类型，从request取出对应的值function extractParameters(req, paramMeta) {    let paramHandlerTpe = {        'query': (paramName: string) => req.query[paramName],        'path': (paramName: string) => req.params[paramName],        'form': (paramName: string) => req.body[paramName],        'cookie': (paramName: string) => req.cookies && req.cookies[paramName],        'header': (paramName) => req.get(paramName),        'request': () => req, // 获取request/response对象，做一些特别操作
        'response': () => res,
    }    let args = [];
    params.forEach(param => {
        args.push(paramHandlerTpe[param.type](param.name))
    })    
    return args;
}

response处理

 @GET('/:id', [ testMidware1 ])
 list( @PathParam('id') id: string, @QueryParam('name') name: string) {    return {name:"tom", age: 10}
 }

一个服务处理完成后，总是要向浏览器返回值的，在回调函数中直接使用return语句，而不是自己调用response.send方法， 如下代码：

// 3. 回调函数let fn = (req, res, next) => {    let result = service.prototype[methodName].apply(serviceInstance, params);    // 支持promise处理
    if (result instanceof Promise) {
        result.then(value => {
            !res.headersSent && res.send(value);
        }).catch(err => {
            next(err);
        });
    } else if (result !== undefined) {
        !res.headersSent && res.send(result);
    }
};

6 总结

以上就是我们框架处理核心代码，核心实现主要有两步：

• 装饰器本身用来存在路由信息

• 注册机制实现express路由注册（回调函数参数处理，返回值处理等）

作者：一箭中的
链接：https://www.jianshu.com/p/02eae6e89d30