利用Decorator怎樣控制Koa路由

小編給大家分享一下利用Decorator怎樣控制Koa路由，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

前言

在Spring中Controller長(zhǎng)這樣

@Controller
public class HelloController{
 @RequestMapping("/hello")
 String hello() {
 return "Hello World"; 
 }
}

還有Python上的Flask框架

@app.route("/hello")
def hello():
 return "Hello World"

兩者都用decorator來(lái)控制路由，這樣寫的好處是更簡(jiǎn)潔、更優(yōu)雅、更清晰。

反觀Express或Koa上的路由

router.get('/hello', async ctx => {
 ctx.body = 'Hello World'
})

完全差了一個(gè)檔次

JS從ES6開始就有Decorator了，只是瀏覽器和Node都還沒(méi)有支持。需要用babel-plugin-transform-decorators-legacy轉(zhuǎn)義。

Decorator基本原理

首先需要明確兩個(gè)概念：

• Decorator只能作用于類或類的方法上

• 如果一個(gè)類和類的方法都是用了Decorator，類方法的Decorator優(yōu)先于類的Decorator執(zhí)行

Decorator基本原理：

@Controller
class Hello{

}

// 等同于

Controller(Hello)

Controller是個(gè)普通函數(shù)，target為修飾的類或方法

// Decorator不傳參
function Controller(target) {

}

// Decorator傳參
function Controller(params) {
 return function (target) {

 }
}

如果Decorator是傳參的，即使params有默認(rèn)值，在調(diào)用時(shí)必須帶上括號(hào)，即：

@Controller()
class Hello{

}

如何在Koa中使用Decorator

我們可以對(duì)koa-router中間件進(jìn)行包裝

先回顧一下koa-router基本使用方法：

var Koa = require('koa');
var Router = require('koa-router');

var app = new Koa();
var router = new Router();

router.get('/', async (ctx, next) => {
 // ctx.router available
});

app
 .use(router.routes())
 .use(router.allowedMethods());

再想象一下最終目標(biāo)

@Controller({prefix: '/hello'})
class HelloController{
 @Request({url: '/', method: RequestMethod.GET})
 async hello(ctx) {
 ctx.body = 'Hello World'
 }
}

類內(nèi)部方法的裝飾器是優(yōu)先執(zhí)行的，我們需要對(duì)方法重新定義

function Request({url, method}) {
 return function (target, name, descriptor) {
 let fn = descriptor.value
 descriptor.value = (router) => {
  router[method](url, async(ctx, next) => {
  await fn(ctx, next)
  })
 }
 }
}

對(duì)RequestMethod進(jìn)行格式統(tǒng)一

const RequestMethod = {
 GET: 'get',
 POST: 'post',
 PUT: 'put',
 DELETE: 'delete'
}

Controller裝飾器需將Request方法添加到Router實(shí)例并返回Router實(shí)例

import KoaRouter from 'koa-router'

function Controller({prefix}) {
 let router = new KoaRouter()
 if (prefix) {
 router.prefix(prefix)
 }
 return function (target) {
 let reqList = Object.getOwnPropertyDescriptors(target.prototype)
 for (let v in reqList) {
  // 排除類的構(gòu)造方法
  if (v !== 'constructor') {
  let fn = reqList[v].value
  fn(router)
  }
 }
 return router
 }
}

至此，裝飾器基本功能就完成了，基本使用方法為：

import {Controller, Request, RequestMethod} from './decorator'

@Controller({prefix: '/hello'})
export default class HelloController{
 @Request({url: '/', method: RequestMethod.GET})
 async hello(ctx) {
 ctx.body = 'Hello World'
 }
}

在App實(shí)例中同路由一樣use即可。

以上是“利用Decorator怎樣控制Koa路由”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！