如何将构面聚合输出转换为true key:value json?

时间:2019-02-02 09:44:32

标签: json mongodb aggregation-framework facet

我写了一个脚本来聚合一些数据,但是输出不是真正的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}

有什么建议吗?

谢谢。

1 个答案:

答案 0 :(得分: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使其具有kv字段的形状,然后使用$unwind获取每个键的单个文档。