如何将整数数组转换为枚举数组？

Question

我有一个整数数组，需要将其转换为枚举数组。

enum my_enum {
    APPLE, BANANA, ORANGE;
    int getColorAsInt() { ... }
};
int[] foo = some_method_i_cannot_modify();
my_enum[] bar = ??? ( foo );

最简单的方法是什么？

Here 我找到了一种将单个整数转换为枚举值的方法（他们在示例中使用了Color 枚举）：

public static Color convertIntToColor(int iColor) {
    for (Color color : Color.values()) {
        if (color.getColorAsInt() == iColor) {
            return color;
        }
    }
    return null;
}

...但我希望有一种更直接的方法来进行这种转换（否则在我的代码中，首先使用枚举是没有意义的）。

Here 是一个关于将单个整数转换为枚举值的 SO 问题。

Answer 1

这取决于您的 foo 数组中的内容。如果是枚举的ordinals，那么简单的事情就足够了。

enum my_enum { APPLE, BANANA, ORANGE };
int[] foo = {0,2,2,1};
public void test(String[] args) {
    my_enum[] bar = new my_enum[foo.length];
    for (int i = 0; i < foo.length; i++) {
        bar[i] = my_enum.values()[foo[i]];
    }
    System.out.println(Arrays.toString(bar));
}

打印

[苹果、橙子、橙子、香蕉]

streams 等效项类似于：

    my_enum[] es = Arrays.stream(foo)
            .mapToObj(i -> my_enum.values()[i])
            .toArray(my_enum[]::new);

Answer 2

您需要使用循环或流来遍历输入的整数数组并将每个整数映射到Color 实例。试试这个：

int[] array = getArrayOfInts();
Color[] colors = Arrays.stream(array)
            .mapToObj(i -> convertIntToColor(i))
            .filter(Objects::nonNull)
            .toArray(Color[]::new);

Answer 3

这一行代码将返回对象数组，您可以访问枚举 通过 arr[0]

Object[] arr = Arrays.stream(my_enum.values()).map(x->x).toArray();

Answer 4

但我希望有更直接的方法来进行这种转换

你没有选择。因为您有 int 值，并且您想将它们转换为枚举值。
因此，对于每个ints，您都必须找到对应的枚举值，并且无论如何。

您有一种更直接的方法，即在其声明中使用枚举的顺序
但这是一种非常容易出错的方法，它假定 int 值从 0 开始并递增 1。

Answer 5

如果是你的enum，你当然可以先给它映射。

public enum MyEnum {
    APPLES(10), ORANGES(20);

    private int asInt;
    private MyEnum(int val) {
        asInt = val;
    }
    public int getIntValue() { return asInt; }

    static final Map<Integer, MyEnum> mapping = new HashMap();
    static {
        for (MyEnum e : MyEnum.values()) {
            mapping.put(e.getIntValue(), e);
        }
    }
    static MyEnum fromInteger(int val) {
        return mapping.get(val);
    }
}

Answer 6

我会制作一个地图，从那里获取一个映射值。当然使用自己的 id 比序数更好，但要显示一个想法：

enum Color {
    RED, GREEN, BLUE;

    public static Color[] create(int[] ids) {
        Map<Integer, Color> colors = Arrays.stream(Color.values())
                .collect(Collectors.toMap(Enum::ordinal, Function.identity()));
        return Arrays.stream(ids)
                .mapToObj(colors::get)
                .filter(Objects::nonNull)
                .toArray(Color[]::new);
    }
};

public static void main(String[] args) {
    int[] foo = new int[]{0,1,2};
    System.out.println(Arrays.toString(Color.create(foo)));
}

输出

[RED, GREEN, BLUE]

Answer 7

最好使用 Java 处理 List，但有转换工具。也更喜欢使用 Java 约定：MyEnum。

enum MyEnum {

    APPLE, BANANA, ORANGE;

    public static void main(String[] arg) {
        int[] foo = new int[] { 2, 0, 1 };
        MyEnum[] bar = Arrays.stream(foo).boxed().map(
            i->MyEnum.values()[i]
        ).toArray(MyEnum[]::new);

    }


};

使用数组和int 使使用流变得复杂，主要是使用：i->MyEnum.values()[i] 从 int 到 enum 转换 要使用此函数转换，您需要可从流中访问的 map 方法。