MongoDB跨两集合文档引用及填充实现方案问询

要实现单个字段引用两个不同集合的文档，核心是额外加一个字段记录目标集合的名称，配合原有的someId来定位文档，查询时动态根据集合名填充数据。

修改后的Schema代码 #

const mongoose = require('mongoose');
const { Schema, ObjectId } = mongoose;

// 定义两个被引用的集合Schema和Model
const schemaOne = new Schema({
  name: String,
  // 其他字段
});
const ModelOne = mongoose.model('SchemaOne', schemaOne);

const schemaTwo = new Schema({
  name: String,
  // 其他字段
});
const ModelTwo = mongoose.model('SchemaTwo', schemaTwo);

// 主集合Schema：新增refModel字段标记目标集合
const schemaThree = new Schema({
  someId: {
    type: ObjectId,
    required: true
  },
  refModel: {
    type: String,
    required: true,
    enum: ['SchemaOne', 'SchemaTwo'] // 限制只能是指定的两个集合，避免出错
  }
});
const ModelThree = mongoose.model('SchemaThree', schemaThree);

查询时动态填充数据 #

因为Mongoose的populate默认需要指定固定的ref，所以要根据refModel的值来动态选择填充的Model：

单个文档填充

async function getPopulatedDoc(docId) {
  const doc = await ModelThree.findById(docId);
  if (!doc) return null;

  if (doc.refModel === 'SchemaOne') {
    return await ModelThree.findById(docId).populate({ path: 'someId', model: ModelOne });
  } else {
    return await ModelThree.findById(docId).populate({ path: 'someId', model: ModelTwo });
  }
}

批量文档填充（更高效）

如果要查多个文档，直接循环填充会重复查询，建议先按集合分组批量获取数据，再手动映射：

async function getPopulatedDocs() {
  // 先获取所有主文档
  const docs = await ModelThree.find();
  
  // 按refModel分组收集需要填充的ID
  const idGroups = docs.reduce((acc, doc) => {
    acc[doc.refModel] = acc[doc.refModel] || [];
    acc[doc.refModel].push(doc.someId);
    return acc;
  }, {});

  // 批量获取对应集合的文档
  const modelOneDocs = await ModelOne.find({ _id: { $in: idGroups['SchemaOne'] || [] } });
  const modelTwoDocs = await ModelTwo.find({ _id: { $in: idGroups['SchemaTwo'] || [] } });

  // 将填充后的文档映射回原数据
  return docs.map(doc => {
    let populatedItem;
    if (doc.refModel === 'SchemaOne') {
      populatedItem = modelOneDocs.find(item => item._id.toString() === doc.someId.toString());
    } else {
      populatedItem = modelTwoDocs.find(item => item._id.toString() === doc.someId.toString());
    }
    return { ...doc.toObject(), someId: populatedItem };
  });
}

为什么这么做 #

• 只用两个字段（someId+refModel）就实现了多集合引用，避免新增两个独立字段带来的冗余和复杂度
• enum限制确保不会传入无效的集合名称，减少出错概率
• 动态填充逻辑清晰，查询效率也能通过批量操作保证

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