Java列舉-通過值查詢對應的列舉

_否極泰來發表於2022-01-30
Java

一、背景

Java 列舉是一個特殊的類，一般表示一組常量，比如一年的 4 個季節，一個年的 12 個月份，一個星期的 7 天，方向有東南西北等。

最近工作中，對接了很多其他的系統，發現對接的同一個系統都有不同的環境（開發、測試、正式環境），並且每個環境的配置資訊通常不會修改，所以發現用列舉 做配置項，使用起來比較簡潔，不同的環境配置 只需多定義一個列舉值就搞定了。

其中使用列舉就會涉及到通過傳入的值，返回對應的列舉。

二、通過一個值 ，查詢返回對應的列舉（示例程式碼）

2.1、列舉類

@Getter
public enum CustomType {
    TEST("test","測試","111"),
    DEV("dev","開發","222");

    String typeCode;
    String typeName;
    String orgId;

    CustomType(String typeCode, String typeName, String orgId) {
        this.typeCode = typeCode;
        this.typeName = typeName;
        this.orgId = orgId;
    }
}

2.2、常用的列舉方法；values(), ordinal() 和 valueOf() 方法

enum 定義的列舉類預設繼承了 java.lang.Enum 類，並實現了 java.lang.Seriablizable 和 java.lang.Comparable 兩個介面。

values(), ordinal() 和 valueOf() 方法位於 java.lang.Enum 類中：

  • values() 返回列舉類中所有的值。
  • ordinal()方法可以找到每個列舉常量的索引，就像陣列索引一樣。
  • valueOf()方法返回指定字串值的列舉常量。

傳入值查詢列舉，就是通過values()方法，返回所以列舉，再遍歷全部列舉，只要傳入的引數值 跟當前列舉的值跟相同，就返回當前列舉；

2.3、通過傳入一個或者多個值，返回對應的列舉
    public CustomType find(String typeCode){
        for (CustomType value : CustomType.values()) {
            if(typeCode.equals(value.getTypeCode())){
                return value;
            }
        }
        //根據自身的業務 查不到可以返回null，或者丟擲異常。
        return null;
    }

    public CustomType find(String orgId,String typeCode){
        if(orgId == null || typeCode == null){
            return null;
        }

        for (CustomType value : CustomType.values()) {
            if(orgId.equals(value.getOrgId()) &&
                    typeCode.equals(value.getTypeCode())){
                return value;
            }
        }
        //根據自身的業務 查不到可以返回null，或者丟擲異常。
        return null;
    }

三、查詢優化

每次通過values()方法遍歷查詢，時間複雜度是O(n),而通過HashMap查詢，時間複雜度是O(1)。

雖說列舉的數量通常都不會很龐大，每次通過values()方法遍歷查詢速度也很快。用HashMap會多佔用一點點記憶體，但是考慮到這一點記憶體能從時間複雜度是O(n)降到O(1)，這種惠而不費的事，還是可以花時間去優化程式碼的。

    private static Map<String,CustomType> orgCustomType = new HashMap<>();
    static {
        for (CustomType value : CustomType.values()) {
            orgCustomType.put(value.getOrgId(),value);
        }
    }

    public CustomType find(String orgId){
        return orgCustomType.get(orgId);
    }

相關文章