如何使用MongoDB聚合進行通用集合操作（​​union，intersection，difference）

Question

我遇到了set操作的一些特殊目的實現，但對於一般情況沒有任何作用。執行集合操作的一般情況是什麼（特別是交集，聯合，對稱差異）。這很容易弄清楚在$ where或map reduce中使用javascript，但我想知道如何在聚合中執行此操作以獲得本機性能。

說明這個問題的更好的方法是舉一個例子。說我有一個2列/套的紀錄：

db.colors.insert({ 
    _id: 1, 
    left : ['red', 'green'], 
    right : ['green', 'blue'] 
}); 

我想要找的「左」和「右」的陣列的集，交集和差異性。更妙的是，形象地我想找到：

聯盟 - > [ '紅'， '綠'， '蘭']

路口 - > ['綠色']

對稱差 - > [' 紅」， '藍']

Answer 1

版本2。6+ Only：

從2.6版的MongoDB開始，這變得容易多了。現在，您可以執行以下操作來解決這個問題：

聯盟

db.colors.aggregate([ 
    {'$project': { 
        union:{$setUnion:["$left","$right"]} 
       } 
    } 
]); 

路口

db.colors.aggregate([ 
    {'$project': { 
        int:{$setIntersection:["$left","$right"]} 
       } 
    } 
]); 

相對補

db.colors.aggregate([ 
    {'$project': { 
        diff:{$setDifference:["$left","$right"]} 
       } 
    } 
]); 

對稱差

db.colors.aggregate([ 
    {'$project': { 
        diff:{$setUnion:[{$setDifference:["$left","$right"]}, {$setDifference:["$right","$left"]}]} 
       } 
    } 
]); 

注：有一個ticket請求對稱差被添加爲核心功能，而不必做兩個集不同的工會。

Answer 2

最簡單的這三個使用聚合的是交叉點**。對於一般的情況下，可以使用聚合像這樣做：

交叉路口：

db.colors.aggregate([ 
    {'$unwind' : "$left"}, 
    {'$unwind' : "$right"}, 
    {'$project': { 
        value:"$left", 
        same:{$cond:[{$eq:["$left","$right"]}, 1, 0]} 
       } 
    }, 
    {'$group' : { 
        _id: {id:'$_id', val:'$value'}, 
        doesMatch:{$max:"$same"} 
       } 
    }, 
    {'$match' :{doesMatch:1}}, 
]); 

另外兩個變得有點更靠譜。據我所知，沒有辦法將同一文件中的兩個單獨字段組合在一起。在$ project管道階段有一個$ add，$ combine或$ addToSet會很好，但這不存在。所以我們能做的最好的就是說是否有某些東西相互交叉。

db.colors.aggregate([ 
    {'$unwind' : "$left"}, 
    {'$unwind' : "$right"}, 
    {'$project': { 
        left:"$left", 
        right:'$right', 
        same:{$cond:[{$eq:["$left","$right"]}, 1, 0]} 
       } 
    }, 
    {'$group' : { 
        _id:{id:'$_id', left:'$left'}, 
        right:{'$addToSet':'$right'}, 
        sum: {'$sum':'$same'}, 
       } 
    }, 
    {'$project': { 
        left:{val:"$_id.left",inter:"$sum"}, 
        right:'$right', 
       } 
    }, 
    {'$unwind' : "$right"}, 
    {'$project': { 
        left:"$left", 
        right:'$right', 
        same:{$cond:[{$eq:["$left.val","$right"]}, 1, 0]} 
       } 
    }, 
    {'$group' : { 
        _id:{id:'$_id.id', right:'$right'}, 
        left:{'$addToSet':'$left'}, 
        sum: {'$sum':'$same'}, 
       } 
    }, 
    {'$project': { 
        right:{val:"$_id.right",inter:"$sum"}, 
        left:'$left', 
       } 
    }, 
    {'$unwind' : "$left"}, 
    {'$group' : { 
        _id:'$_id.id', 
        left:{'$addToSet':'$left'}, 
        right: {'$addToSet':'$right'}, 
       } 
    }, 
]); 

上的問題提供的樣本這種聚合會給這樣的結果：

{ 
     "_id" : 1, 
     "left" : [ 
       { 
         "val" : "green", 
         "inter" : 1 
       }, 
       { 
         "val" : "red", 
         "inter" : 0 
       } 
     ], 
     "right" : [ 
       { 
         "val" : "blue", 
         "inter" : 0 
       }, 
       { 
         "val" : "green", 
         "inter" : 1 
       } 
     ] 
} 

從這裏我們可以通過添加得到交點以下我們可以用下面的開始都聚集到聚合：

{'$project': { 
        left:"$left" 
       } 
    }, 
    {'$unwind' : "$left"}, 
    {'$match' : {'left.inter': 1}}, 
    {'$group' : { 
        _id:'$_id', 
        left:{'$addToSet':'$left'}, 
       } 
    }, 

我們可以通過將以下內容添加到基本聚合的末尾來找到差異以及相關補充：

{'$unwind' : "$left"}, 
    {'$match' : {'left.inter': 0}}, 
    {'$unwind' : "$right"}, 
    {'$match' : {'right.inter': 0}}, 
    {'$group' : { 
        _id:'$_id', 
        left:{'$addToSet':'$left'}, 
        right:{'$addToSet':'$right'}, 
       } 
    }, 

遺憾的是，似乎沒有要到來自不同領域對不同的項目結合起來的好方法。爲了得到工會，最好從客戶那裏做到這一點。或者，如果您想過濾，請分別在每一組上進行。