优化MongoDB $lookup查询：合并重复查询获取关联用户详情

你当前通过两次$lookup实现关联，但多数场景下parent_user_id与user_id相同，可通过以下几种方案合并查询，避免额外的关联开销：

方案1：嵌套$lookup单阶段完成关联 #

把两次关联逻辑整合到一个$lookup的pipeline中，通过条件判断复用当前用户的details，减少聚合阶段数量：

db.tasks.aggregate([
  {
    $lookup: {
      from: "users",
      let: { task_user_id: "$user_id" },
      pipeline: [
        // 获取task对应的用户文档
        { $match: { $expr: { $eq: ["$user_id", "$$task_user_id"] } } },
        // 确定目标用户ID（自身或父用户）
        {
          $addFields: {
            target_id: {
              $cond: {
                if: { $eq: ["$parent_user_id", "$user_id"] },
                then: "$user_id",
                else: "$parent_user_id"
              }
            }
          }
        },
        // 根据目标ID获取details
        {
          $lookup: {
            from: "users",
            let: { target_id: "$target_id" },
            pipeline: [
              { $match: { $expr: { $eq: ["$user_id", "$$target_id"] } } },
              { $project: { details: 1, _id: 0 } }
            ],
            as: "target_details"
          }
        },
        { $unwind: "$target_details" },
        { $project: { details: "$target_details.details", _id: 0 } }
      ],
      as: "associated_details"
    }
  },
  { $unwind: "$associated_details" },
  // 将details合并到task文档
  { $addFields: { details: "$associated_details.details" } },
  { $project: { associated_details: 0 } }
])

说明：嵌套的$lookup把两次关联逻辑打包在一个阶段，当parent_user_id等于user_id时，目标ID就是用户自身，不会产生额外的关联开销，效率更高。

方案2：用$graphLookup一次性获取用户层级 #

利用$graphLookup递归查询用户层级（最多1层，即自身和父用户），再通过过滤和条件判断拿到正确的details：

db.tasks.aggregate([
  {
    $graphLookup: {
      from: "users",
      startWith: "$user_id",
      connectFromField: "parent_user_id",
      connectToField: "user_id",
      maxDepth: 1, // 仅查询自身和父用户
      as: "user_hierarchy"
    }
  },
  // 定位目标用户（父用户或自身）
  {
    $addFields: {
      target_user: {
        $arrayElemAt: [
          {
            $filter: {
              input: "$user_hierarchy",
              cond: { $eq: ["$$this.user_id", { $arrayElemAt: ["$user_hierarchy.parent_user_id", 0] }] }
            }
          },
          0
        ]
      }
    }
  },
  // 选择正确的details
  {
    $addFields: {
      details: {
        $cond: {
          if: { $eq: [{ $arrayElemAt: ["$user_hierarchy.user_id", 0] }, { $arrayElemAt: ["$user_hierarchy.parent_user_id", 0] }] },
          then: { $arrayElemAt: ["$user_hierarchy.details", 0] },
          else: "$target_user.details"
        }
      }
    }
  },
  // 清理冗余字段
  { $project: { user_hierarchy: 0, target_user: 0 } }
])

说明：$graphLookup一次返回所需的用户文档集合，parent_user_id等于user_id时，集合里只有当前用户，直接取其details即可，全程仅一次关联操作。

方案3：简化两次$lookup的逻辑 #

如果偏好简洁易懂的写法，可保留两次$lookup，但通过参数优化避免不必要的处理：

db.tasks.aggregate([
  // 获取当前用户
  {
    $lookup: {
      from: "users",
      localField: "user_id",
      foreignField: "user_id",
      as: "current_user"
    }
  },
  { $unwind: "$current_user" },
  // 获取父用户（允许空结果）
  {
    $lookup: {
      from: "users",
      localField: "current_user.parent_user_id",
      foreignField: "user_id",
      as: "parent_user"
    }
  },
  { $unwind: { path: "$parent_user", preserveNullAndEmptyArrays: true } },
  // 选择details
  {
    $addFields: {
      details: {
        $cond: {
          if: { $eq: ["$current_user.user_id", "$current_user.parent_user_id"] },
          then: "$current_user.details",
          else: "$parent_user.details"
        }
      }
    }
  },
  // 清理冗余字段
  { $project: { current_user: 0, parent_user: 0 } }
])

说明：通过preserveNullAndEmptyArrays: true处理parent_user_id等于user_id的场景（此时父用户结果为空），直接复用当前用户的details，写法直观，容易维护。

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