半小時(shí)掌握koa

• koa分析與實(shí)現(xiàn)
  • 從一個(gè)例子開始
  • koa之流程
  • app對象分析
  • callback實(shí)現(xiàn)
  • 其他function
• koa與他的小伙伴
  • koa-router
    • 內(nèi)存模型
    • 請求處理流程
    • router.verb
    • router.use
    • router.param
  • koa-body

<h3 id="koa分析與實(shí)現(xiàn)">koa分析與實(shí)現(xiàn)</h3>

<h4 id="example">1.從一個(gè)例子開始</h4>

'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('-那是一個(gè)春天');
});

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

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

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

上面代碼，先后添加了3個(gè)middleware。請求的結(jié)果如下：
<pre>
東方紅
-1979年,
--我愛北京,天安門
-那是一個(gè)春天
太陽升
</pre>

<h4 id="abstract">2.koa之流程</h4>

GitHub

• 每一個(gè)請求到達(dá)服務(wù)器后，初始化請求上下文對象
• 將上下文對象按照順序，在每個(gè)中間件中運(yùn)行一遍
  • 執(zhí)行中間件1，next前的邏輯
  • next
    • 執(zhí)行中間件2的邏輯
  • 執(zhí)行中間件1，next后的邏輯
• 將body返回給客戶端

上面的過程就是創(chuàng)建http服務(wù)器的時(shí)候，callback參數(shù)應(yīng)該做的事情。

<h4 id="appObjectAnalyse">3.app對象分析</h4>

• 屬性：由于這里只是簡單的山寨一下koa，所以只需要一個(gè)middleware數(shù)組，用來存儲(chǔ)中間件。如果感興趣，想完善的話可以添加其他的屬性，實(shí)現(xiàn)相關(guān)功能。
• 公共方法：從開始的例子可以輕松看出有如下2個(gè)方法。
  • use
  • listen
• 私有方法：
  • callback：創(chuàng)建http服務(wù)器的時(shí)候需要提供的回調(diào)函數(shù)
  • response：相應(yīng)函數(shù)

<h4 id="callback">4.callback實(shí)現(xiàn)</h4>

先從callback下手。因?yàn)椋鉀Q了這個(gè)函數(shù)，其他函數(shù)的實(shí)現(xiàn)就是輕松加愉快。
不過，開始我們就會(huì)發(fā)現(xiàn)一個(gè)問題。回調(diào)函數(shù)是node底層調(diào)用的，我們沒有辦法把middleware參數(shù)傳遞進(jìn)。
這個(gè)時(shí)候，閉包玩的比較熟練的同學(xué)也許就會(huì)心中竊喜了。
沒錯(cuò)解決這個(gè)問題的常用手段就是使用高階函數(shù)，利用js的閉包這一特性。

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

解決了上邊的問題，我們順著流程寫就好了。

首先構(gòu)造上下文對象

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

之后需要做的事情就是，順序調(diào)用每一個(gè)middleware。但是我們手中有的數(shù)據(jù)結(jié)構(gòu)只是一個(gè)generator function array，顯然完成不了任務(wù)。因此，考慮將它map一下，讓它變成一個(gè)類似function鏈表一樣的東西。

代碼如下：

let next = function*(){};

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

ps: 這里有一點(diǎn)不知道大家發(fā)現(xiàn)了沒有?。≡诿恳粋€(gè)請求到來的時(shí)候，都需要進(jìn)行一次這種map。也就是說，掛載的中間件越多，性能就會(huì)越差。我覺得這也是，koa1在性能上比express差的一個(gè)原因吧。 性能測試

接著，調(diào)用一下鏈表頭的function，就可以實(shí)現(xiàn)middleware的調(diào)用。

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));
  }
}

以上就是整個(gè)callback的代碼，不過錯(cuò)誤處理比較粗糙。大家可以自己完善一下^-。

<h4 id="otherFn">5.其他function</h4>

其他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);
}

最后整個(gè)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;
   

<h3 id="koaCompoment">koa與他的小伙伴</h3>

<h4 id="koa-router">koa-router</h4>

<h5 id="router-memory">1.內(nèi)存模型</h5>

GitHub

<h5 id="router-flow">2.請求處理流程</h5>

在上一部分曾經(jīng)說過，koa的每一個(gè)中間件都是一個(gè)generator function，koa-router也不能免俗。別看他代碼比koa要多，但是他就是一個(gè)函數(shù)而已。
接下來我們就看一下當(dāng)請求到達(dá)服務(wù)器后這個(gè)函數(shù)都干了些什么。

