Node.js中MongoDB多数据库同模型动态处理及请求级连接问询

核心思路 #

不用为每个数据库重复定义模型，基于同一个Schema，在请求到来时动态获取对应数据库的连接，再从该连接生成模型实例即可。

具体实现步骤 #

1. 定义通用Schema #

先写好统一的数据结构，所有数据库的同类型集合都复用这个Schema：

// schemas/userSchema.js
const mongoose = require('mongoose');

const userSchema = new mongoose.Schema({
  name: String,
  email: { type: String, unique: true },
  createdAt: { type: Date, default: Date.now }
});

module.exports = userSchema;

2. 动态管理数据库连接与模型 #

维护一个连接缓存，避免重复创建连接；根据请求指定的数据库名，动态生成对应模型：

// utils/dynamicModel.js
const mongoose = require('mongoose');
const userSchema = require('../schemas/userSchema');

// 缓存已建立的数据库连接
const dbConnections = {};

async function getDbConnection(dbName) {
  // 已存在连接直接返回
  if (dbConnections[dbName]) {
    return dbConnections[dbName];
  }
  // 拼接目标数据库的连接URI，替换dbName即可
  const uri = `mongodb://localhost:27017/${dbName}`;
  const connection = await mongoose.createConnection(uri);
  dbConnections[dbName] = connection;
  return connection;
}

async function getDynamicModel(dbName, modelName, schema) {
  const connection = await getDbConnection(dbName);
  // 连接中已有模型则复用，否则创建新模型
  return connection.models[modelName] || connection.model(modelName, schema);
}

module.exports = { getDynamicModel };

3. 请求级的数据库操作处理 #

在路由中从请求里提取数据库标识（比如请求头、URL参数），再获取对应模型执行操作：

// routes/users.js
const express = require('express');
const { getDynamicModel } = require('../utils/dynamicModel');
const userSchema = require('../schemas/userSchema');
const router = express.Router();

router.get('/users', async (req, res) => {
  // 从请求头X-DB-Name获取目标数据库名
  const dbName = req.headers['x-db-name'];
  if (!dbName) {
    return res.status(400).send('缺少数据库标识');
  }

  try {
    const User = await getDynamicModel(dbName, 'User', userSchema);
    const users = await User.find();
    res.json(users);
  } catch (err) {
    res.status(500).send(err.message);
  }
});

router.post('/users', async (req, res) => {
  const dbName = req.headers['x-db-name'];
  if (!dbName) {
    return res.status(400).send('缺少数据库标识');
  }

  try {
    const User = await getDynamicModel(dbName, 'User', userSchema);
    const newUser = new User(req.body);
    await newUser.save();
    res.status(201).json(newUser);
  } catch (err) {
    res.status(500).send(err.message);
  }
});

module.exports = router;

关键注意事项 #

• 连接缓存：必须缓存已创建的连接，避免每次请求新建连接消耗资源
• 权限校验：要验证请求中的数据库标识是否合法，防止访问未授权数据库
• 连接清理：可定时清理长时间闲置的连接，避免连接数超限
• 事务支持：跨数据库事务需要MongoDB 4.0+，且所有连接需属于同一副本集或分片集群

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