你需要掌握的 Koa 洋葱模型和中间件

大家好，我是前端西瓜哥。

Koa 是一个 nodejs 框架，经常用于写 web 后端服务。它是 Express 框架的原班人马开发的新一代 web 框架，使用了 async / await 来优雅处理无处不在的异步逻辑。

我们常说 Koa 其实是洋葱模型，今天就来深挖下 Koa 的洋葱模型到底是什么。

什么是洋葱模型

在这之前，我们先简单看看 Koa 是如何使用的。

在 Koa 中，我们通过 app.use 方法注册中间件。中间件可以注册多个，它们的执行顺序和注册时机相关，先注册的先执行。

所谓中间件就是一个函数，这个函数接受 Koa 提供的两个参数：

ctx 上下文对象；
next 函数。

ctx 上有各种参数，比如请求对象 request 和响应对象 response。

调用 next 函数会执行下一个的中间件，如果你不调用 next 函数，那下一个中间件就不会执行。

我们看一个例子：

const Koa = require('koa');const app = new Koa();// 中间件 1：记录请求花费时间
app.use(async (ctx, next) => {console.log('中间件 1');const start = new Date().getTime();await next();const t = new Date().getTime() - start;console.log('请求花费时间为', t + ' ms');
})// 中间件 2：获取数据
app.use(async (ctx, next) => {console.log('中间件 2')const data = await getData(); ctx.body = { data };
})// 模拟从数据库获取数据，耗时 1 s
const getData = async () => {await new Promise((resolve) => {setTimeout(() => {resolve();}, 1000);})return 'Hello World!';
}app.listen(3005);

当请求时，服务器的日志是这样的：

中间件
中间件 2
请求花费时间为 1005 ms

从中间件 1 睡觉来看来说，代码逻辑可以分为三部分：

先执行 next() 前面的代码；
然后执行 next() 其后的中间件 2 的所有代码；
最后执行 next() 后面的代码；

这种先执行了当前中间件的前半部分逻辑，然后处理完之后的中间件后，最后继续执行当前中间件的后半部分的特性，可以让我们可以像洋葱一样，从外到内先处理 request 对象，再从内到外处理 response 对象，于是被称为 洋葱模型。

洋葱模式本质是设计模式中的 职责链模式 的变体。

职责链模式，指的是将请求和响应解耦，让多个处理对象有机会依此去处理请求。比如处理对象 A 先处理数据，然后将处理后的数据传给处理对象 B，依此类推形成了一条链。链条上的不同处理对象负责各自的职责。

A -> B -> C

相比经典的职责链模式，洋葱模型可以将一个处理器分成两个部分，在不同时机触发但却拥有相同的上下文，在一些情况下是非常好用的，就比如刚刚提到的打印单个请求花费时长。

A1 -> B1 -> C -> B2 -> A2

Koa 中的源码实现

Koa 是一个非常轻量的库，源码分析起来相对比较容易，所以我们来看看它的洋葱模型，也就是中间件模型的实现吧。不过因为用到了大量闭包的实现，看起来还是容易眼花的。

首先通过 new Koa() 创建的一个 app 对象，它有一个成员属性 middleware，初始值为空数组。这个 middleware 就是保存中间件函数的地方。

每当我们调用 app.use(fn) 时，Koa 会将中间件函数加到 middleware。

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

最后我们调用 app.listen(port)，这个 API 会启动 http 服务器。

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

this.callback 是一个函数，会返回一个封装好的函数给 nodejs 原生的 http.createServer 使用。

callback 实现为：

callback () {const fn = this.compose(this.middleware)if (!this.listenerCount('error')) this.on('error', this.onerror)const handleRequest = (req, res) => {const ctx = this.createContext(req, res)return this.handleRequest(ctx, fn)}return handleRequest
}

再看看这个 this.compose 方法，它将多个中间件函数进行组合，让它们可以依此被调用。这个 compose 被抽成一个名为 koa-compose 的 npm 包，里面的代码很少，我将其中的核心代码抽出来：

function compose (middleware) {return function (context, next) {function dispatch (i) {let fn = middleware[i]if (i === middleware.length) fn = next// 走到最后一个中间件if (!fn) return Promise.resolve()return Promise.resolve(fn(context, dispatch.bind(null, i + 1)))}return dispatch(0)}
}

compose 函数的作用是将中间件函数进行组装，先返回第一个中间件函数的封装，其类型签名为 () => Promise<any>。

当这个被封装的函数被执行时，它会将执行原始的中间件函数，并拿到下一个中间件函数的封装，也就是 next。

我们回到 callback 方法中，其中 this.handleRequest 的实现为：

handleRequest (ctx, fnMiddleware) {const res = ctx.resres.statusCode = 404const onerror = err => ctx.onerror(err)const handleResponse = () => respond(ctx)onFinished(res, onerror)return fnMiddleware(ctx).then(handleResponse).catch(onerror)
}

this.handleRequest 将 compose 返回的第一个中间件，进行调用。

Express 是洋葱模型吗？

Express 发布时，ES6 还没出来，不能使用 Promise，更不用说 ES7 的 async / await 了。

所以 Express 在当时情况下，其实并没有能力实现这种支持异步的洋葱模型，那时候要做异步就只能使用回调的风格。

总的来说，Express 也算是一种 只支持同步的洋葱模型，因为它在实现上没有处理 next 是 async 的情况，这是历史原因导致的。

Express 是在调用 res.send 时，结束数据的处理，返回响应数据给客户的。在一个请求里不能多次调用 res.rend。

Koa 是给 ctx.response 上加内容，等到中间件走完才返回数据。

结尾

洋葱模型，就是将数据顺序传入到多个中间件中，让它们进行处理传递，并利用函数递归的特性，让我们可以在一个中间件内先执行前半部分逻辑，再执行之后的所有中间件的完整逻辑后，再掉转方向继续执行这个中间件的后半部分。

相比一旦进入下个中间件后再不回来，这种实现可以让我们的代码更灵活，在一些场景下很有用。

我是前端西瓜哥，欢迎关注我，学习更多前端知识。