MongoDB索引优化：加速关联文档计数字段查询与聚合性能

一、更高效的聚合实现方式 #

你当前用$lookup+$size的方案本质是对每个父文档单独发起一次子文档查询，父文档数量大时会产生大量重复查询，这是性能瓶颈的核心。推荐换成先全局统计子文档分组计数，再关联父文档的方式，将多次查询合并为一次全局扫描：

db.yourCollection.aggregate([
  // 第一步：统计所有父节点对应的子文档数量
  { $match: { parentId: { $exists: true, $ne: null } } },
  { $group: { _id: "$parentDoc", childCount: { $sum: 1 } } },
  // 第二步：关联回父文档，合并计数字段
  { $lookup: {
      from: "yourCollection",
      localField: "_id",
      foreignField: "_id",
      as: "parentDoc"
    }
  },
  { $unwind: "$parentDoc" },
  { $replaceRoot: {
      newRoot: { $mergeObjects: ["$parentDoc", { childCount: "$childCount" }] }
    }
  },
  // 补充无任何子文档的父节点，计数设为0
  { $unionWith: {
      coll: "yourCollection",
      pipeline: [
        { $match: {
            $expr: { $not: { $in: ["$_id", /* 可通过前序阶段输出的_id列表过滤，或用反向匹配 */ ] } }
          }
        },
        { $addFields: { childCount: 0 } }
      ]
    }
  }
])

这种方式仅需两次集合扫描（分组统计+关联），比原方案的N次查询效率提升显著，配合已有的parentId索引，能进一步加速$match和$group阶段。

二、索引优化建议 #

你已创建的parentId_1单键索引足够支撑分组统计场景，若要进一步优化，可以考虑：

• 确认索引处于活跃状态：通过db.yourCollection.getIndexes()验证索引存在，且db.yourCollection.explain().aggregate(...)显示索引被实际调用
• 创建部分覆盖索引：db.yourCollection.createIndex({ parentId: 1 }, { partialFilterExpression: { parentId: { $exists: true } } })，仅包含parentId字段的索引能减少磁盘IO，加速分组统计过程

三、物化字段（计算模式）的一致性保障 #

如果选择在父文档中新增childCount物化字段，可通过以下方式确保数据一致性：

1. 实时操作触发更新 #

利用MongoDB的Change Streams监听集合的增删改操作，自动同步计数：

• 插入子文档：执行db.yourCollection.updateOne({ _id: <parentId> }, { $inc: { childCount: 1 } })
• 删除子文档：执行db.yourCollection.updateOne({ _id: <parentId> }, { $inc: { childCount: -1 } })
• 修改子文档的parentId：在事务中完成两步操作，先给旧父节点减1，再给新父节点加1，避免中间状态不一致

2. 事务保障多操作原子性 #

对于涉及父节点切换的操作（如移动子文档），必须将更新子文档和两次计数调整放在同一个事务中：

const session = db.getMongo().startSession();
session.startTransaction();
try {
  // 更新子文档的parentId
  db.yourCollection.updateOne(
    { _id: <childId> },
    { $set: { parentId: <newParentId> } },
    { session }
  );
  // 旧父节点计数减1
  db.yourCollection.updateOne(
    { _id: <oldParentId> },
    { $inc: { childCount: -1 } },
    { session }
  );
  // 新父节点计数加1
  db.yourCollection.updateOne(
    { _id: <newParentId> },
    { $inc: { childCount: 1 } },
    { session }
  );
  session.commitTransaction();
} catch (e) {
  session.abortTransaction();
  throw e;
} finally {
  session.endSession();
}

3. 定期校验兜底 #

编写定时任务（如每周一次），重新计算所有父文档的子计数，与物化字段对比并修正不一致：

// 生成计数统计
const countMap = new Map();
db.yourCollection.find({ parentId: { $exists: true } }).forEach(doc => {
  const pid = doc.parentId.toString();
  countMap.set(pid, (countMap.get(pid) || 0) + 1);
});
// 批量更新物化字段
countMap.forEach((count, pid) => {
  db.yourCollection.updateOne(
    { _id: ObjectId(pid) },
    { $set: { childCount: count } }
  );
});
// 给无子女的父节点设置0
db.yourCollection.updateMany(
  { childCount: { $exists: false } },
  { $set: { childCount: 0 } }
);

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