简单实现node中间件机制-仿express

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进行扩展性更好的封装。本文只是简单实现中间件机制，核心并不在业务逻辑处理，所以只是使用了简单的函数传参的方式。