如何从 mongodb 对连接的文档（图的节点）进行分组

Question

假设我有一个带有节点 A、B、C、D、E 的单向图

A->B->C 是连通分量，

D->E 是另一个连通分量。

节点作为文档保存在 MongoDB 中

{name:'A', child: 'B'}, {name:'B', child: 'C'}, {name:'C'},
{name:'D', child: 'E'}, {name:'E'}

如何获取所有连接的组件？

预期结果：2 组

[{name:'A'...},{name:'B'...}, {name:'C'...}],[{name:'D'...}, {name:'E'...}]

Answer 1

使用$graphLookup流水线阶段。

• $group - 从 name 和 child 字段中收集所有可能的值。
• $addFields - 使用根文档的 name 字段值生成数组。
• $unwind - 将 roots 数组拆分为单独的文档。
• $graphLookup - 通过给定的根文档的name 字段值收集相关文档。
• $project - 从结果文档中删除不必要的字段。

查询：

db.getCollection('t').aggregate([
    {
        $group: {
            _id: null,
            names: { $addToSet: "$name" },
            childs: { $addToSet: "$child" }
        }
    },
    {
        $addFields: {
            roots: { $setDifference: ["$names", "$childs"] }
        }
    },
    { $unwind: "$roots" },
    {
        $graphLookup: {
            from: "t",
            startWith: "$roots",
            connectFromField: "child",
            connectToField: "name",
            as: "related"
        }
    },
    {
        $project: {
            "related": 1,
            "_id": 0
        }
    }
])

结果：

/* 1 */
{
    "related" : [ 
        {
            "_id" : ObjectId("5a29316a545eb40950c33bc8"),
            "name" : "C"
        }, 
        {
            "_id" : ObjectId("5a29316a545eb40950c33bc7"),
            "name" : "B",
            "child" : "C"
        }, 
        {
            "_id" : ObjectId("5a29316a545eb40950c33bc6"),
            "name" : "A",
            "child" : "B"
        }
    ]
}

/* 2 */
{
    "related" : [ 
        {
            "_id" : ObjectId("5a29316a545eb40950c33bca"),
            "name" : "E"
        }, 
        {
            "_id" : ObjectId("5a29316a545eb40950c33bc9"),
            "name" : "D",
            "child" : "E"
        }
    ]
}