浅析 Node 中间件 Koa&Express：原理和实现

const Koa=require('koa')const app=new Koa()const Router=require('koa-router')const router=new Router()const cors=require('koa2-cors')const koaBody=require('koa-body')
const ENV='test-mpin2'
app.use(cors({	origin:['http://localhost:9528'],   // 也可以写为：['*']	credentials:true}))app.use(koaBody({	multipart:true}))app.use(async(ctx,next)=>{	console.log('访问全局中间件')	ctx.state.env=ENV   // 全局缓存	await next()})
const playlist=require('./controller/playlist.js')router.use('/playlist',playlist.routes())const blog=require('./controller/blog.js')router.use('/blog',blog.routes())
app.use(router.routes()).use(router.allowedMethods())
app.listen(3000,()=>{	console.log('服务已开启')})

// 最外层中间件，可以用于兜底 Koa 全局错误app.use(async (ctx, next) => {  try {    // 执行下一个中间件    await next();  } catch (error) {    console.log(`[koa error]: ${error.message}`)  }});// 第二层中间件，可以用于日志记录app.use(async (ctx, next) => {  const { req } = ctx;  console.log(`req is ${JSON.stringify(req)}`);  await next();  console.log(`res is ${JSON.stringify(ctx.res)}`);});

use(fn) {    this.middleware.push(fn);    return this;}

// 通过 createServer 方法启动一个 Node.js 服务listen(...args) {    const server = http.createServer(this.callback());    server.listen(...args);}

callback(){	const fn=compose(this.middlewareList)		return (req,res)=>{		const ctx=createContext(req,res)		return this.handleRequest(ctx,fn)	}}
handleRequest(ctx, fn) {    const onerror = err => ctx.onerror(err);    // 将 ctx 对象传递给中间件函数 fn    return fn(ctx).catch(onerror);}

// 组合中间件// 和express中的next函数意义一样function compose(middlewareList){	// return function意思是返回一个函数	return function(ctx,next){		let index=-1;		// 各种中间件调用的逻辑		function dispatch(i){			if(i<=index) return Promise.reject(new Error('next() called multiple times'))			index=i			const fn=middlewareList[i] || next			if(fn){				try{					// koa中都是async，其返回的是一个promise（对象）					return Promise.resolve(fn(ctx,function next(){						return dispatch(i+1)					}))				}catch(err){					return Promise.reject(err)				}			}else{				return Promise.resolve()			}		}		return dispatch(0)	}}

async function middleware1() {  //...  await (async function middleware2() {    //...    await (async function middleware3() {      //...    });    //...  });  //...}

// myKoa.js文件
const http=require('http')
function compose(){}   //见上
class LikeKoa2{	constructor() {	    this.middlewareList=[]	}	use(){}   //见上		// 把所有的req，res属性、事件都交给ctx（这里只是简写）	createContext(req,res){		const ctx={			req,			res		}		// 比如		ctx.query=req,query		return ctx	}	handleRequest(){}   //见上	callback(){}   //见上	listen(){}   //见上}
// koa和express的不同之一：// express在调用时直接调用函数：const app=express();所以暴露出去new过的对象——具体见下面链接中代码// 但是koa调用时以类的方式：const app=new Koa();所以直接暴露出去module.exports=LikeKoa2

((req, res) => {  console.log('第一个中间件');  ((req, res) => {    console.log('第二个中间件');    (async(req, res) => {      console.log('第三个中间件');      await sleep(2000)      res.status(200).send('hello')    })(req, res)    console.log('第二个中间件调用结束');  })(req, res)  console.log('第一个中间件调用结束')})(req, res)