首页 > 其他分享 >koa中间件的实现原理

koa中间件的实现原理

时间:2023-02-21 18:24:37浏览次数:32  
标签:function console log 递归 koa middleware 中间件 next 原理

koa中间件的实现原理如何?先来看一个例子。

koa的执行顺序是这样的:

const middleware = async function (ctx, next) {
  console.log(1)
  await next()
  console.log(6)
}

const middleware2 = async function (ctx, next) {
  console.log(2)
  await next()
  console.log(5)
}

const middleware3 = async function (ctx, next) {
  console.log(3)
  await next()
  console.log(4)
} 

会依次打印1,2,3,4,5,6

 

问题是koa中间件实现原理,也就是洋葱模型的实现原理是什么?

一、问题分析

async await是promise的语法糖,await后面跟一个promise,所以上面的代码可以写成:

const middleware = function (ctx, next) {
  console.log(1)
  next().then(() => {
    console.log(6)
  })
}

const middleware2 = function (ctx, next) {
  console.log(2)
  next().then(() => {
    console.log(5)
  })
}

const middleware3 = function (ctx, next) {
  console.log(3)
  next().then(() => {
    console.log(4)
  })
}

改成这样更好理解一些,所以流程控制的核心在于next的实现。

next要求调用队列中下一个middleware,当达到最后一个的时候resolve。这样最后面的promise先resolve,一直到第一个,这样就是洋葱模型的顺序了。

二、实现

koa-compose的实现是这样的:

function compose(middleware) {
  return function (context, next) {
    let index = -1
    return dispatch(0)

    function dispatch(i) {
      index = i
      let fn = middleware[i]
      if (i === middleware.length) fn = next
      if (!fn) return Promise.resolve()
      try {
        return Promise.resolve(fn(context, dispatch.bind(null, i + 1)))
      } catch (err) {
        return Promise.reject(err)
      }
    }
  }
}

我们把一些参数检查的非核心逻辑去掉了,实现代码就上面那些。每次传入的next都是调用下一个middleware,这样是一个递归的过程,结束条件是最后一个middleware的next是用户传入的。

这里面有一些亮点:

  1. 这是一种尾递归的形式,尾递归的特点是最后返回的值是一个递归的函数调用,这样执行完就会在调用栈中销毁,不会占据调用栈.
  2. 返回的是一个Promise.resolve包装之后的调用,而不是同步的调用,所以这是一个异步递归,异步递归比同步递归的好处是可以被打断,如果中间有一些优先级更高的微任务,那么可以先执行别的微任务
  3. compose是函数复合,把n个middleware复合成一个,参数依然是context和next,这种复合之后依然是一个middleware,还可以继续进行复合。

三、总结

Koa 中间件的实现原理,也就是洋葱模型的实现原理,核心在于next的实现。next需要依次调用下一个middleware,当到最后一个的时候结束,这样后面middleware的promise先resolve,然后直到第一个,这样的流程也就是洋葱模型的流程了。

实现的时候还有一些细节,一个是递归最好做成尾递归的形式,而是用异步递归而不是同步递归,第三就是形式上用函数复合的形式,这样复合之后的中间件还可以继续复合。

标签:function,console,log,递归,koa,middleware,中间件,next,原理
From: https://www.cnblogs.com/qian-fen/p/17141968.html

相关文章

  • FileWriter和FileReader的底层原理:缓存区技术
    FileReader的底层原理FileWriter的底层原理注意事项FileReader的底层原理字节输入流FileReader用于读出文件中的数据到内存中,而且每次只能读取一个字节。这其中内存......
  • 京东开源热key探测(JD-hotkey)中间件单机qps 提升17倍实战
    京东hotkey框架(JD-hotkey)是京东app后台研发的一款高性能热数据探测中间件,用来实时探测出系统的热数据,并将热数据毫秒内推送至系统的业务集群服务器的JVM内存。以下统称为"热......
  • 云小课|MRS基础原理之Flink组件介绍
    阅识风云是华为云信息大咖,擅长将复杂信息多元化呈现,其出品的一张图(云图说)、深入浅出的博文(云小课)或短视频(云视厅)总有一款能让您快速上手华为云。更多精彩内容请单击......
  • Spring Boot自动配置原理懂后轻松写一个自己的starter
    目前很多Spring项目的开发都会直接用到SpringBoot。因为Spring原生开发需要加太多的配置,而使用SpringBoot开发很容易上手,只需遵循SpringBoot开发的约定就行了,也就是约定......
  • [bzoj 1471] 不相交路径 (容斥原理)
    题目描述给出一个结点的有向无环简单图。给出个不同的点,,,,定义不相交路径为两条路径,两条路径的起点分别为和,对应的两条路径的终点为和,要求满足这两条路径不相交,即两条路径上没......
  • [Sdoi2013] [bzoj 3198] spring (hash+容斥原理)
    题目描述给出个维坐标,求有多少对点对满足恰好个位置相等坐标数值在以内题目分析这道题一看就是hash容斥原理,用个位置对应相等个位置对应相等个位置对应相等的…但是不能......
  • [bzoj 2393] Cirno的完美算数教室 (容斥原理+dfs剪枝)
    题目描述发现了一种数,这种数呢只含有和两种数字现在想知道中有多少个数能被数整除  1<L<R<10^{10}题目分析由于R<10^{10},最大只有10位的数可以对答案造成贡献,每一位只能......
  • 多线程并发(二):聊聊AQS中的共享锁实现原理
    在上一篇文章多线程并发(一)中我们通过acquire()详细地分析了AQS中的独占锁的获取流程,提到独占锁,自然少不了共享锁,所以我们这边文章就以AQS中的acquireShared()方法为例,......
  • Nacos服务发现原理分析
    微服务将自己的实例注册到nacos注册中心,nacos服务端存储了注册列表,然后通过ribbon调用服务,具体是如何调用?如果nacos服务挂了,还能正常调用服务吗?调用的服务列表发生变化,调......
  • Zookeeper-ZKFC的原理和功能
    一、前言HADOOP2HA架构引入了ZKFC、Journalnode组件,本篇文章主要介绍ZKFC的功能和原理。HA架构支持两种切换方式:手动切换: 通过命令实现主备之间的切换,可以用HDFS升......