java - 将 EnumSet 转换为整数数组

Question

我有一个 EnumSet 并想将其转换为它的序数值数组。例如：

enum MyEnum { A, B, C; }
EnumSet enumSet = EnumSet.of(MyEnum.A, MyEnum.C);

以及我想要得到的：

[0, 2]

Answer 1

您可以使用流来执行此操作：

 int[] ordinals = enumSet.stream().mapToInt(Enum::ordinal).toArray();

Answer 2

您不应该使用没有该<MyEnum>部​​分的原始类型 EnumSet

EnumSet<MyEnum> enumSet = EnumSet.of(MyEnum.A, MyEnum.C);

Java 8 解决方案：

MyEnum[] values = MyEnum.values();
int[] ordinals = IntStream.range(0, values.length).filter(i -> enumSet.contains(values[i])).toArray();

Java 8 之前的解决方案：

int[] ordinals2 = new int[enumSet.size()];
int index = 0;
for (MyEnum e : enumSet)
    ordinals2[index++] = e.ordinal();

检查它是否有效：

System.out.println(Arrays.toString(ordinals));
System.out.println(Arrays.toString(ordinals2));