Рекурсивный поиск по коллекции в MongoDB

у меня есть список документов в 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',
}

результат агрегации: (задано, name = 'D')

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

Note: Я не могу менять структуру документа. Это вызовет много проблем. Я видел много решений, которые предлагают использовать модели древовидных структур с массивом предков. Но я не могу использовать его сейчас. Есть ли способ достичь этого с помощью вышеуказанного шаблона, используя один запрос агрегации? Спасибо

11
aggregation-framework mongodb tree-structure

2 ответов

начиная с MongoDB 3.4, мы можем сделать это с помощью структуры агрегации.

первый и самый важный этап в нашем трубопроводе $graphLookup этап. $graphLookup позволяет рекурсивно сопоставлять поля" родитель "и" имя". В результате мы получаем предков каждого "имени".

следующим этапом в конвейере является $match этап, где мы просто выбираем" имя", которое нас интересует.

заключительный этап $addFields или $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" }
                    } 
                } 
            }
        }}
    ]
)
15

если вы открыты для использования javascript на стороне клиента, вы можете использовать рекурсию в оболочке mongo для достижения этого:

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);
})

это даст вам полный hirearchy для всех продуктов. Выход образца:

{ "_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" } ] }

если ваше требование не обновить правильную коллекцию, вставьте данные в различную коллекцию и обновите ее. The 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}))
  }
}
2