15

我在 MongoDB 中有一个具有树结构的文档列表，其中使用了具有父引用模式的模型树结构。给定'name'属性，我想要一个返回祖先列表（直到根）的聚合查询。

结构：

{
  '_id': '1',
  'name': 'A',
  'parent': '',
},
{
  '_id': '2',
  'name': 'B',
  'parent': 'A',
},
{
  '_id': '3',
  'name': 'C',
  'parent': 'B',
},
{
  '_id': '4',
  'name': 'D',
  'parent': 'C',
}

聚合结果：（给定，名称='D'）

{
  '_id': '4',
  'name': 'D',
  'ancestors': [{name:'C'}, {name:'B'}, {name:'A'}]
}

Note:我现在无法更改文档结构。会引起很多问题。我看到许多解决方案建议将Model Tree Structures 与 Array of Ancestors 一起使用。但我现在不能使用它。有没有办法使用单个聚合查询通过上述模式来实现它？谢谢

4

2 回答 2

21

从 MongoDB 3.4 开始，我们可以使用聚合框架来做到这一点。

我们管道中的第一个也是最重要的阶段是$graphLookup阶段。$graphLookup允许我们递归匹配“父”和“名称”字段。结果，我们得到了每个“名字”的祖先。

管道的下一个阶段是$match我们简单地选择我们感兴趣的“名称”的阶段。

最后阶段是我们使用数组运算符将表达式应用于“祖先”数组的$addFieldsor阶段。$project$map

当然，我们使用$reverseArray运算符反转我们的数组以获得预期的结果。

db.collection.aggregate(
    [ 
        { "$graphLookup": { 
            "from": "collection", 
            "startWith": "$parent", 
            "connectFromField": "parent", 
            "connectToField": "name", 
            "as": "ancestors"
        }}, 
        { "$match": { "name": "D" } }, 
        { "$addFields": { 
            "ancestors": { 
                "$reverseArray": { 
                    "$map": { 
                        "input": "$ancestors", 
                        "as": "t", 
                        "in": { "name": "$$t.name" }
                    } 
                } 
            }
        }}
    ]
)
于 2016-11-10T13:14:20.017 回答
2

如果您愿意使用客户端 javascript，则可以在 mongo shell 上使用递归来实现：

var pushAncesstors = function (name, doc) {
  if(doc.parent) {
    db.collection.update({name : name}, {$addToSet : {"ancesstors" : {name : doc.parent}}});
    pushAncesstors(name, db.collection.findOne({name : doc.parent}))
  }
}

db.collection.find().forEach(function (doc){
  pushAncesstors(doc.name, doc);
})

这将为您提供所有产品的完整雇佣关系。样本输出：

{ "_id" : "1", "name" : "A", "parent" : "" }
{ "_id" : "2", "name" : "B", "parent" : "A", "ancesstors" : [ { "name" : "A" } ] }
{ "_id" : "3", "name" : "C", "parent" : "B", "ancesstors" : [ { "name" : "B" }, { "name" : "A" } ] }
{ "_id" : "4", "name" : "D", "parent" : "C", "ancesstors" : [ { "name" : "C" }, { "name" : "B" }, { "name" : "A" } ] }

如果您的要求不是更新正确的集合，请将数据插入到不同的集合中并在那里更新。该pushAncesstors功能将更改为：

var pushAncesstors = function (name, doc) {
  if(doc.parent) {
    db.outputColl.save(doc)
    db.outputColl.update({name : name}, {$addToSet : {"ancesstors" : {name : doc.parent}}});
    pushAncesstors(name, db.collection.findOne({name : doc.parent}))
  }
}
于 2017-01-27T17:47:35.507 回答