1. 程式人生 > >恕我直言,我懷疑你並不會用 Java 列舉

恕我直言,我懷疑你並不會用 Java 列舉

開門見山地說吧,enum(列舉)是 Java 1.5 時引入的關鍵字,它表示一種特殊型別的類,預設繼承自 java.lang.Enum。

為了證明這一點,我們來新建一個列舉 PlayerType:

public enum PlayerType {
    TENNIS,
    FOOTBALL,
    BASKETBALL
}

兩個關鍵字帶一個類名,還有大括號,以及三個大寫的單詞,但沒看到繼承 Enum 類啊?彆著急,心急吃不了熱豆腐啊。使用 JAD 檢視一下反編譯後的位元組碼,就一清二楚了。

public final class PlayerType extends Enum
{

    public static PlayerType[] values()
    {
        return (PlayerType[])$VALUES.clone();
    }

    public static PlayerType valueOf(String name)
    {
        return (PlayerType)Enum.valueOf(com/cmower/baeldung/enum1/PlayerType, name);
    }

    private PlayerType(String s, int i)
    {
        super(s, i);
    }

    public static final PlayerType TENNIS;
    public static final PlayerType FOOTBALL;
    public static final PlayerType BASKETBALL;
    private static final PlayerType $VALUES[];

    static 
    {
        TENNIS = new PlayerType("TENNIS", 0);
        FOOTBALL = new PlayerType("FOOTBALL", 1);
        BASKETBALL = new PlayerType("BASKETBALL", 2);
        $VALUES = (new PlayerType[] {
            TENNIS, FOOTBALL, BASKETBALL
        });
    }
}

看到沒?PlayerType 類是 final 的,並且繼承自 Enum 類。這些工作我們程式設計師沒做,編譯器幫我們悄悄地做了。此外,它還附帶幾個有用靜態方法,比如說 values()valueOf(String name)

01、內部列舉

好的,小夥伴們應該已經清楚列舉長什麼樣子了吧?既然列舉是一種特殊的類,那它其實是可以定義在一個類的內部的,這樣它的作用域就可以限定於這個外部類中使用。

public class Player {
    private PlayerType type;
    public enum PlayerType {
        TENNIS,
        FOOTBALL,
        BASKETBALL
    }

    public boolean isBasketballPlayer() {
      return getType() == PlayerType.BASKETBALL;
    }

    public PlayerType getType() {
        return type;
    }

    public void setType(PlayerType type) {
        this.type = type;
    }
}

PlayerType 就相當於 Player 的內部類,isBasketballPlayer() 方法用來判斷運動員是否是一個籃球運動員。

由於列舉是 final 的,可以確保在 Java 虛擬機器中僅有一個常量物件(可以參照反編譯後的靜態程式碼塊「static 關鍵字帶大括號的那部分程式碼」),所以我們可以很安全地使用“==”運算子來比較兩個列舉是否相等,參照 isBasketballPlayer() 方法。

那為什麼不使用 equals() 方法判斷呢?

if(player.getType().equals(Player.PlayerType.BASKETBALL)){};
if(player.getType() == Player.PlayerType.BASKETBALL){};

“==”運算子比較的時候,如果兩個物件都為 null,並不會發生 NullPointerException,而 equals() 方法則會。

另外, “==”運算子會在編譯時進行檢查,如果兩側的型別不匹配,會提示錯誤,而 equals() 方法則不會。

02、列舉可用於 switch 語句

這個我在之前的一篇我去的文章中詳細地說明過了,感興趣的小夥伴可以點選連結跳轉過去看一下。

switch (playerType) {
        case TENNIS:
            return "網球運動員費德勒";
        case FOOTBALL:
            return "足球運動員C羅";
        case BASKETBALL:
            return "籃球運動員詹姆斯";
        case UNKNOWN:
            throw new IllegalArgumentException("未知");
        default:
            throw new IllegalArgumentException(
                    "運動員型別: " + playerType);

    }

03、列舉可以有構造方法

如果列舉中需要包含更多資訊的話,可以為其新增一些欄位,比如下面示例中的 name,此時需要為列舉新增一個帶參的構造方法,這樣就可以在定義列舉時新增對應的名稱了。

public enum PlayerType {
    TENNIS("網球"),
    FOOTBALL("足球"),
    BASKETBALL("籃球");

    private String name;

    PlayerType(String name) {
        this.name = name;
    }
}

04、EnumSet

EnumSet 是一個專門針對列舉型別的 Set 介面的實現類,它是處理列舉型別資料的一把利器,非常高效(內部實現是位向量,我也搞不懂)。

因為 EnumSet 是一個抽象類,所以建立 EnumSet 時不能使用 new 關鍵字。不過,EnumSet 提供了很多有用的靜態工廠方法:

下面的示例中使用 noneOf() 建立了一個空的 PlayerType 的 EnumSet;使用 allOf() 建立了一個包含所有 PlayerType 的 EnumSet。

public class EnumSetTest {
    public enum PlayerType {
        TENNIS,
        FOOTBALL,
        BASKETBALL
    }

    public static void main(String[] args) {
        EnumSet<PlayerType> enumSetNone = EnumSet.noneOf(PlayerType.class);
        System.out.println(enumSetNone);

        EnumSet<PlayerType> enumSetAll = EnumSet.allOf(PlayerType.class);
        System.out.println(enumSetAll);
    }
}

程式輸出結果如下所示:

[]
[TENNIS, FOOTBALL, BASKETBALL]

有了 EnumSet 後,就可以使用 Set 的一些方法了:

05、EnumMap

