1. // 用法
  2. var express = require("express")
  3. var app = express()
  5. app.use((req, res, next) => {
  6.   next()
  7. })
  9. app.get("/", function(req, res) {
  10.   res.send("Hello world!")
  11. })
  13. app.listen(3000)
  15. //************************************************ */
  17. // 中间件实现原理
  18. function use(route, fn) {
  19.   var handle = fn
  20.   var path = route
  22.   // 不传入route则默认为'/'，这种基本是框架处理参数的一种套路
  23.   if (typeof route !== "string") {
  24.     handle = route
  25.     path = "/"
  26.   }
  27.   // 存储中间件，与koa不同，这里会存储路由与函数的映射关系
  28.   this.stack.push({ route: path, handle: handle })
  29.   // 以便链式调用
  30.   return this
  31. }
  33. // 函数自身内部维护自增标识符，next函数，调用next函数，消费队列，把next函数传递给
  34. function handle(req, res, out) {
  35.   let index = 0
  36.   function next(err) {
  37.     // 依次取出中间件
  38.     const { router, handle } = stack[index++]
  39.     // 终止条件
  40.     if (!stack[index++]) {
  41.       defer(done, err)
  42.       return
  43.     }
  44.     // 路由匹配规则,如果不匹配，跳过当前中间件执行
  45.     if (router !== path) {
  46.       return next(err)
  47.     }
  48.     // 执行中间件，传递错误，与上下文对象，next函数
  49.     call(layer.handle, route, err, req, res, next)
  50.   }
  52.   next()
  53. }
  55. function call(handle, route, err, req, res, next) {
  56.   if (err && handle.length === 4) {
  57.     // 错误处理中间件
  58.     handle(err, req, res, next)
  59.     return
  60.   } else if (!err) {
  61.     // 请求处理中间件
  62.     handle(req, res, next)
  63.     return
  64.   }
  65. }

connect 中间件与 koa 中间件差异

  • connect中提供路由匹配的功能，Koa中间件的执行并不需要匹配路由，所以注册的中间件每一次请求都会执行
  • connect在捕获中间件的异常时，通过next携带error一个个中间件验证，直到错误处理中间件，而Koa中则是用Promise包装中间件，一旦中间件发生异常，那么会直接触发reject状态，直接在Promise的catch中处理就行
  • Koa中通过继承event，暴露error事件让开发者自定义异常处理
  • Koa中res.end由中间件执行完成之后自动调用，这样避免在connect忘记调用res.end导致用户得不到任何反馈
  • Koa中采用了async/await语法让开发者利用同步的方式编写异步代码