mongodb - 计算符合给定条件的子文档

我的 Mongo 集合具有以下形式的文档:

{
    ...
    "notifications": [
        {
            "enabled": true,
            "groups": [ "NG1", "NG3" ]
        },
        {
            "enabled": false,
            "groups": []
        }
    ]
}

其中 enabled 是 bool 值，groups 是一个字符串列表。
我需要执行查询来确定 notifications 中有多少条目具有 enabled = true 并在 groups 中包含给定的字符串(例如 NG3 )。

以前，没有后来作为要求引入的 enabled 属性，我的查询很简单

db.collection.find({ "notifications.groups": "NG3" })

我尝试了一些与 $and 运算符的组合，但没有运气，所以欢迎提出任何建议。提前致谢!

最佳答案

建议运行聚合框架管道，该管道在 $filter 管道步骤中结合使用 $size 和 $project 数组运算符。

$filter 运算符将返回一个数组，该数组的元素在数组子集中匹配指定条件。 $size 将简单地返回该过滤数组中的元素数。

因此，总而言之，您可以运行此管道，以便您可以确定通知中有多少条目具有 enabled = true 并在组中包含给定的字符串(例如“NG3”):

var pipeline = [
    { "$match": { "notifications.enabled": true, "notifications.groups": "NG3" } },
    {
        "$project": {
            "numberOfEntries": {
                "$size": {
                    "$filter": {
                        "input": "$notifications",
                        "as": "items",
                        "cond": {
                            "$and": [
                                { "$eq": [ "$$items.enabled", true ] },
                                { "$setIsSubset": [ [ "NG3" ], "$$items.groups" ] }
                            ]
                        }
                    }
                }
            }
        }
    }
];

db.collection.aggregate(pipeline);

以上适用于 MongoDB 版本 3.2.X 和更新版本。但是，对于涵盖 MongoDB 版本 2.6.X up to and including 3.0.X 的解决方案，其他数组运算符(如 $map 、 $setDifference 将是很好的替代运算符)
过滤数组。

考虑使用 $map 运算符来过滤数组，在 $cond 中使用与上述相同的逻辑作为映射表达式。 $setDifference 运算符然后返回一个集合，其中的元素出现在第一个集合中但不在第二个集合中；即相对于第一组执行第二组的相对赞美。在这种情况下，它将通过 enabled 和 groups 属性返回具有与父文档无关的元素的最终通知数组。

var pipeline = [
    { "$match": { "notifications.enabled": true, "notifications.groups": "NG3" } },
    {
        "$project": {
            "numberOfEntries": {
                "$size": {
                    "$setDifference": [
                        {
                            "$map": {
                                "input": "$notifications",
                                "as": "items",
                                "in": {
                                    "$cond": [
                                        { "$and": [
                                            { "$eq": [ "$$items.enabled", true ] },
                                            { "$setIsSubset": [ [ "NG3" ], "$$items.groups" ] }
                                        ] },
                                        "$$items",
                                        false
                                    ]
                                }
                            }
                        },
                        [false]
                    ]
                }
            }
        }
    }
];

db.collection.aggregate(pipeline);

对于没有上述运算符的旧 MongoDB 版本，请考虑使用 $match 、 $unwind 和 $group 运算符来实现相同的目标:

var pipeline = [
    { "$match": { "notifications.enabled": true, "notifications.groups": "NG3" } },
    { "$unwind": "$notifications" },
    { "$match": { "notifications.enabled": true, "notifications.groups": "NG3" } },
    {
        "$group": {
            "_id": "$_id",
            "numberOfEntries": { "$sum": 1 }
        }
    }
];
db.collection.aggregate(pipeline);