Mongodb聚合中的Sum数组元素

Question

我在每个文档中都有以下带有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} ]
  }
}

c每个子文档中的parentId是包含文档的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
   }
});

Answer 1

您需要首先按“类型”对文档进行$group ，使用$sum accumulator运算符返回“k”的总和，并使用$push返回“c”的2D数组。 现在你需要两个“$ unwind”阶段，你可以对“c”二维数组进行反规范化。 您在管道中的最后一个阶段是另一个$group阶段，您可以使用“点符号”计算“p”的总和

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 }

---

从版本3.2开始 ，以前只在$group阶段中可用的累加器表达式现在也可以在$project阶段中使用。

这意味着我们可以利用它并在$project使用$sum accumulator运算符。 当然$map运算符为每个文档返回一个“p”数组。

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 }