本文深入探讨了 java `enumset` 的元素排序机制，阐明其始终遵循枚举常量的自然声明顺序，而非插入顺序。针对需要保留插入顺序的场景，推荐使用 `list`；而对于需要在 `set` 结构中实现自定义排序的情况，文章详细介绍了如何通过为枚举添加排序属性，并结合 `treeset` 与自定义 `comparator` 来实现灵活的排序控制。
理解 EnumSet 的默认排序行为
在 Java 中，EnumSet 是一个高效的 Set 实现，专门用于枚举类型。它的一个关键特性是其元素总是按照枚举类型中常量的 自然声明顺序 进行存储和迭代，而与这些元素被添加到 EnumSet 中的 插入顺序 无关。这意味着，无论您在 EnumSet.of() 方法中如何排列枚举常量，最终迭代时它们的顺序都将是枚举定义时的顺序。
考虑以下 DayOfWeek 枚举（假设其定义顺序为 THURSDAY, FRIDAY, SATURDAY）：
import java.time.DayOfWeek;
import java.util.EnumSet;
import java.util.Set;
public class EnumSetOrderExample {
public static Set<DayOfWeek> getDays() {
// 插入顺序为 SATURDAY, FRIDAY, THURSDAY
return EnumSet.of(
DayOfWeek.SATURDAY,
DayOfWeek.FRIDAY,
DayOfWeek.THURSDAY
);
}
public static void main(String[] args) {
getDays().forEach(System.out::println);
}
}运行上述代码，其输出将是：
THURSDAY FRIDAY SATURDAY
这明确展示了 EnumSet 忽略了 of() 方法中的插入顺序，而是严格按照 DayOfWeek 枚举常量的声明顺序（THURSDAY -> FRIDAY -> SATURDAY）进行输出。
立即学习“Java免费学习笔记（深入）”；
保留插入顺序的替代方案：使用 List
如果您需要集合中的元素严格按照它们被添加时的顺序进行迭代，那么 EnumSet 可能不是最合适的选择，因为 Set 接口本身并不保证元素的顺序。在这种情况下，List 接口是更直接且推荐的替代方案，因为它天生就维护元素的插入顺序。
以下示例演示了如何使用 List 来保留插入顺序：
import java.time.DayOfWeek;
import java.util.List;
public class ListOrderExample {
public static List<DayOfWeek> getDaysInInsertionOrder() {
// 使用 List.of() 保留插入顺序
return List.of(
DayOfWeek.SATURDAY,
DayOfWeek.FRIDAY,
DayOfWeek.THURSDAY
);
}
public static void main(String[] args) {
getDaysInInsertionOrder().forEach(System.out::println);
}
}运行上述代码，其输出将是：
SATURDAY FRIDAY THURSDAY
这与 List.of() 中定义的顺序完全一致，满足了保留插入顺序的需求。
实现自定义排序：结合 TreeSet 与 Comparator
在某些场景下，您可能既需要 Set 的去重特性，又需要按照除了枚举自然顺序或插入顺序之外的 自定义逻辑 来排序元素。在这种情况下，可以结合使用 TreeSet 和自定义 Comparator。TreeSet 能够根据提供的 Comparator 对元素进行排序。
首先，您需要在枚举类型中引入一个用于排序的属性。例如，为枚举常量添加一个 order 字段：
public enum MyEnum {
ONE(3),
TWO(2),
THREE(1);
private final int order;
MyEnum(int order) {
this.order = order;
}
public int getOrder() {
return order;
}
}接下来，您可以使用 Stream API 将枚举常量收集到一个 TreeSet 中，并提供一个基于 order 属性的 Comparator：
import java.util.Comparator;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class CustomEnumOrderExample {
public static Set<MyEnum> getMyEnumsCustomOrdered() {
return Stream.of(MyEnum.ONE, MyEnum.TWO, MyEnum.THREE)
.collect(Collectors.toCollection(
() -> new TreeSet<>(Comparator.comparing(MyEnum::getOrder))
));
}
public static void main(String[] args) {
getMyEnumsCustomOrdered().forEach(System.out::println);
}
}运行上述代码，其输出将是：
THREE TWO ONE
这表明 TreeSet 成功地根据 MyEnum 中定义的 order 属性（1, 2, 3）进行了升序排序，从而实现了自定义的元素顺序。
总结与注意事项
- EnumSet: 始终按照枚举常量的 自然声明顺序 存储和迭代元素，与 of() 方法中的插入顺序无关。它是一个高性能的 Set 实现，适用于只需要去重和自然顺序的场景。
- List: 如果您需要严格保留元素的 插入顺序，List 是最直接和推荐的选择。它不提供 Set 的去重功能。
- TreeSet + Comparator: 当您需要一个 Set 的去重特性，并且需要根据 自定义逻辑 对元素进行排序时，TreeSet 结合自定义 Comparator 是一个强大的解决方案。这通常要求您在枚举类型中添加额外的排序属性。
在选择合适的集合类型时，请根据您的具体需求（是否需要去重、是否需要保留插入顺序、是否需要自定义排序）进行权衡。理解每种集合类型的排序行为对于编写健壮和可预测的 Java 应用程序至关重要。
