前几天面试node,面试官问了koa的中间件是如何实现的,我一想,卧槽,这特么不是我很熟悉的么,然后就哇啦啦啦的一顿操作,自己说的云里雾里,说到最后感觉说的怎么和自己想的不一样呢。结果就是比较惨了。最后想了想还是写下来把,从0到1写一个完整的koa-compose。
- 首先我的思考下脑子没问题,那么自己思路也没问题,那么就是表达不对,导致面试官听了后,心里想,码单,这特么说的啥玩意。算了算了,就当听懂了吧。(其实我自己说的自己都没听)。
- 写之前扯点别的,这个koa洋葱模型用到了闭包,递归,还有Promise(来控制执行流程),偏函数等,从一个简单的来,先写个空壳子。
//v1.0,中间件的两种写法就不多说了,自己去看
[两种不同中间件](https://github.com/demopark/koa-docs-Zh-CN)
// 异步函数不多说。
let middlewares=[
async function(ctx,next){
console.log('one');
await next();
},
async function(ctx,next){
console.log('one');
await next();
}
];
function compose(middleware){
return function(ctx,next){
// 执行流程
}
}
// 返回的是一个闭包函数
let middlewareStart=compose(middlewares);
// 开始执行
middlewareStart(this);
- 重点来了,内部函数怎么来控制执行流程。这个先不看答案,自己思考下,我想想。
- 执行顺序肯定是从第一个函数开始,所以默认是0,
- 洋葱都是一层包裹一层,和我们说的串行化的概念一样。
- 通过Promise来控制执行流程。
function compose(middleware){
return function(context,next){
function dispatch(i){
// 取到第一个函数
let fn=middleware[i];
// 返回一个Promise实例,从第一个函数执行到最后一个函数
return Promise.resolve(fn(context,dispatch.bind(null,i+1)));
}
return dispatch(0);
}
}
- 写好一个1.0版本,我们执行看看,流程可以正跑起来,但是有一个error,查查原因,原来超过了数组长度,最后一个为空,我们做个处理。
function compose(middleware){
return function(context,next){
function dispatch(i){
// 取到第一个函数
let fn=middleware[i];
// 超过数组的长度返回一个空的Promise实例
if(!fn) return Promise.resolve();
// 返回一个Promise实例,从第一个函数执行到最后一个函数
return Promise.resolve(fn(context,dispatch.bind(null,i+1)));
}
return dispatch(0);
}
}
- 我们再次跑起来看看,完美的执行了整个中间件。没有问题。最后我们和原作者的做个对比。发现有个小的区别。
function compose(context, next) {
// last called middleware #
let index = -1
return dispatch(0)
function dispatch (i) {
index = i
let fn = middleware[i]
// 多了一行这个操作。个人感觉没启到作用,不是没用,可以结合整个koa来看。
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)
}
}
}
- 最后来总结下koa中间件的执行流程(洋葱模型)。
- 通过async和await,Promise来控制执行流程,其实就串行执行。
- 通过偏函数将后面一个异步函数(这里在异步函数做了一层包装)作为前一个异步函数的参数(next)。
- 最后面试官来一句嗯,没毛病,那你再说说async和await的原理以及generator函数和yield,最好说下generator函数的流程管理。还有Promise的是是如何实现的也顺带说说。我听了之后,MMP。自己埋雷,炸自己。
- 后面继续分析generator函数和promise的实现原理。