MongoDB Aggregation：使用 $sum 正确计算数组中整数的总和

本文介绍了如何使用 MongoDB 的聚合管道和 $sum 操作符来正确计算嵌套在数组中的整数的总和。重点讲解了在使用 $sum 时可能遇到的问题，并提供了使用 $map 和 $reduce 操作符的两种解决方案，并附带示例代码和 MongoDB Playground 链接，方便读者理解和实践。

在使用 MongoDB 的聚合管道时，经常需要计算数组中元素的总和。$sum 操作符是实现这一功能的关键。然而，当数组嵌套在文档的深层结构中时，直接使用 $sum 可能会导致意外的结果，例如返回 0 而不是实际的总和。本文将深入探讨这个问题，并提供两种有效的解决方案。

问题分析

假设我们有如下结构的文档：

{
  "_id": "2023-04-08",
  "dailyNotes": [
    {
      "therapyDiscipline": "PT",
      "individual": [
        60,
        45,
        30,
        75,
        55
      ],
      "concurrent": [
        69
      ],
      "coTreat": [
        67
      ]
    },
    {
      "therapyDiscipline": "YT",
      "individual": [
        2,
        4
      ]
    }
  ]
}

我们的目标是计算 dailyNotes 数组中每个元素的 individual 数组的总和，并将其作为新的 individual 字段添加到 dailyNotes 数组的每个元素中。

如果直接使用 $sum 操作符，例如：

{
  $group: {
    _id: '$_id',
    individual: {
      $sum: '$dailyNotes.individual'
    }
  }
}

或者：

{
  $project: {
    individual: {
      $sum: '$dailyNotes.individual'
    }
  }
}

可能会错误地返回 0。这是因为 $sum 操作符直接应用于 dailyNotes.individual 时，MongoDB 无法正确地遍历嵌套数组。

解决方案一：使用 $map

$map 操作符允许我们遍历数组中的每个元素，并对每个元素应用一个表达式。我们可以利用 $map 来遍历 dailyNotes 数组，并对每个元素的 individual 数组使用 $sum 操作符。

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

44

查看详情

db.collection.aggregate([
  {
    $project: {
      _id: 1,
      dailyNotes: {
        $map: {
          input: "$dailyNotes",
          in: { individual: { $sum: "$$this.individual" } }
        }
      }
    }
  }
])

在这个聚合管道中：

• $project 阶段用于选择需要保留的字段，并使用 $map 创建新的 dailyNotes 数组。
• $map 操作符遍历 dailyNotes 数组。
• input: "$dailyNotes" 指定要遍历的数组。
• in: { individual: { $sum: "$$this.individual" } } 定义了应用于每个元素的表达式。$$this 代表当前正在遍历的元素，$$this.individual 则指向当前元素的 individual 数组。$sum: "$$this.individual" 计算 individual 数组的总和，并将结果赋值给新的 individual 字段。

MongoDB Playground 示例

解决方案二：使用 $reduce

$reduce 操作符也允许我们遍历数组，并将其归约为单个值。我们可以利用 $reduce 来遍历 dailyNotes 数组，并将每个元素的 individual 数组的总和添加到结果数组中。

db.collection.aggregate([
  {
    $project: {
      _id: 1,
      dailyNotes: {
        $reduce: {
          input: "$dailyNotes",
          initialValue: [],
          in: { $concatArrays: [ "$$value", [ { individual: { $sum: "$$this.individual" } } ] ] }
        }
      }
    }
  }
])

在这个聚合管道中：

• $project 阶段用于选择需要保留的字段，并使用 $reduce 创建新的 dailyNotes 数组。
• $reduce 操作符遍历 dailyNotes 数组。
• input: "$dailyNotes" 指定要遍历的数组。
• initialValue: [] 定义了初始值，即一个空数组。
• in: { $concatArrays: [ "$$value", [ { individual: { $sum: "$$this.individual" } } ] ] } 定义了应用于每个元素的表达式。$$value 代表当前的结果数组，$$this 代表当前正在遍历的元素。$sum: "$$this.individual" 计算 individual 数组的总和，并将结果封装在一个新的对象 { individual: ... } 中。$concatArrays 将新的对象添加到结果数组中。

MongoDB Playground 示例

总结

当需要计算嵌套在数组中的整数的总和时，直接使用 $sum 操作符可能无法得到正确的结果。使用 $map 或 $reduce 操作符可以有效地解决这个问题。$map 适用于直接转换数组中的每个元素，而 $reduce 适用于将数组归约为单个值。选择哪种方案取决于具体的需求和数据结构。在实际应用中，建议根据数据量和性能要求选择合适的方案。

以上就是MongoDB Aggregation：使用 $sum 正确计算数组中整数的总和的详细内容，更多请关注php中文网其它相关文章！