如何将facet聚合输出到true key:value json？

Question

我编写了一个脚本来聚合一些数据，但是输出不是真正的json。

我试图修改聚合管道的$project部分，但我认为我做得不对。

    pipeline = [
        {
            "$match": {
                "manu": {"$ne": "randomized"},
            }},
        {
            "$match": {
                "rssi": {"$lt": "-65db"}
            }
        },
        {"$sort": {"time": -1}},
        {
            "$group": {"_id": "$mac",
                       "lastSeen": {"$first": "$time"},
                       "firstSeen": {"$last": "$time"},
                       }
        },

        {
            "$project":
            {
                "_id": 1,
                "lastSeen": 1,
                "firstSeen": 1,
                "minutes":
                {
                    "$trunc":
                    {
                        "$divide": [{"$subtract": ["$lastSeen", "$firstSeen"]}, 60000]
                    }
                },

            }
        },

        {
            "$facet": {
                "0-5": [
                    {"$match": {"minutes": {"$gte": 1, "$lte": 5}}},
                    {"$count": "0-5"},
                ],
                "5-10": [
                    {"$match": {"minutes": {"$gte": 5, "$lte": 10}}},
                    {"$count": "5-10"},
                ],
                "10-20": [
                    {"$match": {"minutes": {"$gte": 10, "$lte": 20}}},
                    {"$count": "10-20"},
                ],
            }
        },
        {"$project": {
            "0-5": {"$arrayElemAt": ["$0-5.0-5", 0]},
            "5-10": {"$arrayElemAt": ["$5-10.5-10", 0]},
            "10-20": {"$arrayElemAt": ["$10-20.10-20", 0]},

        }},

        {"$sort": SON([("_id", -1)])}

    ]

    data = list(collection.aggregate(pipeline, allowDiskUse=True))

因此，我基本上以{'0-5': 2914, '5-10': 1384, '10-20': 1295}的形式获得输出--它不能用于迭代。

理想情况下应该是

{'timeframe': '0-5', 'count': 262}

有什么建议吗？

提前谢谢。

Answer 1

您可以尝试在聚合下面(替换当前的$facet和以下阶段)：

db.col.aggregate([{
    "$facet": {
            "0-5": [
                {"$match": {"minutes": {"$gte": 1, "$lte": 5}}},
                {"$count": "total"},
            ],
            "5-10": [
                {"$match": {"minutes": {"$gte": 5, "$lte": 10}}},
                {"$count": "total"},
            ],
            "10-20": [
                {"$match": {"minutes": {"$gte": 10, "$lte": 20}}},
                {"$count": "total"},
            ]
        },
    },
    {
        $project: {
            result: { $objectToArray: "$$ROOT" }
        }
    },
    {
        $unwind: "$result"
    },
    {
        $unwind: "$result.v"
    },
    {
        $project: {
            timeframe: "$result.k",
            count: "$result.v.total"
        }
    }
])

$facet返回包含三个字段(子聚合的结果)的单个文档。您可以使用$objectToArray以k和v字段的形式获取它，然后使用$unwind获取每个键的单个文档。