87 lines
2.2 KiB
JavaScript
87 lines
2.2 KiB
JavaScript
const { isConnected, getDatabase } = require("../mongodb");
|
|
const logger = require("./logger");
|
|
|
|
/**
|
|
* 检查数据库连接健康状态
|
|
* @returns {Promise<boolean>} 数据库是否健康
|
|
*/
|
|
async function checkDatabaseHealth() {
|
|
try {
|
|
if (!isConnected()) {
|
|
logger.warn("数据库连接状态异常");
|
|
return false;
|
|
}
|
|
|
|
// 尝试ping几个关键数据库
|
|
const databases = ["admin", "corp", "Internet-hospital"];
|
|
|
|
for (const dbName of databases) {
|
|
try {
|
|
const db = await getDatabase(dbName);
|
|
await db.admin().ping();
|
|
} catch (error) {
|
|
logger.error(`数据库 ${dbName} 连接检查失败:`, error);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
} catch (error) {
|
|
logger.error("数据库健康检查失败:", error);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 安全获取数据库实例，包含错误处理
|
|
* @param {string} dbName 数据库名称
|
|
* @returns {Promise<Object>} 数据库实例
|
|
*/
|
|
async function safeGetDatabase(dbName) {
|
|
try {
|
|
if (!isConnected()) {
|
|
throw new Error("数据库连接已断开");
|
|
}
|
|
|
|
const db = await getDatabase(dbName);
|
|
|
|
// 测试连接
|
|
await db.admin().ping();
|
|
|
|
return db;
|
|
} catch (error) {
|
|
logger.error(`获取数据库 ${dbName} 失败:`, error);
|
|
throw new Error(`数据库连接失败: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 数据库操作包装器，自动处理连接错误
|
|
* @param {Function} operation 数据库操作函数
|
|
* @param {string} operationName 操作名称（用于日志）
|
|
* @returns {Function} 包装后的操作函数
|
|
*/
|
|
function withDatabaseErrorHandling(operation, operationName = "数据库操作") {
|
|
return async (...args) => {
|
|
try {
|
|
return await operation(...args);
|
|
} catch (error) {
|
|
logger.error(`${operationName}失败:`, error);
|
|
|
|
// 检查是否是数据库连接相关的错误
|
|
if (error.message.includes("collection") ||
|
|
error.message.includes("database") ||
|
|
error.message.includes("connection")) {
|
|
throw new Error(`数据库连接错误: ${error.message}`);
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
};
|
|
}
|
|
|
|
module.exports = {
|
|
checkDatabaseHealth,
|
|
safeGetDatabase,
|
|
withDatabaseErrorHandling
|
|
};
|