96 lines
2.8 KiB
JavaScript
96 lines
2.8 KiB
JavaScript
const { MongoClient } = require("mongodb");
|
||
|
||
// 震元堂环境 // 注意后面的配置, 根据实际情况修改
|
||
const username = process.env.CONFIG_DB_USERNAME;
|
||
const password = process.env.CONFIG_DB_PASSWORD;
|
||
const host = process.env.CONFIG_DB_HOST; // 例如：'localhost' 或 MongoDB Atlas提供的连接地址
|
||
|
||
const port = process.env.CONFIG_DB_PORT || "27017"; // MongoDB默认端口为'27017'
|
||
|
||
// 构建连接字符串，包含身份验证信息
|
||
const url = `mongodb://${username}:${password}@${host}:${port}/admin`;
|
||
console.log(`mongoDB url: ${url}`);
|
||
const options = {
|
||
maxPoolSize: 100, // 设置最大连接池大小为 100
|
||
minPoolSize: 10, // 设置最小连接池大小为 10
|
||
connectTimeoutMS: 60000, // 连接超时 60 秒
|
||
socketTimeoutMS: 60000, // 数据传输超时 60 秒
|
||
// useUnifiedTopology 在 MongoDB Driver 4.0+ 中已被废弃，无需设置
|
||
};
|
||
let client;
|
||
let isConnecting = false;
|
||
|
||
async function connectToMongoDB() {
|
||
try {
|
||
console.log("连接 MongoDB");
|
||
if (!client && !isConnecting) {
|
||
isConnecting = true;
|
||
|
||
// 检查必要的环境变量
|
||
if (!username || !password || !host) {
|
||
throw new Error("MongoDB连接参数缺失: 请检查环境变量 CONFIG_DB_USERNAME, CONFIG_DB_PASSWORD, CONFIG_DB_HOST");
|
||
}
|
||
|
||
client = new MongoClient(url, options);
|
||
await client.connect();
|
||
console.log("MongoDB 连接成功");
|
||
isConnecting = false;
|
||
}
|
||
return client;
|
||
} catch (err) {
|
||
isConnecting = false;
|
||
console.error("MongoDB连接失败:", err);
|
||
throw err; // 抛出错误而不是静默失败
|
||
}
|
||
}
|
||
|
||
async function getDatabase(dbName) {
|
||
if (!client) {
|
||
throw new Error("MongoDB客户端未连接，请先调用 connectToMongoDB()");
|
||
}
|
||
|
||
try {
|
||
const db = client.db(dbName);
|
||
// 测试数据库连接
|
||
await db.admin().ping();
|
||
return db;
|
||
} catch (err) {
|
||
console.error(`获取数据库 ${dbName} 失败:`, err);
|
||
throw new Error(`无法访问数据库 ${dbName}: ${err.message}`);
|
||
}
|
||
}
|
||
|
||
async function closeMongoDB() {
|
||
if (client) {
|
||
await client.close();
|
||
client = null;
|
||
}
|
||
}
|
||
|
||
// 检查连接状态
|
||
function isConnected() {
|
||
return client && client.topology && client.topology.isConnected();
|
||
}
|
||
|
||
// exports.connectToMongoDB = async (item, type, dbName) => {
|
||
// try {
|
||
// const db = await getDatabase(dbName);
|
||
// // 获取 serverStatus 的连接数信息
|
||
// const res = await type.main(item, db);
|
||
// return res;
|
||
// } catch (err) {
|
||
// // 细化错误信息
|
||
// return {
|
||
// success: false,
|
||
// message: err.message,
|
||
// };
|
||
// }
|
||
// };
|
||
process.on("SIGINT", async () => {
|
||
console.log("关闭 MongoDB 连接");
|
||
await closeMongoDB();
|
||
process.exit(0);
|
||
});
|
||
|
||
module.exports = { connectToMongoDB, getDatabase, closeMongoDB, isConnected };
|