在使用koa-router的时候会遇到路由的配置,虽然基本的配置很简单,但是当我们的业务越来越复杂,涉及到的路由也越来越多,最开始的单文件路由配置就难以维护了。所以我们需要将路由根据模块进行拆分,从而根据模块配置相应的路由。
但是我们一般在进行拆分的时候可能会遇到下面这样的用法,这属于常规用法,但是模块多了也就不怎么好看了。
app.use(A.routes())
app.use(A.allowedMethods())
app.use(B.routes())
app.use(B.allowedMethods())
app.use(C.routes())
app.use(C.allowedMethods())
因此我们可以采用koa-compose这个方法对其进行优化。
假设我们有三个路由模块分别为A、B、C,首先新建文件夹routes,并在routes文件夹下新建A.js、B.js、C.js、index.js
routes/A.js
const Router = require('koa-router')
const router = new Router()
router.prefix('/A')
router.get('/all',(ctx,next)=>{
ctx.body = "hello A module router"
})
module.exports = router
routes/B.js
const Router = require('koa-router')
const router = new Router()
router.prefix('/B')
router.get('/all',(ctx,next)=>{
ctx.body = "hello B module router"
})
module.exports = router
routes/C.js
const Router = require('koa-router')
const router = new Router()
router.prefix('/C')
router.get('/all',(ctx,next)=>{
ctx.body = "hello C module router"
})
module.exports = router
routes/index.js
const compose = require('koa-compose')
const glob = require('glob')
const { resolve } = require('path')
registerRouter = () => {
let routers = [];
glob.sync(resolve(__dirname, './', '**/*.js'))
.filter(value => (value.indexOf('index.js') === -1))
.map(router => {
routers.push(require(router).routes())
routers.push(require(router).allowedMethods())
})
return compose(routers)
}
module.exports = registerRouter
index.js
const Koa = require('koa')
const registerRouter = require('./routes')
const app = new Koa()
app.use(registerRouter())
app.listen(3000)