你一定需要知道的高階JAVA列舉特性！

程式猿DD發表於2021-01-13
Java

JAVA列舉，比你想象中還要有用！

我經常發現自己在Java中使用列舉來表示某個物件的一組潛在值。

在編譯時確定型別可以具有什麼值的能力是一種強大的能力，它為程式碼提供了結構和意義。

當我第一次瞭解列舉時，當時我認為它們只是一個為常量命名的工具，可以很容易地被靜態常量字串ENUM_VAL_NAME所取代。

後來我發現我錯了。事實證明，Java列舉具有相當高階的特性，可以使程式碼乾淨、不易出錯，功能強大。

讓我們一起來看看Java中的一些高階列舉特性，以及如何利用這些特性使程式碼更簡單、更可讀。

列舉是類！

在Java中，列舉是Object的一個子類。讓我們看看所有列舉的基類，Enum（為簡潔起見進行了修改）。


public abstract class Enum<E extends Enum<E>>
    implements Constable, Comparable<E>, Serializable {
  private final String name;
  
  public final String name() {
      return name;
  }
  
  private final int ordinal;
  
  public final int ordinal() {
      return ordinal;
  }
  
  protected Enum(String name, int ordinal) {
      this.name = name;
      this.ordinal = ordinal;
  }
  
  public String toString() {
      return name;
  }
  
  public final boolean equals(Object other) {
      return this==other;
  }
  
  public final int hashCode() {
      return super.hashCode();
  }
  
  public final int compareTo(E o) {
      Enum<?> other = (Enum<?>)o;
      Enum<E> self = this;
      if (self.getClass() != other.getClass() && // optimization
          self.getDeclaringClass() != other.getDeclaringClass())
          throw new ClassCastException();
      return self.ordinal - other.ordinal;
  }
}

我們可以看到，這基本上只是一個常規的抽象類，有兩個欄位，name和ordinal。

所以說列舉都是類，所以它們具有常規類的許多特性。

我們能夠為列舉提供例項方法、建構函式和欄位。我們可以重寫toString（），但不能重寫hashCode（）或equals（Object other）。

接下來我們看下我們的列舉示例，Operation

  enum Operation {
    ADD,
    SUBTRACT,
    MULTIPLY
  }

這個列舉表示一個Operation可以對兩個值執行，並將生成一個結果。關於如何實現此功能，您最初的想法可能是使用switch語句，如下所示：

  public int apply(Operation operation, int arg1, int arg2) {
    switch(operation) {
      case ADD:
        return arg1 + arg2;
      case SUBTRACT:
        return arg1 - arg2;
      case MULTIPLY:
        return arg1 * arg2;
      default:
        throw new UnsupportedOperationException();
  }
}

當然，這樣子會有一些問題。

第一個問題是，如果我們將一個新操作新增到我們的列舉Operation中，編譯器不會通知我們這個開關不能正確處理新操作。

更糟糕的是，如果一個懶惰的開發人員在另一個類中複製或重新編寫這些程式碼，我們可能無法更新它。

第二個問題是預設情況default，每段程式裡面都是必需的，儘管我們知道在正確的程式碼裡它永遠不會發生。

這是因為Java編譯器知道上面的第一個問題，並且希望確保我們能夠處理在不知情的情況下向Operation中新增了新列舉。

還好，Java8用函數語言程式設計為我們提供了一個乾淨的解決方案。

函式列舉實現

因為列舉是類，所以我們可以建立一個列舉欄位來儲存執行操作的函式。

但是在我們找到解決方案之前，讓我們先來看看一些重構。

首先，讓我們把開關放在enum類中。

  
enum Operation {
  ADD,
  SUBTRACT,
  MULTIPLY;
  
  public static int apply(Operation operation, int arg1, int arg2) {
    switch(operation) {
      case ADD:
        return arg1 + arg2;
      case SUBTRACT:
        return arg1 - arg2;
      case MULTIPLY:
        return arg1 * arg2;
      default:
        throw new UnsupportedOperationException();
    }
  }
}

我們可以這樣做：Operation.apply(Operation.ADD, 2, 3);

因為我們現在從Operation中呼叫方法，所以我們可以將其更改為例項方法並使用this，而不是用Operation.apply（）來實現，如下所示：

public int apply(int arg1, int arg2) {
  switch(this) {
    case ADD:
      return arg1 + arg2;
    case SUBTRACT:
      return arg1 - arg2;
    case MULTIPLY:
      return arg1 * arg2;
    default:
      throw new UnsupportedOperationException();
  }
}

像這樣使用：Operation.ADD.apply(2, 3);

看起來變好了。現在讓我們更進一步，通過使用函數語言程式設計完全消除switch語句。

enum Operation {
              ADD((x, y) -> x + y),
              SUBTRACT((x, y) -> x - y),
              MULTIPLY((x, y) -> x * y);
  
              Operation(BiFunction<Integer, Integer, Integer> operation) {
                      this.operation = operation;
              }
  
              private final BiFunction<Integer, Integer, Integer> operation;
  
              public int apply(int x, int y) {
                      return operation.apply(x, y);
              }
  
  }

這裡我做的是：

  • 新增了一個欄位 BiFunction<Integer, Integer, Integer> operation
  • 用BiFunction建立了用於Operation的建構函式。
  • 呼叫列舉定義中的建構函式，並用lambda指定BiFunction<Integer, Integer, Integer>。

這個java.util.function.BiFunction operation欄位是對採用兩個引數的函式（方法）的引用。

在我們的例子中，兩個引數都是int型，返回值也是int型。不幸的是，Java引數化型別不支援原語，所以我們必須使用Integer。

因為BiFunction是用@functioninterface註釋的，所以我們可以使用Lambda表示法定義一個。

因為我們的函式接受兩個引數，所以我們可以使用（x，y）來指定它們。

然後我們定義了一個單行方法，它使用 ->x+y 返回一個值。這相當於下面的方法，只是更簡潔而已。

  class Adder implements BiFunction<Integer, Integer, Integer> {
          @Override
          public Integer apply(Integer x, Integer y) {
                  return x + y;
    }
  }

我們的新Operation實現採用相同的方式：Operation.ADD.apply(2, 3);.

但是，這種實現更好，因為編譯器會告訴我們何時新增了新Operation，這要求我們更新新函式。如果沒有這一點，如果我們在新增新Operation時還不記得更新switch語句，就有可能得到UnsupportedOperationException（）。

關鍵要點

  • Enum列舉是Enum的擴充套件類。

  • Enum列舉可以有欄位、建構函式和例項方法。

  • Enum列舉欄位可以儲存函式。與lambdas配合使用，可以建立乾淨、安全的特定於列舉的函式實現，並在編譯時強制執行它們（而不是使用switch）。

下面是這個示例的GitHub地址。（https://github.com/alex-power/java-enum-example）

本文參考：https://medium.com/javarevisited/advanced-java-enum-features-you-need-to-know-b516a191c7e2

歡迎關注我的公眾號：程式猿DD，獲得獨家整理的免費學習資源助力你的Java學習之路！另每週贈書不停哦～

相關文章