一步步去阅读koa源码,总体架构分析

阅读好的框架的源码有不少好处,从大神的视角去理解整个框架的设计思想。大到架构设计,小到可取的命名风格,还有设计模式、实现某类功能使用到的数据结构和算法等等。

使用koa

其实某个框架阅读源码的时候,首先咱们要会去用这个框架,由于用了咱们才知道,某个API是怎么用,哪里有坑,哪里设计的精妙。node

下面咱们就简单用一下koa这个框架,以下代码git

const Koa = require('koa')
const app = new Koa()
app.use(async (ctx, next) => {
  ctx.body = 'Hello World'
})
app.listen(4002)

运行结果github

瓦特??这个服务会涉及到从请求到响应返回数据,就这几行代码?? 是的,你没有看错,就是单单这几行代码就能够搭建了一个服务器。web

下面咱们看看一探究竟。面试

阅读源码

去到node_modules文件夹下找到koa模块,先喵几眼README.md文件,里面介绍了koa的一些安装、用法、插件等等,这里咱们跳过,而后转到package.json以下图算法

看到package.json里面的"main": "lib/application.js"没错,这就是咱们的入口,在lib文件夹下面,咱们看到里面有application.jscontext.jsrequrest.jsresponse.js。下面通过我修改简化去掉注释application.js就只有68行代码。阅读起来能够说是很是简单了。以下图:json

第一步是咱们引入各类主要依赖设计模式

// 引入有不少 我只挑我阅读主要框架的代码模块

const response = require('./response'); // 处理response对象
const compose = require('koa-compose'); // 合并处理中间件函数
const context = require('./context'); // 整合处理context对象
const request = require('./request'); // 整合处理request对象
const http = require('http'); // node的 http原生模块

以上就是咱们的主要依赖数组

Application的对象中,有constructor函数,这个主要是初始化Application对象,生成context对象、request对象、response对象服务器

module.exports = class Application extends Emitter {
  // 初始化 Application
  constructor() {
    super(); // 继承Emitter
    this.middleware = []; // 初始化middleware为空数组
    this.context = Object.create(context); // 生成context对象
    this.request = Object.create(request); // 生成request对象
    this.response = Object.create(response); // 生成response对象
  }
}

阅读源码,咱们先不要去扣细节,好比说Object.create(context)生产的对象是什么?this.request对象下面又有什么东西???,咱们如今主要知道的是、this.context是能获取或者设置请求和响应的信息的一个对象,。this.request是请求的对象、里面能够设置或者获取请求信息的一个对象、this.response是响应请求对象、里面能够设置或者获取响应参数和值的一个对象。大概先了解就能够了。继续往下看。

在上面运用的时候,用到了app.use(fn)app.listen(4002) 咱们看看,源码里面试这样子的

module.exports = class Application extends Emitter {
  // 初始化 Application
  constructor() {
    ...
  }
  listen(...args) {
    const server = http.createServer(this.callback());
    return server.listen(...args);
  }
  use(fn) {
    this.middleware.push(fn);
    return this;
  }

上面的代码很简单 use函数就是把传入的fn 推入到this.middleware的数组中,而后返回this,方便链式调用。

而后在listen里面用node原生的http模块建立一个server,在这里顺便说一下,原生 http建立一个服务是这样子滴

const http = http.createServer((req, res) => {
  res.writeHead(200, { 'Content-Type': 'text/plain' });
  res.end('okay');
});
http.listen(8000)

继续看代码 ,在建立服务的时候,参数里面调用了一个this.callback()函数,下面咱们看看这个函数到底是怎么样子的。

module.exports = class Application extends Emitter {
  // 初始化 Application
  constructor() {
    ...
  }
  listen(...args) {
    ...
  }
  use(fn) {
    ...
  }

  callback() {
    const fn = compose(this.middleware); // 集中处理中间件数组
    const handleRequest = (req, res) => {
      const ctx = this.createContext(req, res); // 整合req、res、context、request、response
      return this.handleRequest(ctx, fn); // 返回handleRequest
    };
    return handleRequest;
  }
    
  handleRequest(ctx, fnMiddleware) {
    const handleResponse = () => respond(ctx); // 最终响应函数
    return fnMiddleware(ctx).then(handleResponse) // 处理完中间件,而后传到下一响应函数
  }
  // 建立整合新的 context.
  createContext(req, res) {
    const context = Object.create(this.context);
    const request = context.request = Object.create(this.request);
    const response = context.response = Object.create(this.response);
    context.app = request.app = response.app = this;
    context.req = request.req = response.req = req;
    context.res = request.res = response.res = res;
    request.ctx = response.ctx = context;
    request.response = response;
    response.request = request;
    context.originalUrl = request.originalUrl = req.url;
    context.cookies = new Cookies(req, res, {
      keys: this.keys,
      secure: request.secure
    });
    request.ip = request.ips[0] || req.socket.remoteAddress || '';
    context.accept = request.accept = accepts(req);
    context.state = {};
    return context;
  }
};

上面咱们能够看出在callback函数里面有一个const fn = compose(this.middleware); 这个函数就是把this.middleware数组传进去,而后集中处理中间件,而后会返回处理完中间件的fn。

继续下一行

const handleRequest = (req, res) => {
  const ctx = this.createContext(req, res);
  return this.handleRequest(ctx, fn);
};

继续进入到handleRequest函数里面的const ctx = this.createContext(req, res);这个把原生的http的请求对象req响应对象res做为参数传进去,而后在createContext函数(看上面最大那坨代码)在里面,把this.requestthis.responsethis.context请求对象req响应对象res都整,作各类整合、处理获得新的context对象返回出去。

也就是强大的ctx,获得ctx以后,下一行返回return this.handleRequest(ctx, fn);

this.handleRequest(ctx, fn)代码以下

handleRequest(ctx, fnMiddleware) {
    const handleResponse = () => respond(ctx);
    return fnMiddleware(ctx).then(handleResponse).catch(onerror);
}

这个函数 就是处理完中间件处理以后的返回的函数把ctx传下去,最后流通到respond(ctx);这个函数,

那么咱们看看这个函数被我简化后是怎么样子的,以下

// 一些容错判断或者提示我所有删了
function respond(ctx) {
  const res = ctx.res;
  let body = ctx.body;
  res.end(body);
}

经过ctx拿到响应对象,和响应值、经过end方法会通知服务器,全部响应头和响应主体都已被发送,即服务器将其视为已完成。看上面原生的http的服务方法。

最后附上一个流程图

这个只是介绍application整个流程,还有不少细节都没有一一介绍到,好比、建立contextrequestresponse对象是怎么样子的呀?中间件是如何集中层层深刻处理而后返回的呀?等等这些细节都会在下一篇会讲到(最近公司业务很是忙,不知道到猴年马月)。

写的很差的地方,让你们贱笑了。

而后最后安利一波博客,喜欢的小哥哥小姐姐能够star 哟

websit: https://github.com/naihe138/naice-blog

相关文章
相关标签/搜索