86 lines
1.9 KiB
JavaScript
86 lines
1.9 KiB
JavaScript
const logger = require("./logger");
|
||
|
||
/**
|
||
* 安全发送响应，防止重复发送导致的ERR_HTTP_HEADERS_SENT错误
|
||
* @param {Object} res - Express响应对象
|
||
* @param {*} data - 要发送的数据
|
||
* @param {boolean} isJson - 是否以JSON格式发送
|
||
* @returns {boolean} - 是否成功发送
|
||
*/
|
||
function safeSend(res, data, isJson = true) {
|
||
if (res.headersSent) {
|
||
logger.warn("尝试发送响应但响应头已发送");
|
||
return false;
|
||
}
|
||
|
||
try {
|
||
if (isJson) {
|
||
res.json(data);
|
||
} else {
|
||
res.send(data);
|
||
}
|
||
return true;
|
||
} catch (error) {
|
||
logger.error("发送响应失败:", error);
|
||
return false;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 发送成功响应
|
||
* @param {Object} res - Express响应对象
|
||
* @param {*} data - 成功数据
|
||
* @param {string} message - 成功消息
|
||
*/
|
||
function sendSuccess(res, data = null, message = "请求成功") {
|
||
return safeSend(res, {
|
||
success: true,
|
||
message,
|
||
data
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 发送错误响应
|
||
* @param {Object} res - Express响应对象
|
||
* @param {string} message - 错误消息
|
||
* @param {*} error - 错误详情
|
||
* @param {number} status - HTTP状态码
|
||
*/
|
||
function sendError(res, message = "请求失败!", error = null, status = 400) {
|
||
if (!res.headersSent) {
|
||
res.status(status);
|
||
}
|
||
|
||
return safeSend(res, {
|
||
success: false,
|
||
message,
|
||
error
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 包装异步路由处理器，自动处理错误
|
||
* @param {Function} fn - 异步路由处理函数
|
||
* @returns {Function} - 包装后的处理函数
|
||
*/
|
||
function asyncHandler(fn) {
|
||
return (req, res, next) => {
|
||
Promise.resolve(fn(req, res, next)).catch(error => {
|
||
logger.error("异步路由错误:", error);
|
||
if (!res.headersSent) {
|
||
sendError(res, "服务器内部错误",
|
||
error && error.message ? error.message : '',
|
||
500
|
||
);
|
||
}
|
||
});
|
||
};
|
||
}
|
||
|
||
module.exports = {
|
||
safeSend,
|
||
sendSuccess,
|
||
sendError,
|
||
asyncHandler
|
||
};
|