EnumMap 是一個專門針對列舉型別的 Map 介面的實現類,它可以將列舉常量作為鍵來使用。EnumMap 的效率比 HashMap 還要高,可以直接通過陣列下標(列舉的 ordinal 值)訪問到元素。

和 EnumSet 不同,EnumMap 不是一個抽象類,所以建立 EnumMap 時可以使用 new 關鍵字:

EnumMap<PlayerType, String> enumMap = new EnumMap<>(PlayerType.class);

有了 EnumMap 物件後就可以使用 Map 的一些方法了:

和 HashMap 的使用方法大致相同,來看下面的例子:

EnumMap<PlayerType, String> enumMap = new EnumMap<>(PlayerType.class);
enumMap.put(PlayerType.BASKETBALL,"籃球運動員");
enumMap.put(PlayerType.FOOTBALL,"足球運動員");
enumMap.put(PlayerType.TENNIS,"網球運動員");
System.out.println(enumMap);

System.out.println(enumMap.get(PlayerType.BASKETBALL));
System.out.println(enumMap.containsKey(PlayerType.BASKETBALL));
System.out.println(enumMap.remove(PlayerType.BASKETBALL));

程式輸出結果如下所示:

{TENNIS=網球運動員, FOOTBALL=足球運動員, BASKETBALL=籃球運動員}
籃球運動員
true
籃球運動員

06、單例

通常情況下,實現一個單例並非易事,不信,來看下面這段程式碼

public class Singleton {  
    private volatile static Singleton singleton; 
    private Singleton (){}  
    public static Singleton getSingleton() {  
    if (singleton == null) {
        synchronized (Singleton.class) { 
        if (singleton == null) {  
            singleton = new Singleton(); 
        }  
        }  
    }  
    return singleton;  
    }  
}

但列舉的出現,讓程式碼量減少到極致:

public enum EasySingleton{
    INSTANCE;
}

完事了,真的超級短,有沒有?列舉預設實現了 Serializable 介面,因此 Java 虛擬機器可以保證該類為單例,這與傳統的實現方式不大相同。傳統方式中,我們必須確保單例在反序列化期間不能建立任何新例項。

07、列舉可與資料庫互動

我們可以配合 Mybatis 將資料庫欄位轉換為列舉型別。現在假設有一個數據庫欄位 check_type 的型別如下:

`check_type` int(1) DEFAULT NULL COMMENT '檢查型別(1:未通過、2:通過)',

它對應的列舉型別為 CheckType,程式碼如下:

public enum CheckType {
    NO_PASS(0, "未通過"), PASS(1, "通過");
    private int key;

    private String text;

    private CheckType(int key, String text) {
        this.key = key;
        this.text = text;
    }

    public int getKey() {
        return key;
    }

    public String getText() {
        return text;
    }

    private static HashMap<Integer,CheckType> map = new HashMap<Integer,CheckType>();
    static {
        for(CheckType d : CheckType.values()){
            map.put(d.key, d);
        }
    }

    public static CheckType parse(Integer index) {
        if(map.containsKey(index)){
            return map.get(index);
        }
        return null;
    }
}

1)CheckType 添加了構造方法,還有兩個欄位,key 為 int 型,text 為 String 型。

2)CheckType 中有一個public static CheckType parse(Integer index)方法,可將一個 Integer 通過 key 的匹配轉化為列舉型別。

那麼現在,我們可以在 Mybatis 的配置檔案中使用 typeHandler 將資料庫欄位轉化為列舉型別。

<resultMap id="CheckLog" type="com.entity.CheckLog">
  <id property="id" column="id"/>
  <result property="checkType" column="check_type" typeHandler="com.CheckTypeHandler"></result>
</resultMap>

其中 checkType 欄位對應的類如下:

public class CheckLog implements Serializable {

    private String id;
    private CheckType checkType;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public CheckType getCheckType() {
        return checkType;
    }

    public void setCheckType(CheckType checkType) {
        this.checkType = checkType;
    }
}

CheckTypeHandler 轉換器的類原始碼如下:

public class CheckTypeHandler extends BaseTypeHandler<CheckType> {

    @Override
    public CheckType getNullableResult(ResultSet rs, String index) throws SQLException {
        return CheckType.parse(rs.getInt(index));
    }

    @Override
    public CheckType getNullableResult(ResultSet rs, int index) throws SQLException {
        return CheckType.parse(rs.getInt(index));
    }

    @Override
    public CheckType getNullableResult(CallableStatement cs, int index) throws SQLException {
        return CheckType.parse(cs.getInt(index));
    }

    @Override
    public void setNonNullParameter(PreparedStatement ps, int index, CheckType val, JdbcType arg3) throws SQLException {
        ps.setInt(index, val.getKey());
    }
}

CheckTypeHandler 的核心功能就是呼叫 CheckType 列舉類的 parse() 方法對資料庫欄位進行轉換。

恕我直言,這篇文章看完後,我覺得小夥伴們肯定會用 Java 枚舉了,如果還不會,就過來砍我!

如果覺得文章對你有點幫助,請微信搜尋「 沉默王二 」第一時間閱讀,回覆「併發」更有一份阿里大牛重寫的 Java 併發程式設計實戰,從此再也不用擔心面試官在這方面的刁難了。

本文已收錄 GitHub,傳送門~ ,裡面更有大廠面試完整考點,歡迎 Star。

我是沉默王二,一枚有顏值卻靠才華苟且的程式設計師。關注即可提升學習效率,別忘了三連啊,點贊、收藏、留言,我不挑,嘻嘻。