乐趣区

关于node.js:Koa-洋葱模型

jiezi

微信公众号：[前端一锅煮]
一点技术、一点思考。

前言：Koa 是一个新的 web 框架，由 Express 幕后的原班人马打造，致力于成为 web 利用和 API 开发畛域中的一个更小、更富裕表现力、更强壮的基石。通过利用 async 函数，Koa 帮你抛弃回调函数，并无力地加强错误处理。Koa 并没有捆绑任何中间件，而是提供了一套优雅的办法，帮忙您疾速而欢快地编写服务端应用程序。

什么是洋葱模型？

咱们先来看一个 demo：

const Koa = require('koa')
const app = new Koa() // 应用程序

// 中间件 1
app.use((ctx, next) => {console.log(1)
  next()
  console.log(2)
})
// 中间件 2
app.use((ctx, next) => {console.log(3)
  next()
  console.log(4)
})

app.listen(9000, '0.0.0.0', () => {console.log(`Server is starting`)
})

浏览器输出 localhost:9000，控制台会有如下打印：

1
3
4
2

很显著，在 koa 的中间件中，通过 next 函数，将中间件分成了两局部，next 下面的一部分会首先执行，而上面的一部分则会在所有后续的中间件调用之后执行。

Koa 中间件执行程序：

  • 外层中间件进行后期解决（next 前的逻辑）；
  • 调用 next，将控制流交给下个中间件，并 await 其实现，直到前面没有中间件或者没有 next 函数执行为止；
  • 实现后，一层层回溯执行各个中间件的前期解决（next 后的逻辑）。

自定义实现

从实质上来说，洋葱模型实际上就是一个实现以下成果的函数调用办法。

async function f(){console.log(1);
  await f2()
  console.log(2);
}
async function f2() {console.log(3);
  await f3()
  console.log(4);
}
async function f3() {console.log(5);
  console.log(6);
}
f()

输入：

1
3
5
6
4
2

这里你会发现实际效果是实现相似这样的构造。

function f(){console.log(1);
  new Promise((resolve) => {console.log(3);
      new Promise((resolve) => {console.log(5);
          console.log(6);
      })
      console.log(4);
  })
  console.log(2);
}
f()

从第一个父函数开始，往里面塞子函数，子函数塞孙函数。这个时候调用父函数，就会顺次往里走，先执行同步代码，遇到异步代码放到队列中，同步代码执行结束再顺次取出队列中的代码执行，实质上是利用了 js 的事件循环机制。

上面咱们来看下简版的实现：

const middleware = []
let f1 = async function (next) {console.log(1)
  await next()
  console.log(2)
}
let f2 = async function (next) {console.log(3)
  await next()
  console.log(4)
}
let f3 = async function (next) {console.log(5)
  console.log(6)
}

function use(fn) {middleware.push(fn)
}
use(f1)
use(f2)
use(f3)

// 外围代码
function compose() {return dispatch(0)
  function dispatch(i) {let fn = middleware[i]
    if(!fn) return
    return fn(dispatch.bind(null, i + 1))
  }
}
compose()

从第一个函数开始，顺次把下一个函数当做参数塞进去。

koa-compose 源码

以下是 Koa 洋葱模型的源码：

function compose(middleware) {if (!Array.isArray(middleware))
    throw new TypeError('Middleware stack must be an array!')
  for (const fn of middleware) {if (typeof fn !== 'function')
      throw new TypeError('Middleware must be composed of functions!')
  }

  //  传入对象 context 返回 Promise
  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)
      }
    }
  }
}

代码最外围的两点：

  1. 将 context 一路传下去给中间件
  2. 将 middleware 中的下一个中间件 fn 作为将来 next 的返回值

另一种实现办法 

function compose(middlewares) {return async (ctx) => {function createNext(middleware, oldNext) {return async () => {await middleware(ctx, oldNext)
      }
    }
    let len = middlewares.length
    let next = async () => {return Promise.resolve()
    }
    for (let i = len - 1; i >= 0; i--) {let currentMiddleware = middlewares[i]
      next = createNext(currentMiddleware, next)
    }
    await next()}
}

将中间件从最初一个开始封装，每一次都是将本人的执行函数封装成 next 当做上一个中间件的 next 参数，这样当循环到第一个中间件的时候，只须要执行一次 next()，就能链式的递归调用所有中间件。

退出移动版