Mongodb聚合中的Sum数组元素

时间:2015-12-06 14:42:57

标签: mongodb mongodb-query aggregation-framework

我在每个文档中都有c数组的以下集合

{
  {
    id: 1,
    k: 2.2,
    type: "dog",
    c: [ {parentId:1, p:2.2}, {parentId:1, p:1.4} ]
  },

  {
    id: 2,
    k: 4.3,
    type:"cat",
    c: [ {parentId:2, p:5.2}, {parentId:2, p:4.5} ]
  }
}
parentId中每个子文档中的

c是包含文档的ID。

我希望按type对所有文档进行分组,并在每个组中知道该组所有数组中k和所有p的总和。

目前,我在小组阶段对k求和,但在应用程序的结果数组中对p求和。我想在DB中对p求和!

这就是我现在所做的事情:

db.myCol.aggregate([

{ 
  $group: {
    _id: { type: '$type'},
    k: {$sum: '$k'}, // sum k values, very easy!
    // p: {$sum: '$c.0.p'} <==== Does not work, too
    c: {$addToSet: '$c'} // add to each group all c arrays of group's members
  }   
}
], function(err, res) {
   // go over c-arrays and sum p values
   var accP = 0; // accumulator for p values
   for ( var i=0; i<res.length; i++ ) {
     var c = res[i].c;
     for (var j=0;j<c.length; j++) {
       var c2 = c[j];
       for ( var k=0; k<c2.length; k++) { // finally got to objects c array
          accP += c2[k].p;
       }
     }
     res[i].c = accP; // replace array with accumulated p value
   }
});

1 个答案:

答案 0 :(得分:3)

您需要先通过&#34;键入&#34;来$group您的文档,使用$sum累加器运算符返回&#34; k&#34;的总和。并使用$push返回&#34; c&#34;的2D数组。现在你需要两个"$unwind"阶段,你可以对#34; c&#34;二维数组。您在管道中的最后一个阶段是另一个$group阶段,您可以在其中计算&#34; p&#34;使用&#34;点符号&#34;

db.collection.aggregate([
    { '$group': {
        '_id': '$type', 
        'k': { '$sum': '$k' }, 'c': { '$push': '$c' } 
    } }, 
    { '$unwind': '$c' }, 
    { '$unwind': '$c' },
    { '$group': { 
        '_id': '$_id', 
        'k': { '$first': '$k' }, 
        'c': { '$sum': '$c.p' }
    }}
])

哪个收益率:

{ "_id" : "dog", "k" : 2.2, "c" : 3.6 }
{ "_id" : "cat", "k" : 4.3, "c" : 9.7 }
  

Starting in version 3.2,以前只能在$group阶段使用的累加器表达式现在也可以在$project阶段使用。

这意味着我们可以利用它并在$project中使用$sum累加器运算符。当然$map运算符返回一个&#34; p&#34;对于每个文件。

db.collection.aggregate([
    { '$project': { 
        'type': 1, 
        'k': 1, 
        'c': { 
            '$sum': {
                '$map': { 
                    'input': '$c', 
                    'as': 'subc', 
                    'in': '$$subc.p'
                }
            }
        }
    }}, 
    { '$group': { 
        '_id': '$type', 
        'k': { '$sum': '$k' }, 
        'c': { '$sum': '$c' }
    }}
])

返回:

{ "_id" : "cat", "k" : 4.3, "c" : 9.7 }
{ "_id" : "dog", "k" : 2.2, "c" : 3.6 }