如何在MongoDB中为find查询返回的组文档添加用户计数字段？

要给每个组文档加上关联的用户数量，单纯的find查询或者投影确实做不到——因为这需要跨集合关联并做统计计算，得用MongoDB的聚合框架来实现。下面是具体的解决方案：

聚合查询代码示例 #

db.groups.aggregate([
  // 第一步：关联users集合，获取所有关联当前组的用户列表
  {
    $lookup: {
      from: "users",          // 要关联的集合名
      localField: "name",     // groups集合中用来匹配的字段（组名）
      foreignField: "groups", // users集合中用来匹配的字段（用户所属组的数组）
      as: "matching_users"    // 存储匹配结果的临时字段名
    }
  },
  // 第二步：新增user_count字段，计算匹配到的用户数量
  {
    $addFields: {
      user_count: { $size: "$matching_users" } // $size用来获取数组长度，即用户数
    }
  },
  // 第三步：清理结果，只保留需要的字段
  {
    $project: {
      name: 1,
      desc: 1,
      color: 1,
      user_count: 1,
      _id: 0 // 如果不需要默认的_id字段，可以加上这行，和你的预期结果完全一致
    }
  }
])

代码解释 #

1. $lookup阶段：这是MongoDB的左连接操作，会把每个组对应的所有关联用户都放到matching_users数组里。比如Finance组会匹配到Joe和Jane两个用户，所以matching_users数组长度是2。
2. $addFields阶段：用$size操作符计算matching_users数组的长度，把结果赋值给user_count字段，这样就得到了该组的用户数量。
3. $project阶段：只保留你需要的字段（name、desc、color、user_count），如果不需要默认的_id字段，可以加上_id: 0来移除它，让最终结果和你预期的完全一致。

为什么不能用find投影？ #

find的投影只能处理当前集合（groups）内的字段，没法跨集合去users里查询并统计数量，所以必须借助聚合框架的多阶段处理能力来完成这个需求。

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