MongoDB：针对每个数组位置聚合整数数组

Question

我正在阅读Mongodb有关聚合框架和MapReduce的文档，但仍然不知道如何开始对整数数组的“列”进行聚合。例如，考虑以下文档：

[{ "_id" : "A", "agent" : "006", "score" : [ 1, 0, 0 ], "qv" : [ 1, 0, 1, 0, 1 ] },
 { "_id" : "B", "agent" : "006", "score" : [ 0, 1, 0 ], "qv" : [ 1, 0, 1, 0, 1 ] },
 { "_id" : "C", "agent" : "006", "score" : [ 1, 0, 0 ], "qv" : [ 1, 0, 1, 0, 0 ] },
 { "_id" : "D", "agent" : "007", "score" : [ 1, 0, 0 ], "qv" : [ 1, 0, 1, 0, 0 ] }] 

预期结果应该如下所示：

[
  {"agent": "006", "score": [2, 1, 0], "qv": [3, 0, 3, 0, 2]},
  {"agent": "007", "score": [1, 0, 0], "qv": [1, 0, 1, 0, 0]}
]

聚合框架足以完成这个任务，还是应该使用Mapreduce？

- Ernest

Answer 1

我认为您需要使用MapReduce来编写一个可以访问数组中特定位置的函数。您可以尝试以下代码：

映射函数：

var M = function() { 
    emit( this.agent, { score : this.score, qv : this.qv } )
}

缩减函数：

var R = function(key, values) {
    var result = { score : [0, 0, 0], qv : [0, 0, 0, 0, 0] };
    values.forEach( function(value) {
        for ( var i = 0; i < value.score.length; i ++ ) {
            result.score[i] += parseInt(value.score[i]);
        }
        for ( var i = 0; i < value.qv.length; i ++ ) {
            result.qv[i] += parseInt(value.qv[i]);
        }
    });
    return result;     
}

您可以在集合上运行以下mapReduce函数：

db.foo.mapReduce( M, R, { out : "resultCollection" } )

希望这能给你带来以下期望的结果！

{
"_id" : "006",
"value" : {
    "score" : [2, 1, 0],
    "qv" : [ 3, 0, 3, 0, 2 ]
}
}
{
"_id" : "007",
"value" : {
    "score" : [ 1, 0, 0],
    "qv" : [ 1, 0, 1, 0, 0]
}

}

- Louisa