我刚刚开始使用mongo中的MapReduce,并有一个具有复合键(日期,候选人)的查询,该查询返回与该键相关的值(当天该候选人的总票数)。我真正想要的是每个日期的票数是累积的。也就是说,对于每个日期/候选人键,该值是该日期及之前投给该候选人的所有选票的总和。
下面是我当前的代码:
输入例子:
{
"interaction" : {
"type" : "draft",
"parameters" : {
"value" : [
{
"candidate" : 453510,
"votes" : 2
},
{
"candidate" : 325786,
"votes" : 2
}
]
}
},
"created_at" : 1360796255
}
地图:
var mapFn = function() {
var right = function(str, n) {
return str.substring(str.length, str.length - n);
}
var toDate = function(epochTimeSec) {
var d = new Date(1000 * epochTimeSec);
var yr = d.getYear() + 1900;
var mn = d.getMonth() + 1;
var dt = d.getDate();
return '' + right('0000' + yr, 4) + '-' + right('00' + mn, 2) + '-' + right('00' + dt, 2);
}
for(var i=0; i<this.interaction.parameters.value.length; i++)
{
vote = this.interaction.parameters.value[i];
var creationDate = toDate(this.created_at);
var votedCandidate = vote.candidate;
emit( {date: creationDate, candidate: votedCandidate}, { quantity: candidate.quantity } );
}
}
减少:
var reduceFn = function(key, values) {
var sum = 0;
values.forEach(function(doc) {
sum += doc.quantity;
});
return { quantity: sum };
}
输出:db.collection_mr.drop();
db.collection.mapReduce(mapFn, reduceFn, { out: 'collection_mr' } );
function printData(r) {
print(r._id.date + ", " + r._id.candidate + ", " + r.value.quantity);
}
db.collection_mr.find().forEach(printData);
谢谢你的帮助!
编辑
作为对注释的回应,我包含了一个示例输入。澄清一下-上面的代码正确地返回每个日期的所有投票的总和。但是,我想对它进行调整,以便为每个日期返回在该日期或之前投票的总数。
回答Asya Kamsky,这是在特别的基础上手动运行的,因此需要在输入日期(可能不是当前日期)之前输出所有数据。
使用聚合框架比使用mapReduce更容易解决这个问题。手册节选:
MongoDB聚合框架提供了一种计算方法聚合值,而不必使用map-reduce。而使用映射-规约模式是强大,往往是比必要困难得多的简单吗聚合任务,例如合计或平均字段值。
这个聚合命令返回您期望的结果:
// target date as unix timestamp, example: 2013-02-28T23:59:59)
targetDate = Date.UTC(2013,1,28,23,59,59)/1000; // month -1 !!
db.xx.aggregate([
// select documents created_at up to target date
// note: this match uses an index, if defined on created_at
{ $match: {
"created_at": { $lte: targetDate }
}},
// unwind the "value" array
{ $unwind: "$interaction.parameters.value" },
// group by candidates, sum up votes
{ $group: {
_id: "$interaction.parameters.value.candidate",
votes: { $sum: "$interaction.parameters.value.votes" }
}},
])
结果:{
"result" : [
{
"_id" : 325786,
"votes" : 2
},
{
"_id" : 453510,
"votes" : 2
}
],
"ok" : 1
}