如何在MongoDB MapReduce中生成累积结果



我刚刚开始使用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
}

相关内容

  • 没有找到相关文章

最新更新