深入理解 Koa 框架中间件原理

来源：动视网责编：小采时间：2020-11-27 22:06:07

深入理解 Koa 框架中间件原理

深入理解 Koa 框架中间件原理:Node 主要用在开发 Web 应用，koa 是目前 node 里最流行的 web 框架。在 Node 开启一个 http 服务简直易如反掌,官网 demo。 const http = require(http); const server = http.createServer((req, res) =>

推荐度：

点击下载本文 文档为doc格式

导读深入理解 Koa 框架中间件原理:Node 主要用在开发 Web 应用，koa 是目前 node 里最流行的 web 框架。在 Node 开启一个 http 服务简直易如反掌,官网 demo。 const http = require(http); const server = http.createServer((req, res) =>

理解 Koa 的中间件机制（源码分析）

阅读源码，化繁为简,我们看看 koa 的中间件系统是如何实现的。

class Application extends Emitter {
 constructor() {
 super();
 this.middleware = [];
 },

 use(fn) {
 this.middleware.push(fn);
 return this;
 },

 callback() {
 const fn = compose(this.middleware);

 return function(req, res) {
 return fn(ctx);
 };
 },

 listen(...args) {
 const server = http.createServer(this.callback());
 return server.listen(...args);
 }
}

好了，精简结束，一不小心，去枝末节，最后只剩下不到 20 行代码。

这就是框架的核心，简化后的代码非常清晰，有点不可思议,但核心就是这么简单。

我们先分析以上代码做了什么事。

我们定义了一个 middleware 数组来存储中间件。

我们定一个了一个 use 方法来注册一个中间件。其实就是简单的 push 到自身的 mideware 这个数组中。

我们还使用了一个 compose 方法，传入 middleware ，应该是做了一些处理，返回了一个可执行的方法。

你一定对中间的 compose 方法很好奇，初此之外的代码都容易理解，唯独这个 compose 不太知道究竟做了什么。

其实， compose 就是整个中间件框架的核心。

compose 之外，代码已经很清楚的定义了

中间件的存储

中间件的注册

而 compose 方法做了最为重要的一件事

中间件的执行

核心源码 compose

先上码

function compose(middleware) {
 return function(context, next) {
 // last called middleware #
 let index = -1;
 return dispatch(0);
 function dispatch(i) {
 if (i <= index)
 return Promise.reject(new Error("next() called multiple times"));
 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()传递实现中间件调用, 结合 Promise 采用递归调用的通知机制。

看图