koa中間件的實現(xiàn)原理如何?先來看一個例子。
koa的執(zhí)行順序是這樣的:
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中間件實現(xiàn)原理,也就是洋蔥模型的實現(xiàn)原理是什么?
一、問題分析
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的實現(xiàn)。
next要求調(diào)用隊列中下一個middleware,當達到最后一個的時候resolve。這樣最后面的promise先resolve,一直到第一個,這樣就是洋蔥模型的順序了。
二、實現(xiàn)
koa-compose的實現(xiàn)是這樣的:
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)
}
}
}
}
我們把一些參數(shù)檢查的非核心邏輯去掉了,實現(xiàn)代碼就上面那些。每次傳入的next都是調(diào)用下一個middleware,這樣是一個遞歸的過程,結(jié)束條件是最后一個middleware的next是用戶傳入的。
這里面有一些亮點:
1. 這是一種尾遞歸的形式,尾遞歸的特點是最后返回的值是一個遞歸的函數(shù)調(diào)用,這樣執(zhí)行完就會在調(diào)用棧中銷毀,不會占據(jù)調(diào)用棧.
2. 返回的是一個Promise.resolve包裝之后的調(diào)用,而不是同步的調(diào)用,所以這是一個異步遞歸,異步遞歸比同步遞歸的好處是可以被打斷,如果中間有一些優(yōu)先級更高的微任務(wù),那么可以先執(zhí)行別的微任務(wù)
3. compose是函數(shù)復(fù)合,把n個middleware復(fù)合成一個,參數(shù)依然是context和next,這種復(fù)合之后依然是一個middleware,還可以繼續(xù)進行復(fù)合。
三、總結(jié)
Koa 中間件的實現(xiàn)原理,也就是洋蔥模型的實現(xiàn)原理,核心在于next的實現(xiàn)。next需要依次調(diào)用下一個middleware,當?shù)阶詈笠粋€的時候結(jié)束,這樣后面middleware的promise先resolve,然后直到第一個,這樣的流程也就是洋蔥模型的流程了。
實現(xiàn)的時候還有一些細節(jié),一個是遞歸最好做成尾遞歸的形式,而是用異步遞歸而不是同步遞歸,第三就是形式上用函數(shù)復(fù)合的形式,這樣復(fù)合之后的中間件還可以繼續(xù)復(fù)合。