Java枚举-通过值查找对应的枚举

2022-11-27 12:44:06

一、背景

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

码农公寓