将整数值转换为匹配的Java Enum

Question

我有这样的枚举:

public enum PcapLinkType {
  DLT_NULL(0)
  DLT_EN10MB(1)
  DLT_EN3MB(2),
  DLT_AX25(3),
  /*snip, 200 more enums, not always consecutive.*/
  DLT_UNKNOWN(-1);
    private final int value;   

    PcapLinkType(int value) {
        this.value= value;
    }
}

现在我从外部输入得到一个int并想要匹配的输入 - 如果一个值不存在就抛出一个异常是好的,但最好是DLT_UNKNOWN 在这种情况下我会有.

int val = in.readInt();
PcapLinkType type = ???; /*convert val to a PcapLinkType */

Answer 1

您需要手动执行此操作,方法是在将Integers映射到枚举的类中添加静态映射,例如

private static final Map<Integer, PcapLinkType> intToTypeMap = new HashMap<Integer, PcapLinkType>();
static {
    for (PcapLinkType type : PcapLinkType.values()) {
        intToTypeMap.put(type.value, type);
    }
}

public static PcapLinkType fromInt(int i) {
    PcapLinkType type = intToTypeMap.get(Integer.valueOf(i));
    if (type == null) 
        return PcapLinkType.DLT_UNKNOWN;
    return type;
}

Answer 2

有一个静态方法values(),其被记录,但不是你希望它:http://docs.oracle.com/javase/tutorial/java/javaOO/enum.html

enum MyEnum {
    FIRST, SECOND, THIRD;
    private static MyEnum[] allValues = values();
    public static MyEnum fromOrdinal(int n) {return allValues[n];}
}

原则上,您可以使用just values()[i],但有传言说values()每次调用时都会创建数组的副本.

Answer 3

您必须创建一个新的静态方法,迭代PcapLinkType.values()并比较:

public static PcapLinkType forCode(int code) {
    for (PcapLinkType typ? : PcapLinkType.values()) {
        if (type.getValue() == code) {
            return type;
        }
    }
    return null;
 }

如果它很少被调用,那就好了.如果经常调用它,那么请查看Map其他人建议的优化.

Answer 4

您可以执行类似的操作,将它们全部自动注册到一个集合中,然后可以轻松地将整数转换为相应的枚举.(顺便说一下,不允许在枚举构造函数中将它们添加到地图中.即使在使用Java多年后,学习新东西也很好.:)

public enum PcapLinkType {
    DLT_NULL(0),
    DLT_EN10MB(1),
    DLT_EN3MB(2),
    DLT_AX25(3),
    /*snip, 200 more enums, not always consecutive.*/
    DLT_UNKNOWN(-1);

    private static final Map<Integer, PcapLinkType> typesByValue = new HashMap<Integer, PcapLinkType>();

    static {
        for (PcapLinkType type : PcapLinkType.values()) {
            typesByValue.put(type.value, type);
        }
    }

    private final int value;

    private PcapLinkType(int value) {
        this.value = value;
    }

    public static PcapLinkType forValue(int value) {
        return typesByValue.get(value);
    }
}

Answer 5

如果你有这样的枚举

public enum PcapLinkType {
  DLT_NULL(0)
  DLT_EN10MB(1)
  DLT_EN3MB(2),
  DLT_AX25(3),
  DLT_UNKNOWN(-1);

    private final int value;   

    PcapLinkType(int value) {
        this.value= value;
    }
}

然后就可以使用它了

PcapLinkType type = PcapLinkType.values()[1]; /*convert val to a PcapLinkType */

Answer 6

我知道这个问题已经有几年了，但是由于 Java 8 同时给我们带来了Optional，我想我会提供一个使用它的解决方案（Stream和Collectors）：

public enum PcapLinkType {
  DLT_NULL(0),
  DLT_EN3MB(2),
  DLT_AX25(3),
  /*snip, 200 more enums, not always consecutive.*/
  // DLT_UNKNOWN(-1); // <--- NO LONGER NEEDED

  private final int value;
  private PcapLinkType(int value) { this.value = value; }

  private static final Map<Integer, PcapLinkType> map;
  static {
    map = Arrays.stream(values())
        .collect(Collectors.toMap(e -> e.value, e -> e));
  }

  public static Optional<PcapLinkType> fromInt(int value) {
    return Optional.ofNullable(map.get(value));
  }
}

Optional就像null：它代表没有（有效）值的情况。但它是一种更类型安全的替代null或默认值，例如DLT_UNKNOWN因为您可能忘记检查nullorDLT_UNKNOWN情况。它们都是有效值PcapLinkType！相反，您不能Optional<PcapLinkType>为类型为 的变量赋值PcapLinkType。Optional使您首先检查有效值。

当然，如果你想保留是DLT_UNKNOWN为了向后兼容或其他任何原因，Optional即使在这种情况下你仍然可以使用， usingorElse()将其指定为默认值：

public enum PcapLinkType {
  DLT_NULL(0),
  DLT_EN3MB(2),
  DLT_AX25(3),
  /*snip, 200 more enums, not always consecutive.*/
  DLT_UNKNOWN(-1);

  private final int value;
  private PcapLinkType(int value) { this.value = value; }

  private static final Map<Integer, PcapLinkType> map;
  static {
    map = Arrays.stream(values())
        .collect(Collectors.toMap(e -> e.value, e -> e));
  }

  public static PcapLinkType fromInt(int value) {
    return Optional.ofNullable(map.get(value)).orElse(DLT_UNKNOWN);
  }
}