能否在MongoDB聚合中直接计算$reduce返回数组的$size,无需额外$project?
MongoDB聚合优化:直接计算去重数组长度
完全可以不用额外的$project阶段,直接在$addFields里计算去重后数组的长度,甚至能省去中间存储去重数组的步骤,一步到位得到结果。
优化后的聚合代码
Exercise.aggregate([ {$match: {}}, { $group: { _id: '$userId', correct: {$push: '$stats.correctWords'} } }, { $addFields: { correctsCount: { $size: { $reduce: { input: '$correct', initialValue: [], in: { $setUnion: ['$$value', '$$this'] } } } } } }, // 可选:调整字段名,把_id转为userId,同时移除不需要的correct字段 { $project: { _id: 0, userId: '$_id', correctsCount: 1 } } ])
说明
- 直接将
$reduce返回的去重数组作为$size的输入,省去了单独存储correctItems字段的环节 - 最后的
$project是可选操作:如果需要把聚合结果里的_id显示为userId,同时移除中间生成的correct字段,就加上这一步;如果不需要调整字段,这一步可以省略 - 用示例数据测试的话,输出结果和你期望的完全一致:
[{userId: 'a', correctsCount: 4}, {userId: 'b', correctsCount: 3}]
内容的提问来源于stack exchange,提问作者Roman Nozhenko
相关产品推荐
相关产品推荐

