koa2中间件原理

koa2中间件的执行

koa2中间件的执行就像洋葱圈一样，从外面到最里面，再从最里面到最外面。

const Koa = require('koa');
const app = new Koa();

// logger

app.use(async (ctx, next) => {
    console.log('第一层洋葱圈开始')
    await next();
    const rt = ctx.response.get('X-Response-Time');
    console.log('第一层洋葱圈结束')
    console.log(`${ctx.method} ${ctx.url} - ${rt}`);
});

// x-response-time

app.use(async (ctx, next) => {
    console.log('第二层洋葱圈开始')
    const start = Date.now();
    await next();
    const ms = Date.now() - start;
    ctx.set('X-Response-Time', `${ms}ms`);
    console.log('第二层洋葱圈结束')
});

// response

app.use(async ctx => {
    console.log('第三层洋葱圈开始')
    ctx.body = 'Hello World';
    console.log('第三层洋葱圈结束')
});

app.listen(3000);

执行上述代码的结果是：
第一层洋葱圈开始
第二层洋葱圈开始
第三层洋葱圈开始
第三层洋葱圈结束
第二层洋葱圈结束
第一层洋葱圈结束

koa2中间件原理的实现

知道了，ko2中间件的执行过程，那么我们如何去手写实现这样的原理。
koa2中的use()注册中间件函数不包含拦截路由的功能，只实现next机制，即上一个通过next触发下一个。

like-koa2.js文件中

const http = require('http')
//组合中间件
function compose(middlewareList){
    //compose函数返回一个函数
    return function (ctx){
        //dispath函数，来实现一个函数中的next，代表下一个函数
        function dispatch(i){
            const fn = middlewareList[i]
            try{
                //返回promise是为了没有使用async的情况
                return Promise.resolve(
                    fn(ctx,dispatch.bind(null,i+1)) 
                )
            }catch(err){
                return Promise.reject(err)
            }
        }

        return dispatch(0)  //返回执行dispath函数,传参为0
    }
}

class LikeKo2{
    constructor(){
        this.middlewareList = []
    }

    use(fn){
        this.middlewareList.push(fn)    //往中间件列表加入函数
        return this     //使其可以链式调用
    }

    createContext(req,res){
        const ctx = {
            req,res
        }
        ctx.query = req.query

        return ctx
    }

    handleRequest(ctx,fn){
        return fn(ctx)
    }

    callback(){
        const fn = compose(this.middlewareList)

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

    listen(...args){
        const server = http.createServer(this.callback())
        server.listen(...args)
    }
}

module.exports = LikeKo2

test.js文件中

const Koa = require('./like-koa2');
const app = new Koa();

// logger
app.use(async (ctx, next) => {
  await next();
  const rt = ctx['X-Response-Time'];
  console.log(`${ctx.req.method} ${ctx.req.url} - ${rt}`);
});

// x-response-time
app.use(async (ctx, next) => {
  const start = Date.now();
  await next();
  const ms = Date.now() - start;
  ctx['X-Response-Time'] = `${ms}ms`;
});

// response
app.use(async ctx => {
  ctx.res.end('This is like koa2');
});

app.listen(8000);