express中间件机制原理
首先，我们看一下express使用是怎么使用中间件的。
const express = require('express');
const app = express();
app.use(middlewareA);
app.use(middlewareB);
其实app.use()原理很简单，就是把中间件添加到一个待执行的数组中，当监听到网络请求时，会先依次执行这个数组里面的所有中间件。
使用node实现
实现app.use方法
const task = () => {}
let funcs = [];
let app = {};
app.use = (task) => {
funcs.push(task);
}
app.use(task);
console.log(funcs);
实现express中间件机制
/**
* 仿express实现中间件机制
* @param {func} handler 业务处理逻辑
* @return {object} app 服务器实例
*/
function express(handler) {
// 定义中间件执行数组
const funcs = [];
const app = (req, res) => {
let i = 0;
// 调用下一个中间件
let next = () => {
const task = funcs[i++];
if (!task) {
// 中间件执行结束，进行业务逻辑处理
handler(req, res);
return;
}
task(req, res, next);
}
// 调用中间件
next();
}
/**
* use方法就是把函数添加到函数组中
* @param {func} task 中间件
*/
app.use = (task) => {
funcs.push(task);
}
return app;
}
一个简单的get请求示例
const http = require('http');
const url = require('url');
/**
* 仿express实现中间件机制
* @param {func} handler 业务处理逻辑
* @return {object} app 服务器实例
*/
function express(handler) {
const funcs = [];
const app = (req, res) => {
let i = 0;
let next = () => {
const task = funcs[i++];
if (!task) {
// 中间件执行结束，进行业务逻辑处理
handler(req, res);
return;
}
task(req, res, next);
}
// 调用中间件
next();
}
/**
* use方法就是把函数添加到函数组中
* @param {func} task 中间件
*/
app.use = (task) => {
funcs.push(task);
}
return app;
}
// 业务逻辑
const handleRequest = (req, res) => {
// 模拟用户组数据
const users = [
{
id: '1',
name: 'maik'
}, {
id: '2',
name: 'niub'
}
];
// 获取url参数
const arg = url.parse(req.url, true).query;
const { id } = arg;
// 匹配数据
const data = users.find((i) => (i.id === id));
// 返回数据
res.setHeader('Content-Type', 'application/json;chartset=uft-8');
res.end(JSON.stringify(data));
}
// 实例化服务
const app = express(handleRequest);
// 开启服务监听端口
http.createServer(app).listen('3000', () => {
console.log('listening 3000……');
});
// 定义中间件 - 模拟鉴权
const middlewareA = (req, res, next) => {
if (req.url === '/') {
res.statusCode = 403;
res.end('无权访问！');
}
next();
}
// 定义中间件 - 模拟过滤
const middlewareB = (req, res, next) => {
if (req.url === '/user') {
// 添加参数
req.url = '/user?id=1';
}
next();
}
app.use(middlewareA);
app.use(middlewareB);
写在最后
中间件执行完毕进行业务逻辑处理的模块，还可以用Promise进行扩展性更好的封装。本文只是简单实现中间件机制，核心并不在业务逻辑处理，所以只是使用了简单的函数传参的方式。
