半小时掌握koa(koa源码分析)更新啦~更新啦
发布于 2 个月前 作者 Acceptedlc 993 次浏览 最后一次编辑是 1 个月前 来自 分享

半小时掌握koa

koa分析与实现

1.从一个例子开始

'use strict';

let Application = require('koa');

let app = new Application();

app.use(function*(next) {
  console.log('东方红');
  yield next;
  console.log('太阳升');
  console.log();
});

app.use(function*(next) {
  console.log('-1979年,');
  yield next;
  console.log('-那是一个春天');
});

app.use(function*() {
  let message = '--我爱北京,天安门';
  console.log(message);
  this.body = 'hello world';
});

const PORT = 8888;
app.listen(PORT);

console.log('start at ' + PORT);

上面代码,先后添加了3个middleware。请求的结果如下:

东方红  
-1979年,  
--我爱北京,天安门  
-那是一个春天  
太阳升  

2.koa之流程

GitHub

  • 每一个请求到达服务器后,初始化请求上下文对象
  • 将上下文对象按照顺序,在每个中间件中运行一遍
    • 执行中间件1,next前的逻辑
    • next
      • 执行中间件2的逻辑
    • 执行中间件1,next后的逻辑
  • 将body返回给客户端

上面的过程就是创建http服务器的时候,callback参数应该做的事情。

3.app对象分析

  • 属性:由于这里只是简单的山寨一下koa,所以只需要一个middleware数组,用来存储中间件。如果感兴趣,想完善的话可以添加其他的属性,实现相关功能。
  • 公共方法:从开始的例子可以轻松看出有如下2个方法。
    • use
    • listen
  • 私有方法:
    • callback:创建http服务器的时候需要提供的回调函数
    • response:相应函数

4.callback实现

先从callback下手。因为,解决了这个函数,其他函数的实现就是轻松加愉快。
不过,开始我们就会发现一个问题。回调函数是node底层调用的,我们没有办法把middleware参数传递进。
这个时候,闭包玩的比较熟练的同学也许就会心中窃喜了。
没错解决这个问题的常用手段就是使用高阶函数,利用js的闭包这一特性。

function callback(middleware) {
	return function (req, res) {
	}
}

解决了上边的问题,我们顺着流程写就好了。

首先构造上下文对象

function callback(middleware) {
	return function (req, res) {
		let ctx = {
	      req: req,
	      res: res
	    };
	}
}

之后需要做的事情就是,顺序调用每一个middleware。但是我们手中有的数据结构只是一个generator function array,显然完成不了任务。因此,考虑将它map一下,让它变成一个类似function链表一样的东西。

代码如下:

let next = function*(){};

var i = middleware.length;
while (i--) {
	next = middleware[i].call(ctx, next);
}

ps: 这里有一点不知道大家发现了没有!!在每一个请求到来的时候,都需要进行一次这种map。也就是说,挂载的中间件越多,性能就会越差。我觉得这也是,koa1在性能上比express差的一个原因吧。 性能测试

接着,调用一下链表头的function,就可以实现middleware的调用。

function callback(middleware) {
  return function (req, res) {
    let ctx = {
      req: req,
      res: res
    };
    co(function *() {
      let next = function*(){};

      var i = middleware.length;
      while (i--) {
        next = middleware[i].call(ctx, next);
      }
      return yield next;
    })
  }
}

最后,就是将上下文的body返回给前端。

function callback(middleware) {
  return function (req, res) {
    let ctx = {
      req: req,
      res: res
    };
    co(function *() {
      let next = function*(){};

      var i = middleware.length;
      while (i--) {
        next = middleware[i].call(ctx, next);
      }
      return yield next;
    }).then(() => response.call(ctx))
      .catch(e => console.log(e.stack));
  }
}

以上就是整个callback的代码,不过错误处理比较粗糙。大家可以自己完善一下^-^。

5.其他function

其他function就比较简单了。分别写在下面

app.listen = function () {
  var server = http.createServer(callback(this.middleware));
  return server.listen.apply(server, arguments);
};
app.use = function (fn) {
  this.middleware.push(fn);
};
function response() {
  this.res.writeHead(200, {
    'Content-Type': 'text/plain'
  });
  this.res.end(this.body);
}

最后整个myKoa的代码

'use strict';

