如何在ES6语法下的Node.js模块间正确复用MongoDriver连接

嘿，我明白你遇到的问题了——在ES6模块里复用MongoDB连接时，模块顶部调用getDB()总是拿到undefined，不得不每次在函数内部调用，确实挺麻烦的。让我帮你梳理下问题根源，再给你几个靠谱的解决方案～

问题根源：异步初始化的时机错配 #

你原来的代码里，connectToServer是异步执行的，但你在模块加载阶段（也就是导入后立刻调用getDB()）就试图获取连接，这时候MongoDB的连接还没建立完成，_db自然还是初始的undefined。而且原代码里还忽略了一个细节：MongoClient.connect返回的是客户端实例，不是直接的数据库对象，你需要调用client.db()才能拿到可用的DB实例。

方案一：手动初始化连接（推荐用于Web框架场景） #

我们可以用Promise缓存连接状态，确保在应用启动时先完成连接初始化，之后其他模块就能安全复用连接了。

重构连接管理模块 dbconnections.js #

import { MongoClient } from 'mongodb';
const url = "myurl";
let _db;
let _connectionPromise; // 缓存连接Promise，避免重复发起连接请求

// 初始化连接，返回已就绪的DB实例
export const connectToServer = async () => {
  // 如果已有连接，直接返回
  if (_db) return _db;
  // 如果正在连接中，返回同一个Promise，避免重复连接
  if (_connectionPromise) return _connectionPromise;

  try {
    _connectionPromise = MongoClient.connect(url, {
      useNewUrlParser: true,
      useUnifiedTopology: true
    });
    const client = await _connectionPromise;
    _db = client.db(); // 获取实际的数据库实例
    console.log("MongoDB连接成功");
    return _db;
  } catch (err) {
    console.error("MongoDB连接失败", err);
    throw err;
  }
};

// 获取已就绪的DB实例
export const getDB = () => {
  if (!_db) {
    throw new Error("请先调用 connectToServer 初始化数据库连接");
  }
  return _db;
};

// 断开连接
export const disconnectDB = async () => {
  if (_db) {
    await _db.client.close();
    _db = null;
    _connectionPromise = null;
    console.log("MongoDB已断开连接");
  }
};

在应用入口初始化连接 #

比如你的Express/Koa入口文件（app.js）：

import { connectToServer } from './dbconnections.js';
import express from 'express';

const app = express();

// 先初始化DB连接，再启动服务器
async function startServer() {
  try {
    await connectToServer();
    app.listen(3000, () => {
      console.log("服务器启动在3000端口，数据库连接就绪");
    });
  } catch (err) {
    console.error("服务器启动失败：", err);
    process.exit(1);
  }
}

startServer();

其他模块安全复用连接 #

现在你可以在任意模块里直接调用getDB()，不用再担心拿到undefined：

import { getDB } from '../dbconnections.js';

export const someFunction = async (req, res) => {
  try {
    const db = getDB();
    const user = await db.collection('users').findOne({ id: req.params.id });
    res.json(user);
  } catch (err) {
    res.status(500).json({ error: err.message });
  }
};

方案二：自动初始化连接（适合无明确入口的场景） #

如果你的项目没有统一的启动入口，或者想让连接模块自动完成初始化，可以用立即执行异步函数（IIFE）来处理，使用时只需要await getDB()即可。

修改后的dbconnections.js #

import { MongoClient } from 'mongodb';
const url = "myurl";
let _db;

// 自动执行初始化连接
const initConnection = (async () => {
  try {
    const client = await MongoClient.connect(url, {
      useNewUrlParser: true,
      useUnifiedTopology: true
    });
    _db = client.db();
    console.log("MongoDB连接成功");
  } catch (err) {
    console.error("MongoDB连接失败", err);
    throw err;
  }
})();

// 获取DB实例，自动等待连接初始化完成
export const getDB = async () => {
  await initConnection;
  return _db;
};

export const disconnectDB = async () => {
  if (_db) {
    await _db.client.close();
    _db = null;
    console.log("MongoDB已断开连接");
  }
};

其他模块使用方式 #

这种方式下，调用getDB()时需要加await，确保连接已就绪：

import { getDB } from '../dbconnections.js';

export const someFunction = async (req, res) => {
  try {
    const db = await getDB(); // 等待连接初始化完成
    const user = await db.collection('users').findOne({ id: req.params.id });
    res.json(user);
  } catch (err) {
    res.status(500).json({ error: err.message });
  }
};

关键注意点 #

• 永远不要在模块加载阶段直接调用getDB()，因为异步操作还没完成
• 用Promise缓存连接状态，避免同一时间多次发起连接请求
• 记得从MongoClient实例中获取client.db()，而不是直接把client赋值给_db
• 断开连接时要清理缓存的状态，方便后续重新连接

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