一、背景
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);
}