如何在Mongoose中基于动态父引用使用$graphLookup

问题背景 #

用于餐厅冰箱物品管理的三个MongoDB集合：

• Refrigerators：存储罐子或食品
• Jars：可嵌套在冰箱/其他罐子中，存储食品或子罐子
• Food Items：存储在罐子或冰箱中

各集合结构：

Refrigerators 集合 #

[{
   "name": "abc",
   "temp": 23
}, ...]

Jars 集合 #

[{
   "name": "jar1",
   "parent": {
      "type": "jar",
      "id": "id_of_parent_jar"
   }
}, {
   "name": "jar2",
   "parent": {
      "type": "refrigerator",
      "id": "id_of_parent_refrigerator"
   }
}, ...]

Food Items 集合 #

[{
   "name": "apple",
   "parent": {
      "type": "jar",
      "id": "id_of_parent_jar"
   }
}, ...]

当前痛点：处理food -> jar -> jar -> fridge这类任意深度的嵌套关联时，$graphLookup无法根据parent.type动态切换关联的from集合，无法自动遍历至顶层冰箱。

解决方案 #

MongoDB原生$graphLookup不支持动态切换from集合，可通过统一虚拟集合或条件嵌套关联实现需求，推荐前者适配任意嵌套深度：

方案一：创建统一层级视图 + $graphLookup #

先将Jars、Refrigerators合并为带类型标识的虚拟视图，再用$graphLookup遍历整个层级：

1. 创建合并视图

db.createView(
  "hierarchy_view",
  "food_items",
  [
    { $addFields: { type: "food_item" } },
    { $unionWith: { 
        coll: "jars", 
        pipeline: [{ $addFields: { type: "jar" } }] 
      }
    },
    { $unionWith: { 
        coll: "refrigerators", 
        pipeline: [{ $addFields: { type: "refrigerator" } }] 
      }
    }
  ]
)

2. 执行关联查询

db.food_items.aggregate([
  { $graphLookup: {
      from: "hierarchy_view",
      startWith: "$parent.id",
      connectFromField: "parent.id",
      connectToField: "_id",
      as: "ancestors",
      // 限制仅遍历罐子和冰箱，避免无效关联
      restrictSearchWithMatch: { $or: [{ type: "jar" }, { type: "refrigerator" }] }
    }
  },
  // 提取顶层冰箱节点
  { $addFields: {
      top_refrigerator: {
        $arrayElemAt: [
          { $filter: {
              input: "$ancestors",
              cond: { $eq: ["$$this.type", "refrigerator"] }
            }
          },
          0
        ]
      }
    }
  }
])

方案二：嵌套$lookup + 条件判断（仅适用于已知最大深度） #

若不想创建视图，可通过多层$lookup结合$cond判断parent.type关联对应集合，但需手动扩展嵌套层级：

db.food_items.aggregate([
  // 关联父级罐子
  { $lookup: {
      from: "jars",
      localField: "parent.id",
      foreignField: "_id",
      as: "parent_jar"
    }
  },
  // 判断父级类型并关联对应集合
  { $addFields: {
      parent_entity: {
        $cond: {
          if: { $eq: ["$parent.type", "refrigerator"] },
          then: {
            $arrayElemAt: [
              { $lookup: {
                  from: "refrigerators",
                  localField: "parent.id",
                  foreignField: "_id",
                  as: "parent_fridge"
                }
              },
              0
            ]
          },
          else: { $arrayElemAt: ["$parent_jar", 0] }
        }
      }
    }
  },
  // 如需处理更深层级，手动添加更多嵌套$lookup逻辑
  ...
])

关键说明 #

• 优先选择方案一，可自动适配任意深度的嵌套结构，无需预判层级
• 合并集合时添加type字段，方便后续筛选和识别不同类型的文档
• restrictSearchWithMatch参数可减少无效遍历，提升查询效率

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