let co = require('co');
let http = require('http');

function Application() {
  this.middleware = [];
}

var app = Application.prototype;


app.listen = function () {
  var server = http.createServer(callback(this.middleware));
  return server.listen.apply(server, arguments);
};

app.use = function (fn) {
  this.middleware.push(fn);
};


function callback(middleware) {
  return function (req, res) {
    let ctx = {
      req: req,
      res: res
    };
    co(function *() {
      let next = function*(){};

      var i = middleware.length;
      while (i--) {
        next = middleware[i].call(ctx, next);
      }
      return yield next;
    }).then(() => response.call(ctx))
      .catch(e => console.log(e.stack));
  }
}

//------private function
function response() {
  this.res.writeHead(200, {
    'Content-Type': 'text/plain'
  });
  this.res.end(this.body);
}

module.exports = Application;
   

koa与他的小伙伴

koa-router

1.内存模型

GitHub 上图是一个简化的koa-router的内存快照。
1个router最主要的属性就是stack,他是一个layer的数组。每当我们调用一次router.verb或者router.use就会有一个layer被创造出来,放到这个数组中。
如上图所示,每个layer主要有4个属性。
请大家,务必记住这个内存模型,后文的讨论都会围绕这张图进行。

2.请求处理流程

在上一部分曾经说过,koa的每一个中间件都是一个generator function,koa-router也不能免俗。别看他代码比koa要多,但是他就是一个函数而已。
接下来我们就看一下当请求到达服务器后这个函数都干了些什么。

  • 首先便利router的stack数组,通过每一个layer的正则版本的路径检查当前请求的path是否符合这则表达式。
  • 通过了第一步的检查的layer,进入了第二阶段的筛选。这个阶段的标准是layer的methods属性,methods为空数组或者methods数组中有当前请求的verb便通过筛选。
  • 通过筛选的layer会被放入一个叫做pathAndMethod的数组中
  • 我们下载可以看下下我们又有的数据结构–>一个fn*的二维数组。是否有种似曾相识的感觉囊。那么,下一步就是讲数组变成fn链表,然后调用一下开头的fn。

以上就是koa-router的整个流程。

3.verb
相信大家对于这个方法的用法应该很熟悉了吧,我就在这里不多说了。
这里,我想强调的是。每次使用这个方法的时候就会增加一个layer。也就是说筛选合适的layer的操作消耗的时间就会变多。在设计跟优化程序的时候应该注意这个特征。
另外,推荐一种性能更好的使用方法

//method1
router.get('/blabla',fn1,fn2,...)

//method2
router.use('/blabla',fn1)
router.get('/blabla',fn2)

method1和2可以实现相同的功能,但是method1只会创建一个layer,这个layer会有两个middleware。不难发现method1的效率会更高一些。
好了verb就说到这了吧。

3.use
关于这个方法,我想说的是,use不但可以use一个自己写的中间件,而且还可以use一个router。这种方法可以让我们实现用文件夹定义url的效果,我个人觉得这是一种有没得方法。

var forums = new Router();
var posts = new Router();

posts.get('/', function *(next) {...});
posts.get('/:pid', function *(next) {...});
forums.use('/forums/:fid/posts', posts.routes(), posts.allowedMethods());

// responds to "/forums/123/posts" and "/forums/123/posts/123"
app.use(forums.routes());

我想,也许有人会好奇这个功能是怎么实现的吧,当时看到api文档的时候我也很好奇。下面我就来说一下吧。
直接上源代码:

middleware = middleware.filter(function (fn) {
    if (fn.router) {
      fn.router.stack.forEach(function (layer) {
        if (path) layer.setPrefix(path);
        if (router.opts.prefix) layer.setPrefix(router.opts.prefix);
        router.stack.push(layer);
      });

      if (router.params) {
        Object.keys(router.params).forEach(function (key) {
          fn.router.param(key, router.params[key]);
        });
      }

      return false;
    }

    return true;
  });

从上面可以很轻易的看出,作者是先检测了一下这个fn,发现如果是router,那么读取这个router的所有layer,把每一个layer设置一下前缀,然后直接放到父router的stack中。机智吧^-^.

3.param
最后要说的是param这个方法。

