手写koa2

​一、koa2核心设计

1. 封装node http server，创造Koa类构造函数

2. 构造request、response、及context对象

3. 中间件机制实现

二、koa2核心代码实现

    1.主文件koa.js

const http = require('http');const context = require('./context');const request = require('./request');const response = require('./response');​class Koa {  constructor() {    this.middlewares = [];  }  listen(...args) {    http.createServer(async (req, res) => {      // 创建上下文对象      const ctx = this.createContext(req, res);      // 将middlewares合并成一个      const fn = this.compose(this.middlewares);      await fn(ctx);      // 给用户返回数据      res.end(ctx.body);    }).listen(...args);  }  use(mid) {    this.middlewares.push(mid);  }  createContext(req, res) {    const ctx = Object.create(context);    ctx.request = Object.create(request);    ctx.response = Object.create(response);    ctx.req = ctx.request.req = req;    ctx.res = ctx.response.res = res;    return ctx;    }  compose(middlewares) { // 中间件机制实现    return function(ctx) {      return dispatch(0); // 执行第0个      function dispatch(i) {        let fn = middlewares[i];        if (!fn) {          return Promise.resolve();        }        return Promise.resolve(          fn(ctx, function next() {            // promise完成后，再执行下一个            return dispatch(i + 1);                    });        )         }    }  }}

    2、封装request：request.js

module.exports = {  get url() {    return this.req.url;  }}

    3、封装response：response.js

module.exports = {  get body() {    return this._body;  }  set body(val) {    this._body = val;    }}

    4、封装context：context.js

module.exports = {  get url() {    return this.request.url;  }  get body() {    return this.response.body;  }  set body(val) {    this.response.body = val;  }}

以上为koa2的核心实现，仅作为学习和理解核心原理用，实际中的koa2有更多详尽的实现，具体查阅https://github.com/koajs/koa/tree/master/lib。

参考资料

1. https://study.miaov.com/study/show/chapter/639

延伸阅读

---

▶ Walkthrough007