如何在Mongoose中引用并填充同一集合内的子文档？

我太懂你这个痛点了——Mongoose自带的populate()本来是用来处理跨集合关联的，碰到同一集合里两个子文档数组互相引用的场景，直接用它完全行不通。不过别慌，咱们可以通过自定义逻辑或者虚拟字段来搞定这个需求。

先把你的Schema补全，方便后续说明：

import mongoose from 'mongoose';

const ContactSchema = new mongoose.Schema({
  name: String,
  email: String,
  phone: String
});

const LocationSchema = new mongoose.Schema({
  city: String,
  address: String,
  // 存同文档contacts子文档的ID
  contact: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Contact' // 这里的ref只是标识，Mongoose不会自动识别同集合子文档
  }
});

const MainSchema = new mongoose.Schema({
  title: String,
  contacts: [ContactSchema],
  locations: [LocationSchema]
});

const MainModel = mongoose.model('Main', MainSchema);

方法一：用查询后钩子自动填充 #

我们可以给Schema加个post find钩子，在查询完成后手动匹配关联的子文档：

// 给MainSchema添加查询后钩子，覆盖find、findOne、findById这些方法
MainSchema.post(['find', 'findOne'], function(docs) {
  // 处理单个文档的情况
  const targetDocs = Array.isArray(docs) ? docs : [docs];
  
  targetDocs.forEach(doc => {
    // 把contacts转成以_id为键的映射表，方便快速查找
    const contactMap = new Map(
      doc.contacts.map(contact => [contact._id.toString(), contact])
    );
    
    // 遍历locations，匹配对应的contact
    doc.locations.forEach(location => {
      location._doc.contact = contactMap.get(location.contact.toString()) || null;
    });
  });
});

之后查询的时候就会自动填充了：

const mainDoc = await MainModel.findById('主文档ID');
// mainDoc.locations[0].contact 就是对应的contacts子文档

方法二：定义虚拟字段按需填充 #

如果你不想全局自动填充，也可以给LocationSchema加个虚拟字段，手动触发填充：

LocationSchema.virtual('fullContact', {
  ref: 'Main',
  localField: 'contact',
  foreignField: 'contacts._id',
  justOne: true
});

// 别忘了开启虚拟字段的序列化
MainSchema.set('toObject', { virtuals: true });
MainSchema.set('toJSON', { virtuals: true });

查询时手动指定填充即可：

const mainDoc = await MainModel.findById('主文档ID').populate({
  path: 'locations.fullContact',
  select: 'name email phone' // 只返回需要的字段
});
// mainDoc.locations[0].fullContact 就是关联的子文档

关键注意点 #

• 存location.contact的时候，一定要确保值是同文档中contacts子文档的_id，比如创建时可以这样写：

  const newContact = { name: '张三', email: 'zhangsan@test.com' };
  const newMain = new MainModel({
    title: '测试文档',
    contacts: [newContact],
    locations: [{
      city: '北京',
      contact: newContact._id // 直接用刚创建的子文档ID
    }]
  });
  await newMain.save();
  

• 钩子方法会覆盖所有查询类方法，如果有特殊场景，可以调整钩子的触发时机（比如只针对findById）。

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