本文深入探讨了 Mongoose/MongoDB 脚本在执行 drop() 或 insertMany() 等操作时出现挂起的问题。核心原因在于对 Mongoose 连接状态和事件监听机制的误解，特别是滥用 connection.once(‘connected’, …) 事件。文章将详细解释为何连接已就绪时该事件不再触发，并提供正确的连接管理和操作执行策略，确保脚本可靠运行。
问题现象分析
在开发 Node.js 应用程序时，使用 Mongoose 与 MongoDB 交互是常见的场景。有时，开发者可能会遇到脚本在执行数据库操作（如删除集合 dropCollection() 或插入数据 insertMany()）时无限期挂起的情况。尽管 console.log(connection.readyState) 显示连接状态为 1（表示已连接），但后续的数据库操作却迟迟不执行。
一个典型的导致此问题的代码结构如下：
// ... 其他代码，包括导入 mongoose 和 connection 对象
(async () => {
try {
// ... 文件下载和数据转换逻辑
console.log("Connection state: ", connection.readyState); // 此时可能输出 1
// 问题所在：将数据库操作包裹在 connection.once('connected', ...) 中
connection.once('connected', async () => {
try {
console.log("Dropping collection if it already exists");
await connection.dropCollection('mdprodcats');
console.log("Collection dropped successfully");
} catch (error) {
if (error.message === 'ns not found') {
console.log('Collection does not exist. Continuing...');
} else {
throw error;
}
}
console.log("Inserting data...");
await MdProdCat.insertMany(json);
console.log("Data inserted successfully");
console.log("Closing connection...");
await connection.close();
console.log("Connection closed");
console.log("=== Import completed successfully");
process.exit();
});
} catch (error) {
console.error("An error occurred:", error);
process.exit(1);
}
})();
在上述代码中，当 connection.readyState 已经为 1（表示连接已成功建立）时，将数据库操作包裹在 connection.once(‘connected’, …) 回调函数中，会导致脚本挂起。
根本原因：Mongoose 连接事件的生命周期
Mongoose 的 connection.once(‘connected’, …) 是一个事件监听器，它设计用于在 Mongoose 连接到 MongoDB 服务器首次成功建立时触发一次。
关键在于“首次成功建立时”。如果你的 Mongoose 连接（通常是 mongoose.connection 对象）在脚本执行到 connection.once(‘connected’, …) 这一行时，已经处于 connected 状态（即 readyState 为 1），那么 connected 事件已经触发过了。由于 once 的特性，它不会再次触发。
因此，包裹在 connection.once(‘connected’, …) 内部的所有数据库操作代码将永远不会被执行，导致脚本看起来像“挂起”了，因为它在等待一个永远不会发生的事件。
这种情况尤其容易发生在以下场景：
- 全局连接对象： 如果你的应用程序在启动时就建立了 Mongoose 连接，并将其作为全局或模块导出的单例使用，那么在后续的任何脚本中，当获取到这个连接对象时，它很可能已经处于连接状态。
- 异步操作序列： 如果 mongoose.connect() 在当前脚本的早期阶段就已经完成并成功连接，那么后续尝试使用 once(‘connected’, …) 来等待连接的逻辑就是多余的，甚至是有害的。
解决方案
解决此问题的核心是确保在连接已经就绪时，直接执行数据库操作，而不是等待一个可能永远不会触发的事件。
最直接有效的解决方案是移除 connection.once(‘connected’, …) 这个包裹层，并在确认连接状态后直接执行操作。
const axios = require("axios");
const XLSX = require("xlsx");
const fs = require("fs");
const path = require("path");
const mongoose = require("mongoose");
const connection = require("../../../config/connection"); // 假设 connection 是 mongoose.connection 实例
(async () => {
try {
// 下载和数据转换逻辑
console.log("=== Commencing MD Product Category import...");
console.log("Downloading .xlsx file...");
const response = await axios.get(
"https://assets.mydeal.com.au/content/marketplace/MyDeal_Product_Category_List.xlsx",
{ responseType: "arraybuffer" }
);
console.log("Saving .xlsx file...");
const xlsxFilePath = path.join(
__dirname,
"MyDeal_Product_Category_List.xlsx"
);
fs.writeFileSync(xlsxFilePath, Buffer.from(response.data));
console.log("Saved .xlsx file");
console.log("Converting .xlsx to .csv...");
const workbook = XLSX.read(response.data, { type: "buffer" });
const sheetNameList = workbook.SheetNames;
const csvData = XLSX.utils.sheet_to_csv(workbook.Sheets[sheetNameList[0]]);
console.log("Converted .xlsx to .csv");
console.log("Converting .csv to JSON...");
const rows = csvData.split("\n").slice(1);
const json = rows
.filter((line) => line.trim())
.map((line) => {
const [CategoryID, Breadcrumbs] = line.split(",");
return { _id: parseInt(CategoryID), Breadcrumbs };
});
console.log("Converted .csv to JSON");
// 检查连接状态。如果 connection 已经就绪，直接执行操作。
// 在一个一次性脚本中，更推荐在脚本开始时使用 await mongoose.connect()
// 确保连接建立后再进行后续操作。
console.log("Connection state: ", connection.readyState);
// 直接执行数据库操作，不再等待 'connected' 事件
try {
console.log("Dropping collection if it already exists");
await connection.dropCollection('mdprodcats');
console.log("Collection dropped successfully");
} catch (error) {
if (error.message === 'ns not found') {
console.log('Collection does not exist. Continuing...');
} else {
throw error;
}
}
console.log("Inserting data...");
await MdProdCat.insertMany(json);
console.log("Data inserted successfully");
console.log("Closing connection...");
await connection.close();
console.log("Connection closed");
console.log("=== Import completed successfully");
process.exit();
} catch (error) {
console.error("An error occurred:", error);
process.exit(1);
}
})();
Mongoose 连接管理最佳实践
为了避免类似的挂起问题，并确保 Mongoose 应用程序的健壮性，请遵循以下连接管理最佳实践：
-
使用 mongoose.connect() 的 Promise 返回值：
对于一次性脚本或需要确保连接就绪后才执行操作的场景，直接 await mongoose.connect() 是最清晰和可靠的方式。const mongoose = require('mongoose'); const DB_URI = 'mongodb://localhost:27017/yourdb'; // 你的数据库URI (async () => { try { console.log("Connecting to MongoDB..."); // 等待连接成功 await mongoose.connect(DB_URI, { useNewUrlParser: true, useUnifiedTopology: true, // 其他选项 }); console.log("MongoDB connected successfully."); // 连接成功后，直接执行数据库操作 // 例如： // await mongoose.connection.dropCollection('yourcollection'); // await YourModel.insertMany(data); console.log("Operations completed."); await mongoose.connection.close(); // 完成操作后关闭连接 console.log("Connection closed."); process.exit(0); } catch (error) { console.error("MongoDB connection error or operation failed:", error); process.exit(1); } })(); -
区分应用程序启动和单次脚本：
- 对于长期运行的应用程序（如 Web 服务器）： 在应用程序启动时连接数据库一次，并保持连接。可以在 app.js 或 server.js 中使用 mongoose.connect()，并监听连接事件（connected, error, disconnected）来处理连接状态变化，但业务逻辑不应依赖 once(‘connected’, …)。
- 对于一次性脚本（如数据导入/导出）： 脚本开始时连接数据库，执行操作，然后关闭连接。确保 await mongoose.connect() 或类似机制等待连接就绪。
-
Mongoose 连接状态检查：
Mongoose 提供 mongoose.connection.readyState 属性来检查当前连接状态：- 0: disconnected (断开连接)
- 1: connected (已连接)
- 2: connecting (连接中)
- 3: disconnecting (断开连接中)
在执行数据库操作前，可以通过检查 connection.readyState === 1 来确认连接是否可用，但这通常在 mongoose.connect() 成功后就已满足。
-
错误处理和连接关闭：
- 始终在异步操作中使用 try…catch 进行错误处理。
- 对于一次性脚本，在所有数据库操作完成后，务必调用 mongoose.connection.close() 来关闭连接，以确保脚本正常退出，避免进程挂起。
总结
Mongoose 脚本挂起的问题，通常不是因为数据库连接本身的问题，而是对 Mongoose 连接事件处理机制的误解。connection.once(‘connected’, …) 仅在连接首次建立时触发。如果连接已经处于就绪状态，该回调将永远不会执行。通过直接在连接就绪后执行数据库操作，或使用 await mongoose.connect() 确保连接建立，可以有效解决此类问题，并提高 Mongoose 应用程序的健壮性和可靠性。理解 Mongoose 连接的生命周期和事件模型，是编写高效、无错数据库交互代码的关键。
暂无评论内容