koa-源码分析

来源:互联网 发布:联想笔记本的网络接口 编辑:程序博客网 时间:2024/05/20 01:35

转载:https://segmentfault.com/a/1190000007792512

Koa源码分析

前两篇已经了解了yield和generator的相关概念,这篇文章继续分析Koa的源码,Koa的用法,这里就不在列举了。我觉得Koa主要功能主要是下面几个方面1).    提供了中间件机制2).    封装了request/response,context对象3).    使用了yield,提供了便利的流程控制,使异步编程更优雅4).    便捷的异常处理,使用try catch就可以捕获程序中的异常,不需要考虑同步或者异步的问题

简单的例子

import koa from 'koa';  
const app = koa();
app.experimental = true;
app.proxy = true;//当 app.proxy 设置为 true 时,支持 X-Forwarded-Host
//添加一个中间件
app.use(function*(next) {

     console.log('start');     yield *next;     console.log('end'); })

app.listen(port);

koa会提供很多中间件,比如koa-router,koa-session,koa-bodyparse,koa-server等

koa源码分析之Application

var app = Application.prototype;module.exports = Application;

//函数的入口部分

function Application() {

      if (!(this instanceof Application)) return new Application;      this.env = process.env.NODE_ENV || 'development';      this.subdomainOffset = 2;      this.middleware = [];      this.proxy = false;      this.context = Object.create(context);      this.request = Object.create(request);      this.response = Object.create(response);    }

listen真正的创建了一个http server

app.listen = function(){

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

中间件use方法, fn必须是一个generator function, 可以通过'GeneratorFunction' == fn.constructor.name

app.use = function(fn){      if (!this.experimental) {        // so we have to make sure that `fn` is a generator function        assert(fn && 'GeneratorFunction' == fn.constructor.name, 'app.use() requires a generator function');      }      this.middleware.push(fn);      return this;  };

重要函数来啦,这里只是分析this.experimental = false的情况

app.callback = function(){      var fn = this.experimental        ? compose_es7(this.middleware)        : co.wrap(compose(this.middleware));                //co是使用的tj的co模块,这里先分析compose的函数用法        //compose 函数是将this.middleware数组的函数进行一个简单的处理,保证每个函数都有后一个函数的引用,具体代码看下面的分析        //co.wrap可以将一个generation function 进行转换,使fn能够自动运行              var self = this;          if (!this.listeners('error').length) this.on('error', this.onerror);          return function(req, res){        res.statusCode = 404;        var ctx = self.createContext(req, res);        onFinished(res, ctx.onerror);        //这里执行中间件函数,然后将处理的结果都挂在ctx上,最后respond函数输出到前端,注意catch函数进行错误逻辑处理        fn.call(ctx).then(function () {          respond.call(ctx);        }).catch(ctx.onerror);      }};

compose 代码分析, compse是将middleware里面的middleware都有下一个函数的引用

function compose(middleware){      return function *(next){        if (!next) next = noop();            var i = middleware.length;            while (i--) {          next = middleware[i].call(this, next);        }            return yield *next;      }    }

至此,koa的源码已经分析完毕了



原创粉丝点击