router
  .param('user', function *(id, next) {
    this.user = users[id];
    if (!this.user) return this.status = 404;
    yield next;
  })
  .get('/users/:user', function *(next) {
    this.body = this.user;
  })
  .get('/users/:user/friends', function *(next) {
    this.body = yield this.user.getFriends();
  })
  // /users/3 => {"id": 3, "name": "Alex"}
  // /users/3/friends => [{"id": 4, "name": "TJ"}]

当时,看到这段代码的时候,确实激动了一把。如作者说的,这个功能可以很方便的实现auto-loading作者是validation。
但是,但是,但是!!!!!!!同志们一定要注意,这里面有一个小小的坑。 闲言碎语不要说,我们直接上代码

var app = require('koa')();
var router = require('koa-router')();
       
router
  .get('/test/:id', function *() {
    console.log('get', this.test);
    this.body = 'hello world';
  })
  .use('/test/:id', function*(next) {
    console.log('use', this.test)
    yield next;
  })
  .param('id', function * (id, next) {
    this.test = this.test || 0;
    this.test = this.test + id;
    yield next;
  });

app.use(router.routes())
  .use(router.allowedMethods());

app.listen(8888);

/**
输出结果
use 01
get 011
**/

相信眼尖的通知们已经知道发生了什么!没错,param对应的fn执行了两遍,这是因为param会把中间件放到layer的middleware数组中。
好了,虽然有点小小的瑕疵,但是,用的时候注意就好了。这个功能还是很好用的,吼吼吼。

以上就是koa-router我想说的全部内容,至于怎么用的话,看一下api文档就好了,我只能说作者设计的很简洁,一切说明都是废话。

希望我也能早日设计出这么有没得api。

koa-body

这个组件使用起来也是非常方便的,在这里说一下的目的是,由于前一阵子公司使用的body parse用起来难用的不要不要的。所以在这里推荐一下吧。

下面的内容基本就是readme的翻译,没兴趣的可以直接不看。

1.简介
koa-body支持multipart, urlencoded 和 json 请求体,提供跟express的multer一样的功能。他是对co-body和formidable的封装。

2.使用方法
koa-body可以向multer一样使用,非常简单,因为你可以从ctx.request.body 或者 ctx.req.body 中获得fields和files

var app      = require('koa')(),
    koaBody   = require('koa-body');

app.use(koaBody({formidable:{uploadDir: __dirname}}));
app.use(function *(next) {
  if (this.request.method == 'POST') {
    console.log(this.request.body);
    // => POST body
    this.body = JSON.stringify(this.request.body);
  }
  yield next;
});
app.listen(3131)

koa-body还可以跟koa-router一起使用

var app     = require('koa')(),
    router  = require('koa-router')(),
    koaBody = require('koa-body')();

router.post('/users', koaBody,
  function *(next) {
    console.log(this.request.body);
    // => POST body
    this.body = JSON.stringify(this.request.body);
  }
);

app.use(router.routes());

3.参数

  • patchNode bool 把request的body给node的ctx.req,默认为false。
  • patchKoa bool 把request的body给koa的ctx.req,默认为false。
  • jsonLimit String|Integer json body的最大byte数,默认为1mb
  • formLimit String|Integer form body的最大byte数,默认为56kb
  • textLimit String|Integer text body的最大byte数,默认为56kb
  • encoding String 设置field的编码,默认为utf-8
  • multipart Boolean 是否解析multipart body,默认为false
  • formidable Object 床底给formidable的设置项
  • strict bool 如果激活,koa-body不会解析GET,HEAD,DELETE请求,默认为true
13 回复

不错不错 mark

可以改下标题 不需要半天, 半小时吧

2333

话说既然都用es6了,那ctx那部分就可以直接用{res,req}了吧。。。。

来自酷炫的 CNodeMD

@wssgcg1213 还没完成囊 ^-^.不过完成了,估计也用不了半天,半小时肯定搞不定。求合适标题

@codehz node 4 支持吗?我司node版本还没升级囊

@Acceptedlc …那你觉得你koa是怎么用的呢—生成器函数都能用 - 现在在用5.x,这个特性一直在用,未测试4

@codehz 肯定不会用的,要不然让我们这些node4.x的怎么活

@Acceptedlc LTS4支持generator, 赶紧升吧 稳定的

看见标题就已经被吸引了,

回到顶部