Spring Data MongoDB计算数组对象amount总和（含空值判断）

我来帮你分析问题并修正代码，你遇到的空数组返回问题主要是字段路径错误、条件处理和结果映射的问题，以下是具体解决方案：

核心问题分析 #

你的代码里有几个关键错误：

• sum("budgets")指定的字段路径不对：unwind展开的是pmv.budgets，单个元素的amount字段路径应该是pmv.budgets.amount
• 用String.class接收聚合结果不合适：聚合返回的是包含总和的文档，应该用Document或自定义DTO类
• 没有处理pmv.budgets不存在/为空的场景：直接match过滤会导致这类文档被排除，无法返回0的结果

方案1：高效计算单个文档的总和（推荐用$reduce） #

MongoDB 3.2+支持$reduce操作，可以直接在文档内计算数组总和，无需unwind和分组，性能更优，同时能处理数组不存在的情况：

// 假设你要匹配特定文档，比如根据_id
AggregationOperation matchDoc = match(Criteria.where("_id").is(new ObjectId("你的文档ID")));

// 用$reduce计算总和，$ifNull处理budgets不存在的情况（替换为空数组）
AggregationOperation projectTotal = project()
    .and(new Document("$reduce",
        new Document("input", new Document("$ifNull", Arrays.asList("$pmv.budgets", Collections.emptyList())))
            .append("initialValue", 0)
            .append("in", new Document("$add", Arrays.asList("$$value", "$$this.amount")))
    )).as("totalAmount")
    .andExclude("_id");

Aggregation aggregation = newAggregation(matchDoc, projectTotal);
AggregationResults<Document> results = mongoTemplate.aggregate(aggregation, "Iniciativas", Document.class);

// 提取结果，无结果时返回0
Integer total = results.getMappedResults().stream()
    .findFirst()
    .map(doc -> doc.getInteger("totalAmount"))
    .orElse(0);

方案2：计算整个集合的所有budgets.amount总和 #

如果需要统计集合中所有文档的budgets.amount总和，可以用unwind+group的方式，同时处理空数组场景：

// 将不存在的pmv.budgets替换为空数组
AggregationOperation handleEmptyBudgets = addFields()
    .field("pmv.budgets")
    .withValue(new Document("$ifNull", Arrays.asList("$pmv.budgets", Collections.emptyList())))
    .build();

// 展开数组，保留空数组的文档（preserveNullAndEmptyArrays=true）
AggregationOperation unwindBudgets = unwind("pmv.budgets", true);

// 全局分组计算总和
AggregationOperation sumTotal = group()
    .sum("pmv.budgets.amount")
    .as("totalAmount");

// 只保留总和字段
AggregationOperation projectResult = project("totalAmount").andExclude("_id");

Aggregation aggregation = newAggregation(handleEmptyBudgets, unwindBudgets, sumTotal, projectResult);
AggregationResults<Document> results = mongoTemplate.aggregate(aggregation, "Iniciativas", Document.class);

Integer total = results.getMappedResults().stream()
    .findFirst()
    .map(doc -> doc.getInteger("totalAmount"))
    .orElse(0);

方案3：修正你原来的代码（针对单个文档） #

如果你想基于自己的代码逻辑修改，调整如下：

// 匹配目标文档，同时允许budgets为空（去掉not().size(0)，后续处理空数组）
AggregationOperation filter = match(Criteria.where("_id").is(new ObjectId("你的文档ID")));

// 处理budgets不存在的情况
AggregationOperation addFields = addFields()
    .field("pmv.budgets")
    .withValue(new Document("$ifNull", Arrays.asList("$pmv.budgets", Collections.emptyList())))
    .build();

AggregationOperation unwind = unwind("pmv.budgets", true);
// 修正sum的字段路径为pmv.budgets.amount，按文档_id分组
AggregationOperation sum = group("_id").sum("pmv.budgets.amount").as("amount");
AggregationOperation project = project("amount").andExclude("_id");

Aggregation aggregation = newAggregation(filter, addFields, unwind, sum, project);
AggregationResults<Document> aggregationa = mongoTemplate.aggregate(aggregation,"Iniciativas",Document.class);

Integer total = aggregationa.getMappedResults().stream()
    .findFirst()
    .map(doc -> doc.getInteger("amount"))
    .orElse(0);

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