假设我们的集合中有以下文档:
{ "_id" : ObjectId("5759658e654456bf4a014d01"), "a" : [ 1, 3, 9, 2, 9, 0 ] }
{ "_id" : ObjectId("5759658e654456bf4a014d02"), "a" : [ 0, 8, 1 ] }
{ "_id" : ObjectId("5759658e654456bf4a014d03"), "a" : [ 0, 8, 432, 9, 34, -3 ] }
{ "_id" : ObjectId("5759658e654456bf4a014d04"), "a" : [ 0, 0, 4, 3, 2, 7 ] }
以及下面的输入数组和n = 2
var inputArray = [1, 3, 0];
我们可以使用聚合框架返回数组字段至少包含给定数组的 n 个元素的文档。
$match 只选择数组长度大于或等于n 的文档。这减少了管道中要处理的数据量。
$redact 管道运算符使用逻辑条件处理,使用 $cond 运算符和特殊操作 $$KEEP 来“保留”逻辑条件为真的文档或 $$PRUNE 来“丢弃”其中的文档条件为假。
在我们的例子中,条件是$gte,如果我们使用$setIntersection 运算符计算的两个数组的交集的$size 大于或等于2,则返回true。
db.collection.aggregate(
[
{ "$match": { "a.1": { "$exists": true } } },
{ "$redact": {
"$cond": [
{ "$gte": [
{ "$size": { "$setIntersection": [ "$a", inputArray ] } },
2
]},
"$$KEEP",
"$$PRUNE"
]
}}
]
)
产生:
{ "_id" : ObjectId("5759658e654456bf4a014d01"), "a" : [ 1, 3, 9, 2, 9, 0 ] }
{ "_id" : ObjectId("5759658e654456bf4a014d02"), "a" : [ 0, 8, 1 ] }
{ "_id" : ObjectId("5759658e654456bf4a014d04"), "a" : [ 0, 0, 4, 3, 2, 7 ] }