本文介绍了如何使用 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 操作符。
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 正确计算数组中整数的总和的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1509865.html
微信扫一扫 
支付宝扫一扫 
