Chuyển đổi giá trị số nguyên thành phù hợp với Java Enum

Tôi có một enum như thế này:

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;
    }
}

Bây giờ tôi nhận được một int từ đầu vào bên ngoài và muốn đầu vào phù hợp - ném một ngoại lệ nếu một giá trị không tồn tại là ok, nhưng tốt hơn là tôi nên đặt nó DLT_UNKNOWN trong trường hợp đó.

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

Bạn sẽ cần thực hiện việc này theo cách thủ công, bằng cách thêm aa bản đồ tĩnh trong lớp ánh xạ Số nguyên thành enum, chẳng hạn như

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;
}

Có một phương pháp tĩnh values()mà được ghi nhận, nhưng không phải là nơi mà bạn mong đợi nó: 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];}
}

Về nguyên tắc, bạn có thể sử dụng chỉ values()[i], nhưng có tin đồn rằng values()sẽ tạo ra một bản sao của mảng mỗi khi nó được gọi.

Bạn sẽ phải tạo một phương thức tĩnh mới, nơi bạn lặp lại PcapLinkType.values ​​() và so sánh:

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

Điều đó sẽ tốt nếu nó được gọi là hiếm khi. Nếu nó được gọi thường xuyên, thì hãy xem cách Maptối ưu hóa do những người khác đề xuất.

Bạn có thể làm như thế này để tự động đăng ký tất cả chúng vào một tập hợp để sau đó dễ dàng chuyển đổi các số nguyên sang enum tương ứng. (BTW, việc thêm chúng vào bản đồ trong hàm tạo enum không được phép . Thật tuyệt khi học những điều mới ngay cả sau nhiều năm sử dụng 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);
    }
}

nếu bạn có enum như thế này

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;
    }
}

sau đó bạn có thể sử dụng nó như

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

Như @MeBigFatGuy đã nói, ngoại trừ việc bạn có thể tạo static {...} khối cách sử dụng một vòng lặp qua values()bộ sưu tập:

static {
    for (PcapLinkType type : PcapLinkType.values()) {
        intToTypeMap.put(type.getValue(), type);
    }
}

Tôi biết câu hỏi này đã có từ vài năm trước, nhưng vì Java 8, trong khi chờ đợi, đã mang lại cho chúng tôi Optional, tôi nghĩ tôi sẽ đưa ra một giải pháp bằng cách sử dụng nó (và Streamvà 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));
  }
}

Optionalgiống như null: nó đại diện cho một trường hợp khi không có giá trị (hợp lệ). Nhưng nó là một sự thay thế an toàn hơn cho kiểu nullhoặc một giá trị mặc định, chẳng hạn như DLT_UNKNOWNvì bạn có thể quên kiểm tra nullhoặc DLT_UNKNOWNcác trường hợp. Cả hai đều là PcapLinkTypegiá trị hợp lệ ! Ngược lại, bạn không thể gán Optional<PcapLinkType>giá trị cho biến kiểu PcapLinkType. Optionalkhiến bạn kiểm tra giá trị hợp lệ trước.

Tất nhiên, nếu bạn muốn giữ lại DLT_UNKNOWNđể tương thích ngược hoặc bất kỳ lý do nào khác, bạn vẫn có thể sử dụng Optionalngay cả trong trường hợp đó, sử dụng orElse()để chỉ định nó làm giá trị mặc định:

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);
  }
}

Bạn có thể thêm một phương thức tĩnh trong enum của mình để chấp nhận intmột tham số và trả về a PcapLinkType.

public static PcapLinkType of(int linkType) {

    switch (linkType) {
        case -1: return DLT_UNKNOWN
        case 0: return DLT_NULL;

        //ETC....

        default: return null;

    }
}

Đây là những gì tôi sử dụng:

public enum Quality {ENOUGH,BETTER,BEST;
                     private static final int amount = EnumSet.allOf(Quality.class).size();
                     private static Quality[] val = new Quality[amount];
                     static{ for(Quality q:EnumSet.allOf(Quality.class)){ val[q.ordinal()]=q; } }
                     public static Quality fromInt(int i) { return val[i]; }
                     public Quality next() { return fromInt((ordinal()+1)%amount); }
                    }

static final PcapLinkType[] values  = { DLT_NULL, DLT_EN10MB, DLT_EN3MB, null ...}    

...

public static PcapLinkType  getPcapLinkTypeForInt(int num){    
    try{    
       return values[int];    
    }catch(ArrayIndexOutOfBoundsException e){    
       return DLT_UKNOWN;    
    }    
}    

Không có cách nào để xử lý các kiểu liệt kê dựa trên số nguyên một cách thanh lịch. Bạn có thể nghĩ đến việc sử dụng kiểu liệt kê dựa trên chuỗi thay vì giải pháp của mình. Không phải là một cách ưa thích mọi lúc, nhưng nó vẫn tồn tại.

public enum Port {
  /**
   * The default port for the push server.
   */
  DEFAULT("443"),

  /**
   * The alternative port that can be used to bypass firewall checks
   * made to the default <i>HTTPS</i> port.
   */
  ALTERNATIVE("2197");

  private final String portString;

  Port(final String portString) {
    this.portString = portString;
  }

  /**
   * Returns the port for given {@link Port} enumeration value.
   * @return The port of the push server host.
   */
  public Integer toInteger() {
    return Integer.parseInt(portString);
  }
}