Spring Data MongoDB聚合：如何按枚举ordinal实现自定义排序？

我希望基于枚举属性的ordinal（顺序）对数据集进行排序。假设有如下枚举：

public enum Status {
  IN_PROGRESS,
  RESOLVED,
  NEW
}

我需要数据集按NEW、IN_PROGRESS、RESOLVED的顺序排序，而非枚举名称的字母顺序。目前我已使用聚合操作进行过滤，现需在聚合中实现该自定义排序，代码如下：

Aggregation.newAggregation(Case.class,
            match(filterCriteria),
            sort(generateCustomSort()),
            limit(limit),
            skip(skipCount));

我知道在JPA中可通过@Enumerated注解设置使用枚举ordinal排序，请问Spring Data MongoDB中是否有类似方式？或能否通过投影实现？

1. 投影添加临时排序权重（推荐） #

由于你的目标排序顺序与枚举自身的ordinal不一致（原枚举ordinal为IN_PROGRESS=0、RESOLVED=1、NEW=2，但需要NEW排在最前），可以通过$addFields聚合阶段添加临时排序权重字段，再基于该字段排序：

// 构建聚合流程
Aggregation aggregation = Aggregation.newAggregation(
    match(filterCriteria),
    // 添加临时排序权重字段
    addFields()
        .addFieldWithValue("sortWeight",
            ConditionalOperators.when(Criteria.where("status").is(Status.NEW)).then(0)
                .when(Criteria.where("status").is(Status.IN_PROGRESS)).then(1)
                .when(Criteria.where("status").is(Status.RESOLVED)).then(2)
                .otherwise(3)) // 处理未定义的枚举值
        .build(),
    // 按临时权重字段升序排序
    sort(Sort.by(Sort.Order.asc("sortWeight"))),
    // 可选：移除临时字段，避免返回结果包含该字段
    project().andExclude("sortWeight"),
    limit(limit),
    skip(skipCount)
);

2. 与枚举ordinal一致时的简化排序 #

如果目标排序顺序和枚举自身的ordinal完全匹配，Spring Data MongoDB默认会将枚举以ordinal形式存储（除非通过@Field(targetType = FieldType.STRING)配置为字符串存储），此时直接按枚举字段排序即可：

Sort sort = Sort.by(Sort.Order.asc("status"));
Aggregation aggregation = Aggregation.newAggregation(
    match(filterCriteria),
    sort(sort),
    limit(limit),
    skip(skipCount)
);

3. 全局自定义枚举转换器 #

若希望全局统一按自定义顺序存储枚举的排序权重，可以实现MongoDB的Converter接口，将枚举转换为对应权重值存储，后续查询直接按该字段排序即可。注意这种方式会修改数据存储结构，需谨慎使用：

@Component
public class StatusToIntegerConverter implements Converter<Status, Integer> {
    @Override
    public Integer convert(Status source) {
        return switch (source) {
            case NEW -> 0;
            case IN_PROGRESS -> 1;
            case RESOLVED -> 2;
        };
    }
}

@Component
public class IntegerToStatusConverter implements Converter<Integer, Status> {
    @Override
    public Status convert(Integer source) {
        return switch (source) {
            case 0 -> Status.NEW;
            case 1 -> Status.IN_PROGRESS;
            case 2 -> Status.RESOLVED;
            default -> throw new IllegalArgumentException("无效的状态编码: " + source);
        };
    }
}

配置后，MongoDB中存储的是权重数值而非枚举的ordinal或名称，查询时直接按status字段排序就能得到目标顺序。

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