95 # express 二级路由的实现

上一节实现了兼容老的路由写法，这一节来实现二级路由

二级路由实现核心：

• 进入中间件后，让对应的路由系统去进行匹配操作
• 中间件进去匹配需要删除 path，存起来出去时在加上

示意图：

代码实现如下：

router/index.js

const url = require("url");
const Route = require("./route");
const Layer = require("./layer");
const methods = require("methods");

function Router() {
    // 创建路由系统
    let router = (req, res, next) => {
        // 二级路由
        // 让对应的路由系统去进行匹配操作
        router.handle(req, res, next);
    };
    // 兼容老的路由写法
    // 维护所有的路由
    router.stack = [];
    // router 链上得有 get 等方法
    // 让路由的实例可以通过链找到原来的方法
    router.__proto__ = proto;
    return router;
}

let proto = {};

proto.route = function (path) {
    // 产生 route
    let route = new Route();
    // 产生 layer 让 layer 跟 route 进行关联
    let layer = new Layer(path, route.dispatch.bind(route));
    // 每个路由都具备一个 route 属性，稍后路径匹配到后会调用 route 中的每一层
    layer.route = route;
    // 把 layer 放到路由的栈中
    this.stack.push(layer);
    return route;
};

methods.forEach((method) => {
    proto[method] = function (path, handlers) {
        // 1.用户调用 method 时，需要保存成一个 layer 当道栈中
        // 2.产生一个 Route 实例和当前的 layer 创造关系
        // 3.要将 route 的 dispatch 方法存到 layer 上
        let route = this.route(path);
        // 让 route 记录用户传入的 handler 并且标记这个 handler 是什么方法
        route[method](handlers);
    };
});

proto.use = function (path, ...handlers) {
    // 默认第一个是路径，后面是一个个的方法，路径可以不传
    if (typeof path === "function") {
        handlers.unshift(path);
        path = "/";
    }
    // 如果是多个函数需要循环添加层
    for (let i = 0; i < handlers.length; i++) {
        let layer = new Layer(path, handlers[i]);
        // 中间件不需要 route 属性
        layer.route = undefined;
        this.stack.push(layer);
    }
};

proto.handle = function (req, res, out) {
    console.log("请求到了");
    // 需要取出路由系统中 Router 存放的 layer 依次执行
    let { pathname } = url.parse(req.url);
    let idx = 0;
    let removed = "";
    let next = (err) => {
        // 遍历完后没有找到就直接走出路由系统
        if (idx >= this.stack.length) return out();
        let layer = this.stack[idx++];
        // 中间件出来需要加上 removed，继续往下匹配
        if (removed) {
            req.url = removed + pathname; // 匹配其他中间件
            removed = "";
        }
        if (err) {
            console.log("统一对中间件跟路由错误处理");
            // 找错误处理中间件
            if (!layer.route) {
                // 如果是中间件自己处理
                layer.handle_error(err, req, res, next);
            } else {
                // 路由则跳过，继续携带错误向下执行
                next(err);
            }
        } else {
            // 需要判断 layer 上的 path 和当前请求路由是否一致，一致就执行 dispatch 方法
            if (layer.match(pathname)) {
                // 中间件没有方法可以匹配，不能是错误处理中间件
                if (!layer.route) {
                    if (layer.handler.length !== 4) {
                        // 中间件进去匹配需要删除 path，存起来出去时用
                        // 比如我们访问 /user/add，那么需要删除 /user，用 /add 去匹配用户里的路由是否有 /add
                        if (layer.path !== "/") {
                            removed = layer.path;
                            req.url = pathname.slice(removed.length);
                        }
                        layer.handle_request(req, res, next);
                    } else {
                        // 错误中间件
                        next();
                    }
                } else {
                    // 将遍历路由系统中下一层的方法传入
                    // 加速匹配，如果用户注册过这个类型的方法在去执行
                    if (layer.route.methods[req.method.toLowerCase()]) {
                        layer.handle_request(req, res, next);
                    } else {
                        next();
                    }
                }
            } else {
                next();
            }
        }
    };
    next();
};

module.exports = Router;

route.js：主要改造一下报错问题，需要判断 handlers 是否是数组，上一节是临时写了 ...handlers

const Layer = require("./layer");
const methods = require("methods");

function Route() {
    this.stack = [];
    // 用来描述内部存过哪些方法
    this.methods = {};
}

Route.prototype.dispatch = function (req, res, out) {
    // 稍后调用此方法时，回去栈中拿出对应的 handler 依次执行
    let idx = 0;
    console.log("this.stack----->", this.stack);
    let next = (err) => {
        // 如果内部迭代的时候出现错误，直接到外层的 stack 中
        err && out(err);
        // 遍历完后没有找到就直接走出路由系统
        if (idx >= this.stack.length) return out();
        let layer = this.stack[idx++];
        console.log("dispatch----->", layer.method);
        if (layer.method === req.method.toLowerCase()) {
            layer.handle_request(req, res, next);
        } else {
            next();
        }
    };
    next();
};
methods.forEach((method) => {
    Route.prototype[method] = function (handlers) {
        console.log("handlers----->", handlers);
        // 需要判断 handlers 是否是数组
        if (!Array.isArray(handlers)) {
            handlers = [handlers];
        }
        handlers.forEach((handler) => {
            // 这里的路径没有意义
            let layer = new Layer("/", handler);
            layer.method = method;
            // 做个映射表
            this.methods[method] = true;
            this.stack.push(layer);
        });
    };
});

module.exports = Route;