如何使用MongoDB递归查询树结构?

时间:2017-01-09 06:34:51

标签: node.js mongodb recursion

例如树结构为;

[
    {id: 1 , childrenIdList: [2, 3]},
    {id: 2 , childrenIdList: [4, 5]},
    {id: 3 , childrenIdList: []},
    {id: 4 , childrenIdList: [6, 7]},
    {id: 5 , childrenIdList: []},
    {id: 6 , childrenIdList: []},
    {id: 7 , childrenIdList: []}
]

就像;

               1
          2        3
       4    5
    6    7

如何跟踪树从启动叶节点(id = 7)到root(id = 1)?

查找id=7的父级很容易;

db.document.find({childrenList: { $in: [7]}}, {id: 1}).toArray(function(err), result{
  /*result gives 
  {"id" : NumberInt(4)}
  now I should look the parent of id=4, and parent of id=2 as you know.
  */
})

是否可以在mongodb上进行递归查询?我该如何实现它?

2 个答案:

答案 0 :(得分:4)

根据您的使用案例,MongoDB v3.4提供了一个名为aggregation pipeline$graphLookup运算符。聚合运算符能够对集合执行递归搜索。在$graphLookup definition上查看更多信息。

使用上面的文档层次结构和值作为示例,您可以尝试在聚合下面运行:

db.collectionName.aggregate([

                {$unwind:{
                        path:"$childrenIdList", 
                        preserveNullAndEmptyArrays: true}
                  }, 
                {$graphLookup:{
                        from:"collectionName", 
                        startWith:"$_id", 
                        connectFromField:"_id", 
                        connectToField:"childrenIdList", 
                        as:"myparents",  
                        restrictSearchWithMatch: {"_id"}}
                  }, 
                {$match: {"_id": 7 } },
                {$group:{
                        _id:"$_id", 
                        parents:{$addToSet:"$myparents._id"}
                  }}
]);

上面应该返回如下结果:

{ "_id" : 7, "parents" : [ [ 1, 2, 4 ] ] }

话虽如此,如果您有一个大型集合,上述查询可能无法执行,因为您将对每个文档执行$unwind,并且无法使用索引。正如其他人所建议的那样,您应该重新考虑您的文档模型结构。见Data Models Tree Structures。根据您的应用程序逻辑和查询用例进行优化,并遵循灵活的文档架构。

答案 1 :(得分:0)

无法进行递归查询,您必须对其进行编码,以便在代码中递归调用find。或者,您可以重新设计模式以将父ID保存在子级中,或将所有祖先保存为数组或物化路径。

请参阅MongoDB已提供的documentation树结构数据。