koa源码

最新推荐文章于 2024-04-10 08:51:40 发布

原创最新推荐文章于 2024-04-10 08:51:40 发布 · 790 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#koa #node

javascript 专栏收录该内容

35 篇文章

订阅专栏

写在前面

本文将会大家来看下koa的源码，当然本文需要大家了解koa的中间件机制，如果大家之前没有了解过其实现原理，可以关注下这篇文章。
koa的源码非常的精简，与express不同，koa只是为开发者搭起了一个架子，没有任何的功能，包括路由，全部由中间件实现；下面就来看下koa的实现：

koa

创建应用时，一般都会利用app.listen指定一个端口号，这个方法的本质就是http.createServer：

listen() {
    debug('listen');
    const server = http.createServer(this.callback());
    return server.listen.apply(server, arguments);
}

最为关键的就是这个callback的实现：

这里写图片描述

callback() {
    const fn = compose(this.middleware);

    if (!this.listeners('error').length) this.on('error', this.onerror);

    const handleRequest = (req, res) => {
        res.statusCode = 404;
        const ctx = this.createContext(req, res);
        const onerror = err => ctx.onerror(err);
        const handleResponse = () => respond(ctx);
        onFinished(res, onerror);
        return fn(ctx).then(handleResponse).catch(onerror);
    };

    return handleRequest;
}

需要注意下面几点：
- onFinished(res, onerror)，应对的是返回的body为Stream的情况，为其添加一个finished事件。
- respond()根据ctx的status,body,method来决定如何响应这次请求：
- status为204,304，不需要有响应体，res.end()就好
- method为HEAD，HEAD的意义是不请求资源内容但是需要了解资源情况，所以只需要请求头，指定了资源length后res.end()就好
- 加入body为空，则body为statuses包中status对应的文字描述，如404 => Not Found

context对象

koa将request对象response封装成了一个对象，提供了一些别名，具体可以参见context对象，例如：当访问ctx.url实则是访问的ctx.request.url。具体的实现利用了tj写的delegates这个npm包来对context对象添加属性，koa中利用了其中三个api:

method：添加方法引用
getter：利用__defineGetter__，添加getter属性
access：添加getter与setter

对于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;
  }