MongoDB聚合查询中_id字段的类型映射问题求解

我来帮你搞定这个聚合分组时_id类型不兼容的问题！核心原因是MongoDB聚合返回的_id类型会随分组字段数量变化：

• 0个分组字段：_id为null
• 1个分组字段：_id是该字段的原始类型（比如Integer、String）
• 多个分组字段：_id是包含所有分组字段的键值对文档（对应Java的Map<String, Object>）

之前用List<Object>肯定不行，因为多字段返回的是键值对而非列表；用Object本身是可行的，但需要在业务代码中根据分组字段数量做类型判断和转换。下面是具体的解决方案：

步骤1：修正Metric类的_id字段定义 #

把_id字段定义为Object，它可以兼容null、单个值、Map三种情况：

public class Metric {
    private Object _id;
    private Integer count;
    private Double profit;
    private Double averageOdd;

    // 生成getter和setter方法
    public Object get_id() {
        return _id;
    }

    public void set_id(Object _id) {
        this._id = _id;
    }

    // 其他字段的getter/setter...
}

步骤2：在业务代码中处理不同场景的_id #

在获取聚合结果后，根据传入的分组字段列表fields的长度，判断_id的实际类型并处理：

public List<Metric> getMetric(List<String> fields) {
    MatchOperation match = match(where("status").in("WIN", "LOSS", "PUSH").and("year").gte(2015));
    GroupOperation group = group(fields.toArray(new String[0]))
            .count().as("count")
            .avg("odd").as("averageOdd")
            .sum("profit").as("profit")
            .push("$$ROOT").as("picks");
    Aggregation aggregation = Aggregation.newAggregation(match, group);

    AggregationResults<Metric> aggregationResults = mongoTemplate.aggregate(aggregation, "pick", Metric.class);
    List<Metric> metrics = aggregationResults.getMappedResults();

    // 遍历处理每个Metric的_id
    for (Metric metric : metrics) {
        Object id = metric.get_id();
        if (fields.isEmpty()) {
            // 空分组场景：id为null，无需额外处理
            System.out.println("空分组，总统计");
        } else if (fields.size() == 1) {
            // 单字段分组：id是该字段的原始值，直接使用
            String fieldName = fields.get(0);
            System.out.printf("单字段[%s]分组，值：%s%n", fieldName, id);
        } else {
            // 多字段分组：id是Map<String, Object>，可以按字段名获取对应值
            Map<String, Object> groupIdMap = (Map<String, Object>) id;
            for (String field : fields) {
                System.out.printf("多字段分组，%s：%s%n", field, groupIdMap.get(field));
            }
        }
    }

    return metrics;
}

可选：更优雅的类型处理（自定义转换器） #

如果不想在业务代码里做类型判断，可以自定义一个MongoDB转换器，自动处理_id的类型转换：

import org.bson.Document;
import org.springframework.core.convert.converter.Converter;
import org.springframework.data.convert.ReadingConverter;

@ReadingConverter
public class MetricReadConverter implements Converter<Document, Metric> {
    @Override
    public Metric convert(Document source) {
        Metric metric = new Metric();
        metric.setCount(source.getInteger("count"));
        metric.setProfit(source.getDouble("profit"));
        metric.setAverageOdd(source.getDouble("averageOdd"));
        
        // 直接赋值_id，自动兼容三种类型
        metric.set_id(source.get("_id"));
        return metric;
    }
}

然后在MongoDB配置类中注册这个转换器：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.convert.MongoCustomConversions;

import java.util.List;

@Configuration
public class MongoConfig {
    @Bean
    public MongoCustomConversions mongoCustomConversions() {
        return new MongoCustomConversions(List.of(new MetricReadConverter()));
    }
}

这样无论分组字段数量多少，_id都能正确映射到Metric类中，你只需要在使用时根据分组字段数量做类型强转即可。

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