MongoDB聚合$lookup数组匹配：用户与群组关联查询问题

我来帮你搞定这个$lookup的实现！你要的就是在用户集合的聚合管道里，通过用户_id匹配群组的members数组，把用户所属的群组关联出来对吧？直接上可行的代码示例，再给你拆解每一步的逻辑：

db.users.aggregate([
  {
    $lookup: {
      from: "groups",       // 指定要关联的目标集合是groups
      localField: "_id",    // 用户集合中用来匹配的字段：当前用户的ID
      foreignField: "members", // 群组集合中用来匹配的字段：成员ID数组
      as: "user_groups"     // 关联结果会被存到这个字段里，格式是数组
    }
  }
])

关键参数解释 #

• from: "groups"：明确告诉MongoDB要和哪个集合做关联查询
• localField: "_id"：用当前用户文档的_id作为匹配键
• foreignField: "members"：MongoDB会自动检查用户的_id是否存在于某个群组的members数组中，所有匹配成功的群组都会被加入结果数组
• as: "user_groups"：每个用户文档都会新增这个字段，里面是该用户所属的所有群组数据；如果用户没有加入任何群组，这个数组会是空的

进阶优化方案 #

1. 只返回群组的关键字段

如果不需要群组的全部字段（比如只需要群组ID和名称），可以在$lookup后加$project做字段过滤：

db.users.aggregate([
  {
    $lookup: {
      from: "groups",
      localField: "_id",
      foreignField: "members",
      as: "user_groups"
    }
  },
  {
    $project: {
      // 保留用户自身需要的字段
      username: 1,
      email: 1,
      // 只提取群组的_id和name字段
      user_groups: {
        $map: {
          input: "$user_groups",
          as: "group",
          in: {
            _id: "$$group._id",
            name: "$$group.name"
          }
        }
      }
    }
  }
])

2. 过滤掉无所属群组的用户

如果只想保留至少加入了一个群组的用户，可以在$lookup后加$match：

db.users.aggregate([
  {
    $lookup: {
      from: "groups",
      localField: "_id",
      foreignField: "members",
      as: "user_groups"
    }
  },
  {
    $match: {
      $expr: { $gt: [ { $size: "$user_groups" }, 0 ] }
    }
  }
])

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