• 首先便利router的stack數(shù)組，通過每一個(gè)layer的正則版本的路徑檢查當(dāng)前請求的path是否符合這則表達(dá)式。
• 通過了第一步的檢查的layer，進(jìn)入了第二階段的篩選。這個(gè)階段的標(biāo)準(zhǔn)是layer的methods屬性，methods為空數(shù)組或者methods數(shù)組中有當(dāng)前請求的verb便通過篩選。
• 通過篩選的layer會(huì)被放入一個(gè)叫做pathAndMethod的數(shù)組中
• 我們下載可以看下下我們又有的數(shù)據(jù)結(jié)構(gòu)-->一個(gè)fn*的二維數(shù)組。是否有種似曾相識的感覺囊。那么，下一步就是講數(shù)組變成fn鏈表，然后調(diào)用一下開頭的fn。

以上就是koa-router的整個(gè)流程。

<h5 id="router.verb">3.verb</h5>
相信大家對于這個(gè)方法的用法應(yīng)該很熟悉了吧，我就在這里不多說了。
這里，我想強(qiáng)調(diào)的是。每次使用這個(gè)方法的時(shí)候就會(huì)增加一個(gè)layer。也就是說篩選合適的layer的操作消耗的時(shí)間就會(huì)變多。在設(shè)計(jì)跟優(yōu)化程序的時(shí)候應(yīng)該注意這個(gè)特征。
另外，推薦一種性能更好的使用方法

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

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

method1和2可以實(shí)現(xiàn)相同的功能，但是method1只會(huì)創(chuàng)建一個(gè)layer，這個(gè)layer會(huì)有兩個(gè)middleware。不難發(fā)現(xiàn)method1的效率會(huì)更高一些。
好了verb就說到這了吧。

<h5 id="router.use">3.use</h5>
關(guān)于這個(gè)方法，我想說的是，use不但可以use一個(gè)自己寫的中間件，而且還可以use一個(gè)router。這種方法可以讓我們實(shí)現(xiàn)用文件夾定義url的效果，我個(gè)人覺得這是一種有沒得方法。

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());

我想，也許有人會(huì)好奇這個(gè)功能是怎么實(shí)現(xiàn)的吧，當(dāng)時(shí)看到api文檔的時(shí)候我也很好奇。下面我就來說一下吧。
直接上源代碼：

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;
  });

從上面可以很輕易的看出，作者是先檢測了一下這個(gè)fn，發(fā)現(xiàn)如果是router，那么讀取這個(gè)router的所有l(wèi)ayer，把每一個(gè)layer設(shè)置一下前綴，然后直接放到父router的stack中。機(jī)智吧^-.

<h5 id="router.param">3.param </h5>
最后要說的是param這個(gè)方法。

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"}]

當(dāng)時(shí)，看到這段代碼的時(shí)候，確實(shí)激動(dòng)了一把。如作者說的，這個(gè)功能可以很方便的實(shí)現(xiàn)auto-loading作者是validation。
但是，但是，但是?。。。。。?！同志們一定要注意，這里面有一個(gè)小小的坑。
閑言碎語不要說，我們直接上代碼

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);

/**
輸出結(jié)果
use 01
get 011
**/

相信眼尖的通知們已經(jīng)知道發(fā)生了什么！沒錯(cuò)，param對應(yīng)的fn執(zhí)行了兩遍，這是因?yàn)閜aram會(huì)把中間件放到layer的middleware數(shù)組中。
好了，雖然有點(diǎn)小小的瑕疵，但是，用的時(shí)候注意就好了。這個(gè)功能還是很好用的，吼吼吼。

以上就是koa-router我想說的全部內(nèi)容，至于怎么用的話，看一下api文檔就好了，我只能說作者設(shè)計(jì)的很簡潔，一切說明都是廢話。

希望我也能早日設(shè)計(jì)出這么有沒得api。

<h4 id="koa-body">koa-body</h4>

這個(gè)組件使用起來也是非常方便的，在這里說一下的目的是，由于前一陣子公司使用的body parse用起來難用的不要不要的。所以在這里推薦一下吧。

下面的內(nèi)容基本就是readme的翻譯，沒興趣的可以直接不看。

<h5>1.簡介</h5>
koa-body支持multipart, urlencoded 和 json 請求體，提供跟express的multer一樣的功能。他是對co-body和formidable的封裝。

<h5>2.使用方法</h5>
koa-body可以向multer一樣使用，非常簡單，因?yàn)槟憧梢詮腸tx.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());

<h5>3.參數(shù)</h5>

• patchNode bool 把request的body給node的ctx.req，默認(rèn)為false。
• patchKoa bool 把request的body給koa的ctx.req，默認(rèn)為false。
• jsonLimit String|Integer json body的最大byte數(shù)，默認(rèn)為1mb
• formLimit String|Integer form body的最大byte數(shù)，默認(rèn)為56kb
• textLimit String|Integer text body的最大byte數(shù)，默認(rèn)為56kb
• encoding String 設(shè)置field的編碼，默認(rèn)為utf-8
• multipart Boolean 是否解析multipart body，默認(rèn)為false
• formidable Object 床底給formidable的設(shè)置項(xiàng)
• strict bool 如果激活，koa-body不會(huì)解析GET,HEAD,DELETE請求，默認(rèn)為true