学习koa源码，弄懂中间件和路由的实现原理

// 引入koa的中间件和路由构造函数
const { compose } = require('koa-compose');
const Layer = require('./layer');
 
class Router {
  constructor() {
    this.methods = ['GET', 'POST', 'PUT', 'HEAD', 'DELETE', 'OPTIONS'];
    this.stack = []; // 存储路由层的数组
  }
 
  // 将路由层添加到栈中
  route(path) {
    const route = new Layer(path, [], this.methods);
    this.stack.push(route);
    return route.route;
  }
 
  // 将中间件应用到路由层
  register(path, methods, middleware) {
    const route = new Layer(path, methods, this.methods);
    const stack = this.stack;
    let layer;
 
    // 为每个中间件创建一个新的层并将其添加到栈中
    for (let i = 0; i < middleware.length; i++) {
      layer = new Layer(path, methods[i], this.methods);
      layer.route = middleware[i];
      stack.push(layer);
    }
  }
 
  // 处理请求，根据路径和HTTP方法匹配中间件执行
  handle(ctx, next) {
    let idx = 0;
    const stack = this.stack;
 
    const dispatch = (i) => {
      let layer = stack[i];
      let route = layer.route;
 
      if (i === stack.length) {
        return next();
      }
 
      if (layer.match(ctx.path, ctx.method)) {
        // 由layer的route执行，即中间件函数
        route.call(layer, ctx, () => {
          dispatch(i + 1);
        });
      } else {
        dispatch(i + 1);
      }
    };
 
    dispatch(idx);
  }
}
 
// 使用koa-compose来合并所有层的中间件到一个处理函数
Router.prototype.middleware = function middleware() {
  const middleware = this.stack.map((layer) => {
    return layer.middleware.bind(layer);
  });
  return compose(middleware);
};
 
module.exports = Router;

这个示例代码展示了如何实现一个简单的路由和中间件机制。它定义了一个Router类，并实现了route和register方法来添加路由层和中间件层到栈中。handle方法负责根据请求的路径和方法分发到正确的中间件执行。最后，它提供了一个middleware方法，该方法使用koa-compose将所有中间件层合并为一个处理请求的函数。