Как получить общие элементы массива mongodb - PullRequest
Новая
       19

Как получить общие элементы массива mongodb

0 голосов
/ 21 января 2019

Первый документ: 

    {
    "_id" : 1,
    "array" : [ 
        {
            "userID" : 1,
            "name" : "name 1"
        }, 
        {
            "userID" : 2,
            "name" : "name 2"
        }, 
        {
            "userID" : 3,
            "name" : "name 3"
        }, 
        {
            "userID" : 6,
            "name" : "name 6"
        }
    ]
}

Второй документ: 

    {
    "_id" : 2,
    "array" : [ 
        {
            "userID" : 1,
            "name" : "name 1"
        }, 
        {
            "userID" : 2,
            "name" : "name 2"
        }, 
        {
            "userID" : 3,
            "name" : "name 3"
        }, 
        {
            "userID" : 4,
            "name" : "name 4"
        }, 
        {
            "userID" : 5,
            "name" : "name 5"
        }
    ]
}

есть ли способ получить вывод, подобный этому: 

   {    
    "array" : [  
        {  
            "userID" : 1,  
            "name" : "name 1" 
        },  
        {  
            "userID" : 2,  
            "name" : "name 2"  
        },   
        {  
            "userID" : 3,  
            "name" : "name 3"  
        }  
    ]  

    }

Я хочу сравнить поля массива двух документов и извлечь из них общие элементы. И сравнивать поля массива только по общим 'userID'. Можем ли мы сделать это по $setIntersection? Мне нужно проверять только два поля массива одновременно.

1 Ответ

0 голосов
/ 21 января 2019

вы можете использовать $setIntersection

db.t67.aggregate([
    //{$match : {}} add filter criteria to filter documents
    {$group : {_id : null, array : {$push : "$array"}}},
    {$project : {_id :0, array : {$reduce : {input : "$array", initialValue : {$arrayElemAt : ["$array", 0]}, in : {$setIntersection : ["$$this", "$$value"]}}}}}
])

выход 

> db.t67.aggregate([
...     {$group : {_id : null, array : {$push : "$array"}}},
... {$project : {_id :0, array : {$reduce : {input : "$array", initialValue : {$arrayElemAt : ["$array", 0]}, in : {$setIntersection : ["$$this", "$$value"]}}}}}
... ])
{ "array" : [ { "userID" : 1, "name" : "name 1" }, { "userID" : 2, "name" : "name 2" }, { "userID" : 3, "name" : "name 3" } ] }

или $unwind и $group

db.t67.aggregate([
    // $match stages to filter 2 documents
    {$unwind : "$array"},
    {$group : {_id : "$array", count : {$sum : 1}}},
    {$match : {count : {$gt : 1}}},
    {$project : {count : 0}},
    {$group : {_id : null, array : {$push : "$_id"}}},
    {$project : {_id : 0}}
])

выход 

{ "_id" : null, "array" : [ { "userID" : 3, "name" : "name 3" }, { "userID" : 2, "name" : "name 2" }, { "userID" : 1, "name" : "name 1" } ] }
...