Skip to content

waynerQiu/koa-decorator-router

 
 

Repository files navigation

koa使用装饰器动态创建路由(router)

前言

在node项目,不管是koa/express路由的使用中,我们创建路由一般都是这样的姿势

router.post("/api/test", middleware, handler); // 创建路由

比如:

router.jpg

我们一般创建路由的handler会单独抽到其它文件也就是所说的controller,这样就会多了一步编写router的过程。那么这一步是否可以省略呢?

当然可以,本文带你一步步使用装饰器统一处理构建路由,这样不用在写完某一个controller的方法后再进行创建router啦,使用装饰器,我们只需要在某一个接口方法上添加路由的装饰就可以进行创建router。那么精彩来啦~

项目结构

项目结构.jpg

开始啦,一步步教你使用装饰器构建路由喽!

初始化项目

初始化package.json

新建一个目录,用vscode打开。执行npm init初始化创建package.json

安装koa项目使用的第三方包:

npm install -S  koa koa-router koa-bodyparser koa-compress
  • koa-router 管理路由
  • koa-bodyparser 读取post,put数据转化对象格式
  • koa-compress 压缩请求数据提高传输速度

配置babel支持ES6 ES7

因为项目使用装饰器配置路由,必须支持ES7语法,才需要配置babel

本文配置的是babel7版本

下面我带大家一步步配置babel

npm install -D @babel/core 
npm install -D @babel/preset-env
npm install -D @babel/plugin-proposal-decorators
npm install -D @babel/register

babel7版本 使用的是@babel,7以下是babel-xxx 这点很容易区分

下面介绍一下这些包的功能

  • @babel/core babel 核心代码
  • @babel/preset-env 编译新版的语法 如:箭头函数,但是并不转换新版api 如:Array.include 转换新版api及兼容浏览器之间的差异(兼容ie)需要 babel-polyfill
  • @babel/plugin-proposal-decorators 解析装饰器
  • @babel/register 在运行时进行即时编译,不是进行先编译在运行

然后在项目根目录创建.babelrc文件

配置预设,插件

{
  "presets": [
    ["@babel/preset-env", {
      "targets": {
        "node": "current"
      }
    }]
  ],
  "plugins": [
    ["@babel/plugin-proposal-decorators", { "legacy": true }]
  ]
}

到此,配置结束。项目比较简单,这些babel配置在本项目已经足够使用,接下来就可以愉快的使用es6,装饰器新语法了

创建node服务

新建index.js,app.js启动一个服务

index.js 主要是进行babel注册,以及启动服务的中介文件。 为什么会这样呢?在这个文件进行使用动态编译,但是编译的时候是不会编译index.js文件的,所以,这个文件还是需要使用e5的旧语法

require("@babel/register")
require("./app")

app.js才会去真正启动node服务

import Koa from "koa"
import bodyparser from "koa-bodyparser"
import compress from "koa-compress"

app.use(compress());
app.use(bodyparser());


const PORT = 8081;
app.listen(PORT,()=>{
    console.log(`启动成功! env=${process.env.NODE_ENV}`)
    console.log(`Listening at http://localhost:${PORT}`)
})

配置运行命令,启动服务

然后需要添加package.json的执行命令

"dev":"cross-env NODE_ENV=development nodemon ./index.js"

这里用到了两个node开发中常用的两个插件cross-envnodemon其主要用途如下

安装npm install -D cross-env nodemon,安装到devDependencies

  • cross-env 给项目添加环境变量区分是开发环境和生产环境,一般在执行npm命令前添加环境变量通过

这个地方添加了NODE_ENV变量,在程序中可以通过process.env.NODE_ENV获取执行命令添加的变量值

  • nodemon 可以替代node启动服务,只是nodemon功能丰富,可以监听代码的更改重新启动项目

下面执行npm run dev可以看到服务启动成功

1565227107(1).jpg

添加路由装饰器

在项目根目录新建common目录,然后在common目录下新建decorator目录,在这个目录新建index.js

no bi bi ,上代码:

/**
 * 请求方法
 */
export const RequestMethod = {
    "GET":"get",
    "POST": "post",
    "PUT": "pust",
    "DELETE": "delete",
    "OPTION": "option",
    "PATCH": "patch"
}

/**
 * 定义注册的路由数组
 */
export const controllers = [];

/**
 * 给controller添加装饰
 * @param {*} path 
 */
export function Controller(path=""){
    return function(target){
        // 给controller类添加路由前缀
        console.log(target)
        target.prefix = path;
    }
}

/**
 * 给controller类的方法添加装饰
 * url 可选
 * method 请求方法
 * middleware 中间件
 */
export function RequestMapping({url="",method="",middleware=[]}){
    return function(target,name,descriptor){
        let path = "";
        // 判断有没有定义url
        if(!url){
            // 取方法名作为路径
            path = `/${name}`;
        }else{
            // 自己定义的url
            path = url;
        }
        // 创建router需要的数据 url,method,middleware(可以没有),最终执行的方法,装饰器队对象的构造函数
        const item = {
            url:path,
            method:method,
            middleware:middleware,
            handler:target[name],
            constructor:target.constructor,
        };
        controllers.push(item);
    }
}

路由的统一创建

创建router文件夹,再继续创建index.js文件

no bi bi ,上代码:

此文件是统一创建路由的入口文件,需要传入koa实例和koa-router实例进行创建路由和路由装箱

import { controllers } from "./../common/decorator"
/**
 * 初始化路由
 */
export default (app,router) => {
    controllers.forEach((item) => {
        // 获取每个路由的前缀
        const prefix = item.constructor.prefix; 
        let url = item.url;
        if(prefix) url = `${prefix}${url}`; // 组合真正链接
        console.log(item.method,url); // 打印请求的路由method,url
        router[item.method](url, ...item.middleware, item.handler); // 创建路由
    });
    app.use(router.routes()).use(router.allowedMethods()) // 路由装箱
}

然后在 app.js进行初始化路由:

initRoutes(app,router)

使用

创建controler/index.js , middleware/index.js

middleware/index.js的测试代码

export default async (ctx,next)=>{
    console.log("middleware")
    await next();
}

controler/index.js的测试代码:




import {RequestMethod,Controller,RequestMapping } from "./../common/decorator/index"
import TestFun from './../middleware/index'

// 添加Controller前缀
@Controller("/api/test")
export default class TestController {

    // 基本面使用 /api/test/login
    @RequestMapping({
        url:"/login",
        method:RequestMethod.GET, // 定义请求方法
    })
    async login(ctx){
        ctx.body = {
            code:0,
            message:"success"
        }
    }

    // 定义有中间件的router  /api/test/test
    @RequestMapping({
        method:RequestMethod.POST, // 定义请求方法
        middleware: [TestFun] // 使用中间件
    })
    async test(ctx){
        ctx.body = {
            code:0,
            message:"success"
        }
    }

    
}

把conttroller/index.js 在 router/index.js导入并导出 export * from "../controller/index" 如果还有其它controller,那么只需要在router/index.js添加就可以了~

重启服务

1565241373(1).jpg

使用postman测试

1565240592(1).jpg

1565240617(1).jpg

ok,搞定~,以后再也不用写router啦,感觉爽了很多!

代码注释很详细,若有什么不理解或者更好的建议,欢迎各位同学留言交流

About

使用装饰器创建路由

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • JavaScript 100.0%