MongoDB 2.1聚合框架与名称匹配的数组元素的总和

时间:2012-08-28 15:32:48

标签: mongodb

这是一个关于在数组中添加一系列数据的最佳方法的问题,我必须匹配另一个元素。我正在尝试使用2.2 Aggregation框架,我可以用一个简单的组来完成这个。

因此,对于给定的一组文档,我试图获得这样的输出;

{
    "result" : [
            {
                "_id" : null,
                "numberOf": 2,
                "Sales" : 468000,
                "profit" : 246246,
            }
    ],
    "ok" : 1
}

现在,我最初有一个文档列表,其中包含分配给命名属性的值,例如;

[
{
    _id : 1,
    finance: {
        sales: 234000,
        profit: 123123,
    }
}
,
{
    _id : 2,
    finance: {
        sales: 234000,
        profit: 123123,
    }
}
]

这很容易加起来,但结构因其他原因不起作用。例如,可能有其他列,如“财务”,我希望能够索引它们而不创建数千个索引,所以我需要转换为这样的结构;

[
{
    _id : 1,
    finance: [
        {
            "k": "sales",
            "v": {
                "description":"sales over the year",
                v: 234000,
            }
        },
        {
            "k": "profit",
            "v": {
                "description":"money made from sales",
                v: 123123,
            }
        }
    ]
}
,
{
    _id : 2,
    finance: [
        {
            "k": "sales",
            "v": {
                "description":"sales over the year",
                v: 234000,
            }
        },
        {
            "k": "profit",
            "v": {
                "description": "money made from sales",
                v: 123123,
            }
        }
    ]
}
]

如果我愿意,我可以为finance.k编制索引,但是我正在努力构建一个聚合查询来添加与特定键匹配的所有数字。这就是我最初选择命名属性的原因,但这确实需要在有数千个“k”标签的情况下工作。

有没有人知道如何使用新框架为此构建聚合查询?我试过这个;

db.projects.aggregate([
    {
        $match: {
            // QUERY
            $and: [
                // main query
                {},
            ]
        }
    },
    {
        $group: {
            _id: null,
            "numberOf": { $sum: 1 },
            "sales":    { $sum: "$finance.v.v" },
            "profit":   { $sum: "$finance.v.v" },
        }
    },
])

但我得到了;

{
    "errmsg" : "exception: can't convert from BSON type Array to double",
    "code" : 16005,
    "ok" : 0
}

**对于额外的荣誉,我也需要能够在MapReduce查询中执行此操作。

2 个答案:

答案 0 :(得分:18)

您可以使用汇总框架来获取销售利润您在密钥/值对表示中存储的任何其他值。

对于您的示例数据:

var pipeline = [
    {
        "$unwind" : "$finance"
    },
    {
        "$group" : {
            "_id" : "$finance.k",
            "numberOf" : {
                "$sum" : 1
            },
            "total" : {
                "$sum" : "$finance.v.v"
            }
        }
    }
]

R = db.tb.aggregate( pipeline );
printjson(R);
{
        "result" : [
            {
                "_id" : "profit",
                "numberOf" : 2,
                "total" : 246246
            },
            {
                "_id" : "sales",
                "numberOf" : 2,
                "total" : 468000
            }
        ],
        "ok" : 1
}

如果你有额外的k / v对,那么你可以添加一个只能通过[“sales”,“profit”]中的k值的匹配。

答案 1 :(得分:6)

您必须使用'$ unwind'来分解数组中的值,这意味着您无法在单个聚合命令中获得销售额和利润的总和。鉴于此,查询本身很容易:

var pipeline = [ 
        {"$unwind": "$finance" } ,
        {"$match": {"finance.k": "sales"} },
        { $group: 
            { _id: null,
                numberOf: { "$sum": 1 },
                sales: {"$sum": "$finance.v.v" }
            }
        }
    ];

R = db.tb.aggregate( pipeline );
printjson(R);

{
        "result" : [
                {
                        "_id" : null,
                        "numberOf" : 2,
                        "sales" : 236340
                }
        ],
        "ok" : 1
}

您可以为利润运行类似的查询,只需在“$ match”运算符中将“profit”替换为“sales”。

哦,这是map / reduce示例:

map = function() {
    var ret = { sales: 0.0 , profit: 0.0, count: 1 };

    // iterate over 'finance[]' array
    this.finance.forEach( function (i) { 
        if ( i.k == "sales" ) ret.sales =  i.v.v ;
        if ( i.k == "profit" ) ret.profit =  i.v.v ;
    } );

    emit( 1, ret );
}

reduce = function(key, values) {
    var ret = { sales: 0.0 , profit: 0.0, count: 0 };

    values.forEach(function(v) {
        ret.sales += v.sales;
        ret.profit += v.profit;
        ret.count += v.count;
    });

    return ret;
    };
//
// run map/reduce
//
res = SOURCE.mapReduce( map, reduce );