（七）Koa原生路由的实现

（一） 代码

const Koa = require('koa')
const app = new Koa()
const fs = require('fs')
const path = require('path')

app.use(async (ctx) => {
  const url = ctx.url
  let html = await getPage(url)
  ctx.body = html.toString() // 不然返回的是Buffer，toString转一下
})

app.listen(3000, () => {
  console.log('server is running at 3000')
})

async function getPage (url) {
  let page = ''
  switch (url) {
    case '/':
      page = 'index.html'
      break;
    case '/todo':
      page = 'todo.html'
      break;
    default: 
      page = '404.html'
  }
  let content = await render(page)
  return content
}

function render (page) {
  return new Promise((resolve, reject) => {
    let setPage = path.join(__dirname, 'views', page)
    fs.readFile(setPage, (err, data) => {
      if (err) {
        reject(err)
      } else {
        resolve(data)
      }
    })
  })
}

（二）层级路由

一般在项目中，我们一个功能一个前缀路由

router/index.js

const router = require('koa-router')()
// 前缀
router.prefix('/home')

router.get('/index', (ctx) => {
  ctx.body = '/home/index'
})

router.get('/detail', (ctx) => {
  ctx.body = '/home/detail'
})

module.exports = router

router/mine.js

const router = require('koa-router')()
// 前缀
router.prefix('/mine')

router.get('/index', (ctx) => {
  ctx.body = '/mine/index'
})

router.get('/detail', (ctx) => {
  ctx.body = '/mine/detail'
})

module.exports = router

index.js

const Koa = require('koa')
const app = new Koa()
// 引入
const HomeRouter = require('./router/index')
const MineRouter = require('./router/mine')
// 定义
app.use(HomeRouter.routes())
app.use(MineRouter.routes())

app.listen(3001, () => {
  console.log('server is running at 3001')
})