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

17

这是一个关于如何将数组中一系列数据相加的问题,需要匹配另一个元素。我尝试使用2.2聚合框架,并且可能可以通过简单的分组来实现。

所以对于给定的文档集,我想要得到以下输出:

{
    "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查询中完成此操作。


为什么需要在MapReduce中完成这个任务?聚合框架通常比M/R更快,更容易阅读。 - Asya Kamsky
除了 agg fx 还未投入生产之外 :), 对于大多数情况,MR 的速度将更快,因为我可以预缓存输出。当我们没有 MR 缓存版本时,agg fx 解决方案将会派上用场。 - cirrus
2.2已经发布到生产环境,因此agg框架也已进入生产环境。 - Asya Kamsky
是的,但实际上只是在我最后一条评论和你的评论之间的时间内 :) 在像MongoLab和MongoHQ这样的地方,它仍然需要一段时间才能成为标准提供,但是除非$out功能正在工作,否则能够预先计算和存储MR结果仍然是一个要求。 - cirrus
2个回答

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
}

如果您有其他键/值对,那么您可以添加一个仅通过["sales",“profit”]中的k值的匹配。


顺便说一句,回答很好!可惜我不能将两个答案都标为正确。 - cirrus
我认为这正是我所需要的 - 如果我想过滤可能存在的其他字段,并仅返回“sales”和“profit”,那么$match应该如何呈现? - cirrus
$match语法与find()的第一个参数完全相同 - 因此{$match:{"finance.k":{$in:["sales","profit"]}}},您可以在展开之后放置它(实际上您可以在展开之前和之后都放置它 :) - Asya Kamsky

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"操作符中将"sales"替换为"profit"即可。

哦,这里是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 );

非常好的答案,谢谢!MapReduce 很有效。可惜我不能将两个答案都标为正确。 - cirrus

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接