MongoDB递归查询：获取指定用户的所有嵌套所属分组

问题背景 #

现有users和groups两个MongoDB集合，groups集合的members字段既可以存储用户名，也能嵌套其他分组名。需求是查询指定用户所属的所有分组（包含多层嵌套的上级分组）。当前使用$graphLookup的查询仅能返回用户直接所属的分组（如查询Joe Blogs仅返回contractors），无法获取users、all这类嵌套的上级分组。

集合结构 #

users集合 #

{
  "users": [
    {
      "_id": 1,
      "username": "Fred Smith",
      "level": 12
    },
    {
      "_id": 2,
      "username": "Bob Brown",
      "level": 20
    },
    {
      "_id": 3,
      "username": "Joe Blogs",
      "level": 1
    }
  ]
}

groups集合 #

{
  "groups": [
    {
      "_id": 1,
      "groupname": "admin",
      "members": ["Fred Smith"]
    },
    {
      "_id": 2,
      "groupname": "users",
      "members": ["Fred Smith", "Bob Brown", "contractors"]
    },
    {
      "_id": 3,
      "groupname": "contractors",
      "members": ["Joe Blogs"]
    },
    {
      "_id": 4,
      "groupname": "all",
      "members": ["users"]
    }
  ]
}

错误查询分析 #

原查询的问题在于$graphLookup的connectFromField设置错误：

db.users.aggregate([
  { $match: { "username": "Joe Blogs" } },
  {
    "$graphLookup": {
      "from": "groups",
      "startWith": "$username",
      "connectFromField": "username", // 错误：找到分组后无法用username继续匹配上级分组
      "connectToField": "members",
      "as": "groups",
      "maxDepth": 10
    }
  },
  { $project: { "username": 1, "groups": "$groups.groupname" } }
])

该查询只能找到直接包含用户的分组，无法递归向上查找包含该分组的上级分组，因为connectFromField指定的是username，而后续需要用分组名去匹配其他分组的members字段。

解决方案 #

调整$graphLookup的connectFromField为groupname，让查询在找到用户的直接分组后，能递归用分组名继续查找上级分组：

db.users.aggregate([
  // 匹配目标用户
  {
    $match: {
      "username": "Joe Blogs"
    }
  },
  // 递归查找所有关联分组
  {
    "$graphLookup": {
      "from": "groups",
      "startWith": "$username", // 初始起点：当前用户的用户名
      "connectFromField": "groupname", // 递归时用分组名作为下一轮的匹配值
      "connectToField": "members", // 匹配目标：其他分组的members字段
      "as": "all_groups",
      "maxDepth": 10 // 限制最大递归深度，避免无限循环
    }
  },
  // 格式化输出结果
  {
    $project: {
      "username": 1,
      "all_groups": "$all_groups.groupname" // 仅提取分组名
    }
  }
])

正确返回结果 #

[
  {
    "_id": 3,
    "username": "Joe Blogs",
    "all_groups": ["contractors", "users", "all"]
  }
]

关键说明 #

• connectFromField改为groupname是核心调整：第一次遍历找到包含用户的contractors分组后，后续递归会用contractors这个分组名去匹配其他分组的members字段，从而找到包含它的users分组，再找到包含users的all分组，实现多层嵌套分组的查询。
• 该查询无需拆分两次查询再串联结果，$graphLookup本身支持递归遍历多层关联关系，只要正确配置连接字段即可一次性完成所有层级的查询。

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