聚合查询中由另一个字段分组的两个 int 数组的总和

Question

我是 MongoDB 新手，遇到了一些问题。我的文档包含我应该求和的固定大小的 int 数组。

mongo 可以在一个查询中对两个 int 数组求和吗？在我的例子中，字符串日期。

示例数据：

{d:[1,2], date:"17-01-2020"} {d:[3,4], date:"17-01-2020"} {d:[5,6], date:"18-01-2020"}

我想要的查询结果：

{d:[4, 6], date:"17-01-2020"} {d:[5,6], date:"18-01-2020"}

Answer 1

如果你有两个值的固定大小，那么你可以使用这个：

db.collection.aggregate([
   { $group: { _id: null, data: { $push: "$d" } } },
   {
      $set: {
         d: [
            {$sum:{ $map: { input: "$data", in: { $first: "$$this" } } }},
            {$sum:{ $map: { input: "$data", in: { $last: "$$this" } } }}
         ]
      }
   },
   {$unset: "data"}
])

任何长度都可以使用这个：

db.collection.aggregate([
   { $group: { _id: null, data: { $push: "$d" } } },
   {
      $set: {
         d: {
            $map: {
               input: { $range: [0, { $size: { $first: "$data" } }] },
               as: "idx",
               in: { $sum: { $map: { input: "$data", in: { $arrayElemAt: ["$$this", "$$idx"] } } } }
            }
         }
      }
   }
])

第一个数组决定了所有其他数组的长度。

另一种方法是：

db.collection.aggregate([
  { $unwind: { path: "$d", includeArrayIndex: "idx" } },
  { $group: { _id: "$idx", d: { $sum: "$d" } } },
  { $sort: { _id: 1 } },
  { $group: { _id: null, d: { $push: "$d" } } }
])

或者如果您想添加其他字段：

db.collection.aggregate([
  { $unwind: { path: "$d", includeArrayIndex: "idx" } },
  { $group: { _id: "$idx", d: { $sum: "$d" }, date: { $first: "$date" } } },
  { $sort: { _id: 1 } },
  { $group: { _id: "$date", d: { $push: "$d" } } },
  { $project: { d: 1, date: "$_id" } }
])

即使数组的长度不同，这也有效。