基于用户认证的MongoDB多数据库连接管理方案咨询

问题描述 #

我正在基于Mongo Atlas DB、Node.js和Express.js开发项目，目前所有客户的用户（通过local-passport-js管理）及数据都集中在集群的一个数据库中。现在我想优化架构，为每个客户创建独立数据库来管理其用户与数据。

当前我的数据库连接代码如下：

const MongoUrl = `mongodb+srv://*myuser**:${process.env.password}@testdb.9unfgh9.mongodb.net/?retryWrites=true&w=majority`

mongoose.connect(MongoUrl, 
    { useNewUrlParser: true, useUnifiedTopology: true})
.then(()=> {
    console.log('DB Connection, Successful')
})
.catch(err => {
    console.log('Error connecting DB')
    console.log(err)
})

请问如何创建全局用户集合，以确定每个用户所属的数据库并加载对应数据库？

实现方案 #

1. 搭建全局元数据库与用户集合 #

先在Mongo Atlas集群中创建一个全局元数据库（比如命名为global_db），专门存储全量用户的归属映射信息。在该库下创建users集合，文档结构示例：

{
  _id: ObjectId("..."),
  username: "user123",
  email: "user@client-a.com",
  clientDbName: "client_a_db", // 该用户对应的客户独立数据库名
  passwordHash: "...", // 仅存密码哈希，用于Passport验证
  // 其他通用用户字段（如创建时间、状态等）
}

2. 修改Passport验证逻辑，绑定用户与客户库 #

调整Passport的Local策略，先从全局元数据库查询用户信息，验证通过后将clientDbName存入会话，后续请求可直接调用：

const passport = require('passport');
const LocalStrategy = require('passport-local').Strategy;
const mongoose = require('mongoose');
const bcrypt = require('bcrypt');

// 单独连接全局元数据库
const globalDbConn = mongoose.createConnection(
  `mongodb+srv://*myuser**:${process.env.password}@testdb.9unfgh9.mongodb.net/global_db?retryWrites=true&w=majority`,
  { useNewUrlParser: true, useUnifiedTopology: true }
);

// 定义全局用户模型
const GlobalUser = globalDbConn.model('User', new mongoose.Schema({
  username: { type: String, unique: true },
  email: String,
  clientDbName: String,
  passwordHash: String
}));

// 配置Local策略
passport.use(new LocalStrategy(
  async (username, password, done) => {
    try {
      const user = await GlobalUser.findOne({ username });
      if (!user) return done(null, false, { message: '用户不存在' });

      // 验证密码哈希
      const isMatch = await bcrypt.compare(password, user.passwordHash);
      if (!isMatch) return done(null, false, { message: '密码错误' });

      // 返回带客户库名的用户对象
      return done(null, { ...user.toObject(), clientDbName: user.clientDbName });
    } catch (err) {
      return done(err);
    }
  }
));

// 序列化/反序列化用户，保留客户库名
passport.serializeUser((user, done) => {
  done(null, { id: user._id, clientDbName: user.clientDbName });
});

passport.deserializeUser(async (userData, done) => {
  try {
    const user = await GlobalUser.findById(userData.id);
    done(null, { ...user.toObject(), clientDbName: userData.clientDbName });
  } catch (err) {
    done(err);
  }
});

3. 封装动态连接客户数据库的工具函数 #

创建工具类缓存已建立的连接，避免重复创建连接消耗资源，同时提供动态模型生成能力：

// db-utils.js
const mongoose = require('mongoose');
const connections = {}; // 缓存已连接的数据库实例

// 获取客户数据库连接
async function getClientDbConnection(clientDbName) {
  // 已存在连接则直接返回
  if (connections[clientDbName]) {
    return connections[clientDbName];
  }

  // 新建连接并缓存
  const conn = await mongoose.createConnection(
    `mongodb+srv://*myuser**:${process.env.password}@testdb.9unfgh9.mongodb.net/${clientDbName}?retryWrites=true&w=majority`,
    { useNewUrlParser: true, useUnifiedTopology: true }
  );

  connections[clientDbName] = conn;
  return conn;
}

// 动态生成客户数据库的模型
function getClientModel(clientDbName, modelName, schema) {
  const conn = connections[clientDbName];
  return conn.model(modelName, schema);
}

module.exports = { getClientDbConnection, getClientModel };

4. 在路由中使用动态连接 #

在业务路由里，从会话中获取clientDbName，动态连接对应数据库并操作数据：

const express = require('express');
const router = express.Router();
const { getClientDbConnection, getClientModel } = require('./db-utils');
const mongoose = require('mongoose');

// 示例：获取当前用户的订单数据
router.get('/orders', async (req, res) => {
  // 校验用户会话是否携带客户库名
  if (!req.user?.clientDbName) {
    return res.status(401).send('未授权访问');
  }

  try {
    // 获取客户数据库连接
    await getClientDbConnection(req.user.clientDbName);
    // 动态生成订单模型
    const Order = getClientModel(
      req.user.clientDbName,
      'Order',
      new mongoose.Schema({
        userId: mongoose.Schema.Types.ObjectId,
        product: String,
        amount: Number,
        createTime: Date
      })
    );

    // 查询当前用户的订单
    const orders = await Order.find({ userId: req.user._id });
    res.json(orders);
  } catch (err) {
    res.status(500).send('数据查询失败');
    console.error(err);
  }
});

module.exports = router;

关键注意事项 #

• 连接缓存：用connections对象复用已建立的连接，大幅提升请求处理效率。
• 权限隔离：严格校验会话中的clientDbName，禁止用户跨数据库访问。
• 客户库初始化：新增客户时，自动创建对应数据库及必要的集合、索引。
• 密码安全：全局用户集合仅存储密码哈希，推荐使用bcrypt等工具加密。

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