MongoDB 聚合框架 - 如何进行多个 $group 查询

Question

我有以下 MongoDB 聚合查询，它查找指定月份内的所有记录，按天对记录进行 $groups，然后返回每天的平均价格。我还想返回整个月的平均价格。我可以通过使用多个 $groups 来做到这一点，如果可以，如何？

        PriceHourly.aggregate([
                { $match: { date: { $gt: start, $lt: end } } },
                { $group: { 
                    _id: "$day", 
                    price: { $avg: '$price' },
                    system_demand: { $avg: '$system_demand'}
                }}
        ], function(err, results){

                results.forEach(function(r) {
                    r.price          = Helpers.round_price(r.price);
                    r.system_demand  = Helpers.round_price(r.system_demand);
                });
                console.log("Results Length: "+results.length, results);
                res.jsonp(results);
        }); // PriceHourly();  

这是我的模型：

// Model
var PriceHourlySchema = new Schema({
    created: {
        type: Date,
        default: Date.now
    },
    day: {
        type: String,
        required: true,
        trim: true
    },
    hour: {
        type: String,
        required: true,
        trim: true
    },
    price: {
        type: Number,
        required: true
    },
    date: {
        type: Date,
        required: true
    }
}, 
{ 
    autoIndex: true 
});

Answer 1

简短的回答是“仅将日期范围扩大到包括一个月中的所有天有什么问题？”因此，您只需更改即可获得结果。

你能“嵌套”分组阶段吗？是的，您可以向管道添加额外的阶段，这就是管道的用途。因此，如果您首先想“平均”每天，然后取一个月中所有日子的平均值，您可以这样形成：

PriceHourly.aggregate([
   { "$match": { 
       "date": { 
           "$gte": new Date("2014-03-01"), "$lt": new Date("2014-04-01")
       }  
   }},
   { "$group": { 
       "_id": "$day", 
       "price": { "$avg": "$price" },
       "system_demand": { "$avg": "$system_demand" }
   }},
   { "$group": { 
       "_id": null, 
       "price": { "$avg": "$price" },
       "system_demand": { "$avg": "$system_demand" }
   }}
])

尽管这可能是相当多余的，因为这可以通过一个单独的 group 语句来完成。

但对这个架构有更长的评论。除了获得平均值或模式打算包含的内容之外，您实际上并没有说明您所做的大部分目的。所以我想描述一些可能有点不同的东西。

假设您有一个集合，其中包含“产品”、“类型”、“当前价格”和“时间戳”作为“价格”“更改”的日期。让我们将集合称为“PriceChange”。因此，每次发生此事件时都会创建一个新文档。

{
    "product": "ABC",
    "type": 2,
    "price": 110,
    "timestamp": ISODate("2014-04-01T00:08:38.360Z")
}

这可能在一小时、一天或任何情况下发生多次变化。

因此，如果您对一个月内每种产品的“平均”价格感兴趣，您可以这样做：

PriceChange.aggregate([
    { "$match": {
        "timestamp": { 
            "$gte": new Date("2014-03-01"), "$lt": new Date("2014-04-01")
        }
    }},
    { "$group": {
        "_id": "$product",
        "price_avg": { "$avg": "$price" }
    }}
])

此外，无需任何其他字段，您就可以获得每月每一天的每种产品的平均价格：

PriceChange.aggregate([
    { "$match": {
        "timestamp": { 
            "$gte": new Date("2014-03-01"), "$lt": new Date("2014-04-01")
        }
    }},
    { "$group": {
        "_id": {
            "day": { "$dayOfMonth": "$timestamp" },
            "product": "$product"
        },
        "price_avg": { "$avg": "$price" }
    }}
])

或者您甚至可以获得全年每个月的最后价格：

PriceChange.aggregate([
    { "$match": {
        "timestamp": { 
            "$gte": new Date("2013-01-01"), "$lt": new Date("2014-01-01")
        }
    }},
    { "$group": {
        "_id": {
            "date": {
                "year": { "$year" : "$timestamp" },
                "month": { "$month": "$timestamp" }
            },
            "product": "$product"
        },
        "price_last": { "$last": "$price" }
    }}
])

所以这些是您可以使用 Date Aggregation Operators 中的构建来实现各种结果的一些事情。这些甚至可以帮助收集这些信息以写入新的“预聚合”集合，以便更快地进行分析。

我想有一种方法可以使用 mapReduce 将“运行”平均值与所有价格结合起来。再次从我的示例中：

PriceHourly.mapReduce(
    function () {
        emit( this.timestamp.getDate(), this.price )
    },
    function (key, values) {
        var sum = 0;
        values.forEach(function(value) {
            sum += value;
        });
        return ( sum / values.length );
    },
    { 
        "query": {
            "timestamp": {
                "$gte": new Date("2014-03-01"), "$lt": new Date("2014-04-01")
            }
        },
        "out": { "inline": 1 },
        "scope": { "running": 0, "counter": 0 },
        "finalize": function(key,value) {
            running += value;
            counter++;
            return { "dayAvg": value, "monthAvg": running / counter };
        }
    }
)

这将返回如下内容：

{
    "results" : [
        {
            "_id" : 1,
            "value" : {
                "dayAvg" : 105,
                "monthAvg" : 105
            }
        },
        {
            "_id" : 2,
            "value" : {
                "dayAvg" : 110,
                "monthAvg" : 107.5
           }
        }
    ],
}

但是，如果您希望看到日期和月份的离散值，那么如果不运行单独的查询，这将是不可能的。