Spring Data MongoDB：MergeOperation执行聚合后返回整个集合的原因及解决方法

我完全懂你的困扰——用Spring Data MongoDB执行聚合+$merge的写操作，结果aggregate()居然返回了所有经过管道处理的文档，数据量大的时候简直是性能灾难。下面给你两种针对性的解决方案：

方案一：完全不返回任何结果（推荐用于批量写操作） #

问题出在mongoTemplate.aggregate()的默认行为上：它会返回聚合管道所有阶段输出的文档，哪怕最后一步是$merge这种写操作。要避免这个问题，你可以绕开Spring Data的封装，直接用MongoDB原生驱动执行聚合命令，只获取操作的统计结果，而不是返回所有文档。

修改后的代码示例：

// 先构建好你的聚合管道（match、group、merge阶段不变）
Aggregation agg = Aggregation.newAggregation(match, group, merge);

// 获取源集合的原生MongoCollection实例
MongoCollection<Document> collection = mongoTemplate.getCollection("prices");

// 执行聚合并仅获取写入统计信息
AggregateIterable<Document> aggregateResult = collection.aggregate(agg.toPipeline());
Document writeStats = aggregateResult.first();

// 可以从统计文档中提取操作详情，比如修改的文档数
long updatedCount = writeStats.getLong("nUpdated");
long matchedCount = writeStats.getLong("nMatched");

这样执行后，你只会得到一个包含nMatched、nUpdated等统计字段的文档，不会返回所有经过管道的文档，彻底解决性能问题。

你也可以用executeCommand直接执行聚合命令，效果是一样的：

Document aggregateCommand = new Document("aggregate", "prices")
        .append("pipeline", agg.toPipeline())
        .append("cursor", new Document());

Document result = mongoTemplate.executeCommand(aggregateCommand);
// 从result中提取你需要的统计信息

方案二：仅返回被修改的文档 #

如果你确实需要获取被修改的文档，要明确一点：聚合管道中group阶段输出的文档，就是$merge用来更新目标集合的源文档。如果你的场景确实是“仅对单个文档生效”，那group阶段应该只输出一个文档，这时候aggregate()返回它完全没问题。但如果之前误返回了大量文档，大概率是match阶段的过滤条件不够精准，导致group输出了过多结果，这时候要先检查你的match条件是否正确。

如果想获取目标集合中修改后的最终文档，建议先执行聚合写操作（用方案一的方式），再单独查询目标集合：

// 先执行聚合写操作（方案一的代码）
// 然后根据目标文档的_id查询修改后的结果
Item updatedItem = mongoTemplate.findById(targetDocumentId, Item.class, "items");

这样既能避免返回大量文档，又能拿到你需要的最终结果。

内容的提问来源于stack exchange，提问作者IsaacLevon