Node.js中如何在控制器函数内使用中间件？

问题描述 #

我想给request.body添加Schema校验的中间件，需求是：

• 校验通过则调用next()继续执行控制器函数；
• 校验不通过则调用next(error)跳转至错误处理中间件。

这个中间件在路由配置中使用正常，但我想把它放到需要的控制器函数开头，而非路由里。现在遇到错误："Cannot set headers after they are sent to the client"，虽然中间件没直接操作response，但我不确定能不能在控制器内集成中间件？

附上相关代码：

List Controller (app/controllers/list.controller.js) #

updateListById: async function (request, response, next) {
        validateList(request.body, next);

        const { position, name } = request.body;
        const { id } = request.params;

        try {
            const list = await List.findByPk(id);
            list.name = name;
            list.position = position;
            await list.save();

            response.json(list);
        } catch (error) {
            console.trace(error);
            next(error);
        }
    },

List Validator (app/validator/list.validator.js) #

const z = require("zod");
const validateData = require("../utils/validateData");

const schemaData = z.object({
    name: z.string().min(2, { message: "Le nom doit faire au moins deux caractères." }),
    position: z.number({ message: "Veuillez remplir la position." }),
});

const listValidator = {
    validateList: function (listData, next) {
        return validateData(listData, schemaData, next);
    },
};

module.exports = listValidator;

Data Validator (app/utils/validateData.js) #

function validateData(data, schema, next) {
    const z = schema.parse(data);
    if (z.errors) return next(errors);
    next();
}

module.exports = validateData;

问题分析与解决 #

1. 错误原因 #

你遇到的Cannot set headers after they are sent to the client错误，本质是重复发送响应：

• 当校验不通过时，你的校验逻辑调用next(error)，触发错误处理中间件发送响应；
• 但控制器内后续的数据库操作和response.json(list)代码仍会继续执行，尝试再次发送响应，导致冲突。

另外，你的Zod校验逻辑也有错误：Zod的parse方法在校验失败时会直接抛出ZodError，而不是返回包含errors字段的对象，所以原来的if (z.errors)判断完全无效。

2. 解决方案 #

步骤一：修复Zod校验逻辑

修改app/utils/validateData.js，改用safeParse方法（它会返回包含校验结果的对象，不会直接抛出错误）：

function validateData(data, schema, next) {
    const result = schema.safeParse(data);
    if (!result.success) {
        // 传递Zod的错误对象给错误处理中间件
        return next(result.error);
    }
    next();
    // 返回校验结果，方便控制器判断是否继续执行
    return true;
}

或者用try/catch包裹parse方法：

function validateData(data, schema, next) {
    try {
        schema.parse(data);
        next();
        return true;
    } catch (error) {
        next(error);
        return false;
    }
}

步骤二：控制器内终止后续代码执行

在控制器调用校验函数后，必须判断校验结果，若失败则立即return，阻止后续代码执行：

updateListById: async function (request, response, next) {
        // 调用校验，若失败则直接返回，不执行后续逻辑
        const isValid = validateList(request.body, next);
        if (!isValid) return;

        const { position, name } = request.body;
        const { id } = request.params;

        try {
            const list = await List.findByPk(id);
            list.name = name;
            list.position = position;
            await list.save();

            response.json(list);
        } catch (error) {
            console.trace(error);
            next(error);
        }
    },

3. 结论 #

完全可以在控制器内集成校验逻辑，但要注意两个核心点：

• 正确处理Zod的校验错误，避免无效的错误判断；
• 校验失败时必须终止控制器后续代码的执行，防止重复发送响应。

内容的提问来源于stack exchange，提问作者Enzo