如何使用MongoDB聚集管道将同一记录分为多组

Question

我有两个收藏。

orgstructure（将其视为树结构）

示例文档：

{
    "id": "org1",
    "nodes": [
        {
            "nodeId": "root",
            "childNodes": ["child1"]
        },
        {
            "nodeId": "child1",
            "childNodes": ["child2"]
        },
        {
            "nodeId": "child2",
            "childNodes": []
        }
    ]
}

活动

示例文档：

[
    {
        "id":"A1",
        "orgUnit": "root"
    },
    {
        "id":"A2",
        "orgUnit": "child1"
    },
    {
        "id":"A3",
        "orgUnit": "child2"
    }
]

现在，我的期望是通过orgunit将其分组也考虑孩子节点。

在这里，我不想进行查找，我需要将一个访问者文档视为输入，以便我可以使用该文档构建某种条件，以至于查询将返回以下结果。

预期的结果

[
    {
        "_id": "root",
        "activities": ["A1","A2","A3"]
    },
    {
        "_id": "child1",
        "activities": ["A2","A3"]
    },
    {
        "_id": "child2",
        "activities": ["A3"]
    }
]

询问类似的汇总查询。

{
    "$group": {
        "_id": {
            "$switch": {
                "branches": [
                    {
                        "case": {"$in": ["$orgUnit",["root","child1","child2"]]},
                        "then": "root"
                    },
                    {
                        "case": {"$in": ["$orgUnit",["child1","child2"]]},
                        "then": "child1"
                    },
                    {
                        "case": {"$in": ["$orgUnit",["child2"]]},
                        "then": "child2"
                    }
                ],
                "default": null
            }
        }
    }
}

因此，我会提前

Answer 1

您将需要2个步骤：

1. 为递归查找创建另一个集合节点。原始orgstructure很难执行$ graphlookup

db.OrgStructure.aggregate([
  {
    "$unwind": "$nodes"
  },
  {
    "$replaceRoot": {
      "newRoot": "$nodes"
    }
  },
  {
    $out: "nodes"
  }
])

2. persir $ graphlookup on nodes集合以获取所有子节点。执行$查找到活动，并进行一些争吵。

db.nodes.aggregate([
  {
    "$graphLookup": {
      "from": "nodes",
      "startWith": "$nodeId",
      "connectFromField": "childNodes",
      "connectToField": "nodeId",
      "as": "nodesLookup"
    }
  },
  {
    "$lookup": {
      "from": "Activity",
      "let": {
        nodeId: "$nodesLookup.nodeId"
      },
      "pipeline": [
        {
          $match: {
            $expr: {
              $in: [
                "$orgUnit",
                "$nodeId"
              ]
            }
          }
        },
        {
          $group: {
            _id: "$id"
          }
        }
      ],
      "as": "activity"
    }
  },
  {
    $project: {
      _id: "$nodeId",
      activities: "$activity._id"
    }
  }
])

这是 mongo Playground 供您参考。