koa2---源码解析---koa-compose

最新推荐文章于 2024-08-24 07:50:17 发布

原创最新推荐文章于 2024-08-24 07:50:17 发布 · 268 阅读

0 ·

CC 4.0 BY-SA版权

koa 专栏收录该内容

7 篇文章

订阅专栏

本文深入探讨了中间件的组合方式及执行流程，通过递归调用Promise.resolve实现中间件链的同步与异步执行，确保每个中间件正确处理上下文(context)并传递给下一个中间件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

function compose (middleware) {
  if (!Array.isArray(middleware)) throw new TypeError('Middleware stack must be an array!')
  for (const fn of middleware) {
    if (typeof fn !== 'function') throw new TypeError('Middleware must be composed of functions!')
  }

  // 返回闭包,this.callback 中的 fn 后续一定会使用这个闭包传入过滤后的 context(ctx)
  return function (context, next) {
    // last called middleware #
    // 初始化中间件函数数组执行的下标值
    let index = -1

    // 返回递归执行的 Promise.resolve 去执行整个中间件数组，初始从第一个开始
    return dispatch(0)
    function dispatch (i) {
      // 校验上次执行的下标 index 不能大于本次执行的传入下标 i，如果大于可能是 next(下个中间件) 执行了多次导致
      if (i <= index) return Promise.reject(new Error('next() called multiple times'))

      // 赋值当前执行的中间件函数下标值
      index = i

      let fn = middleware[i]  // 取出当前要执行的中间件函数
      if (i === middleware.length) fn = next  //如果i === middleware.length，那么此时 fn = next 一定会是空
      if (!fn) return Promise.resolve()   //返回 Promise.resolve() 宣布中间件数组执行完毕

      // 若数组下标并未到达最后一位，且存在当前中间件函数则执行当前函数并传入 dispatch(i + 1)，
      //就可看出是数组中的下一个中间件了，此时作为 next 传入了中间件函数中；            
      try {
     	 // bind 时指向 null 以防在执行过程中改变了指向
     	  //  Promise.resolve 中去实现递归 dispatch 函数
        return Promise.resolve(fn(context, dispatch.bind(null, i + 1))); 
      } catch (err) {
        return Promise.reject(err)
      }
    }
  }
}