Koa compose源码解析

compose是koa的核心组件，负责中间件注册后的调用，可以实现多函数的可控链式调用。compose的使用如下：

function A(context, next) {
  console.log('A');
  next();
}

function B(context, next) {
  console.log('B');
}

function C(context, next) {
  console.log('C');
  next();
}

const handler = compose([A, B, C]);
handler(null);
console:
  A
  B

上述示例中C不会触发因为函数B中未调用next()。

看了以上的代码，很多人使用koa框架时，对中间件函数通过next()触发下一个中间件的原理不甚了解。接下来会逐步还原compose的实现，同时对源码实现进行解读。

function compose(funcs) {
    return function(context) {
        return dispatch(0)；
        function dispatch(i) {
            return funcs[i](context, dispatch.bind(null, i+1))
        }
    }
}

以上简单实现了compose函数，当然compose函数官方实现比这个复杂，但这段代码可以看出compose函数的核心原理，compose函数的核心是建立dispatch函数和funcs函数数组的联系。

dispatch调用流程

compose函数运行流程如以上所示，我们可以发现执行compose(funcs)(ctx)等价于执行disptch(0)，执行dispatch(i)等价于执行funcs[i](ctx, dispatch(null, i+1))。 意不意外，传给中间件函数的next参数就是dispatch.bind(null, i+1)，执行这个next函数就会触发下一个中间件函数。

现在考虑compose函数的边际条件，来完成官方实现。

1. 首先校验compose函数的输入，其输入必须是一个函数列表；
2. 限制每个中间件函数中next()的调用次数，最多只能调用一次；
3. 对dispatch的参数进行校验，i不能大于函数列表的长度，即i < funcs.length；
   接下来实现上述边际条件：

function compose(funcs) {
    if (!Array.isArray(funcs)) {
      throw new Error('param funcs must be an array');
    }
    for (let i = 0; i < funcs.length; i++) {
      if (typeof funcs[i] !== 'function') {
        throw new Error('param funcs must be an array of functions')
      }
    }
    return function(context) {
        let index = -1;
        if (index === i) {
           throw new Error('next function triggered more than once');
        }
        return dispatch(0);
        function dispatch(i) {
            index++;
            return i < funcs.length ? funcs[i](context, dispatch.bind(null, i+1)) : null;
        }
    }
}