Слияние полей массива в агрегации MongoDB

Question

Можно ли объединить поля массива при использовании структуры агрегации MongoDB? Вот краткое изложение проблема, которую я пытаюсь решить:

входные выборки документов агрегирования:

{ 
    "Category" : 1, 
    "Messages" : ["Msg1", "Msg2"], 
    "Value" : 1 
}, 
{ 
    "Category" : 1, 
    "Messages" : [], 
    "Value" : 10 
}, 
{ 
    "Category" : 1, 
    "Messages" : ["Msg1", "Msg3"], 
    "Value" : 100 
}, 
{ 
    "Category" : 2, 
    "Messages" : ["Msg4"], 
    "Value" : 1000 
}, 
{ 
    "Category" : 2, 
    "Messages" : ["Msg5"], 
    "Value" : 10000 
}, 
{ 
    "Category" : 3, 
    "Messages" : [], 
    "Value" : 100000 
} 

Мы хотим группе «Категория» при подведении итогов «Value» и объединения «Сообщения». Я попробовал этот агрегацию трубопровод:

{group : { 
     _id : "$Category", 
     Value : { $sum : "$Value"}, 
     Messages : {$push : "$Messages"} 
    } 
}, 
{$unwind : "$Messages"}, 
{$unwind : "$Messages"}, 
{$group : { 
     _id : "$_id", 
     Value : {$first : "$Value"}, 
     Messages : {$addToSet : "$Messages"} 
    } 
} 

Результат является:

"result" : [{ 
     "_id" : 1, 
     "Value" : 111, 
     "Messages" : ["Msg3", "Msg2", "Msg1"] 
    }, 
    { 
     "_id" : 2, 
     "Value" : 11000, 
     "Messages" : ["Msg5", "Msg4"] 
    } 
] 

Однако это совершенно упускает категории 3, так как документы, в которых «Категория» является 3 не имеют каких-либо «Сообщения», и они отбрасываются вторым разворачиванием. Мы хотели бы, чтобы в результате также было включено следующее:

{ 
    "_id" : 3, 
    "Value" : 100000, 
    "Messages" : [] 
} 

Есть ли опрятный способ достижения этого по структуре агрегации?

Answer 1

Вот трюк, вы можете использовать, если сообщение гарантированно будет массивом:

> db.messages.find() 
    { "Category" : 1, "Messages" : [ "Msg1", "Msg2" ], "Value" : 1 } 
    { "Category" : 1, "Messages" : [ ], "Value" : 10 } 
    { "Category" : 1, "Messages" : [ "Msg1", "Msg3" ], "Value" : 100 } 
    { "Category" : 2, "Messages" : [ "Msg4" ], "Value" : 1000 } 
    { "Category" : 2, "Messages" : [ "Msg5" ], "Value" : 10000 } 
    { "Category" : 3, "Messages" : [ ], "Value" : 100000 } 

> var group1 = { 
    "$group": { 
     "_id":  "$Category", 
     "Value": { 
      "$sum":  "$Value" 
     }, 
     "Messages": { 
      "$push": "$Messages" 
     } 
    } 
}; 

> var project1 = { 
    "$project": { 
     "Value": 1, 
     "Messages": { 
      "$cond": [ 
       { 
        "$eq": [ 
         "$Messages", 
         [ [ ] ] 
        ] 
       }, 
       [ [ null ] ], 
       "$Messages" 
      ] 
     } 
    } 
}; 

> db.messages.aggregate(group1, project1) 
    { "_id" : 3, "Value" : 100000, "Messages" : [ [ null ] ] } 
    { "_id" : 2, "Value" : 11000, "Messages" : [ [ "Msg4" ], [ "Msg5" ] ] } 
    { "_id" : 1, "Value" : 111, "Messages" : [ [ "Msg1", "Msg2" ], [ ], [ "Msg1", "Msg3" ] ] } 

Теперь расслабьтесь дважды и перегруппировать, чтобы получить один массив Messages.

> var unwind = {"$unwind":"$Messages"}; 

> var group2 = { 
    $group: { 
     "_id":  "$_id", 
     "Value": { 
      "$first":  "$Value" 
     }, 
     "Messages": { 
      "$addToSet": "$Messages" 
     } 
    } 
}; 

> var project2 = { 
    "$project": { 
     "Category": "$_id", 
     "_id":  0, 
     "Value": 1, 
     "Messages": { 
      "$cond": [ 
       { 
        "$eq": [ 
         "$Messages", 
         [ null ] 
        ] 
       }, 
       [ ], 
       "$Messages" 
      ] 
     } 
    } 
}; 

> db.messages.aggregate(group1, project1, unwind, unwind, group2 ,project2) 
    { "Value" : 111, "Messages" : [ "Msg3", "Msg2", "Msg1" ], "Category" : 1 } 
    { "Value" : 11000, "Messages" : [ "Msg5", "Msg4" ], "Category" : 2 } 
    { "Value" : 100000, "Messages" : [ ], "Category" : 3 } 

Answer 2

Вы можете попробовать выполнить запрос агрегации в версии 3.2.

db.messages.aggregate([ 
    {"$group":{"_id":"$Category","Value":{"$sum":"$Value"},"Messages":{"$push":"$Messages"}}}, 
    {"$unwind":{"path":"$Messages","preserveNullAndEmptyArrays":true}}, 
    {"$unwind":{"path":"$Messages","preserveNullAndEmptyArrays":true}}, 
    {"$group":{"_id":"$_id","Value":{"$first":"$Value"},"Messages":{"$addToSet":"$Messages"}}} 
]) 

Для 3.4 версии и за ее пределами

db.messages.aggregate([ 
    {"$group":{"_id":"$Category","Value":{"$sum":"$Value"},"Messages":{"$push":"$Messages"}}}, 
    {"$addFields":{ 
    "Messages":{ 
     "$reduce":{ 
     "input":"$Messages", 
     "initialValue":[], 
     "in":{"$setUnion":["$$value","$$this"]} 
     } 
    } 
    } 
    } 
])