在switch中使用枚举替换if...else方式

傷城~ 2022-11-19 07:55 392阅读 0赞

在项目中经常会遇到要判断枚举值的。很多人都会使用if .... else 的方式来简单处理下。

枚举值少的还好说，当一项枚举值多了之后，你就会发现满屏尽带if ... else

体验下（密集恐惧症请略过）

if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }else if(){
    //TODO ...
    }

用这种方式有个很大的问题就是代码的维护性比较差，差的原因主要是容易出错！一个不小心就写错了判断条件。

很多人会尝试用其他的方式来替换。例如使用switch的方式来替换。

switch基本格式

switch(){
        case a:
            //TODO
            break;
        case b:
            //TODO
            break;
        ...
        default:
        //TODO
    
    }

在使用枚举进行switch的时候，会提示需要常量表达式。switch中使用的都是常量的方式。解决这个问题的方式很简单，在枚举中定义一个static方法。就可以解决这个问题了。

代码如下：

public static TypeEnum getByValue(Integer value){
                for(OriginTypeEnum typeEnum : values()){
                    if (typeEnum.getCode().equals(value) ) {
                        return typeEnum;
                    }
                }
                return null;
            }

完整代码如下：

enum TypeEnum {
            
            XX(0,"小学"), 
              
            CZ(1,"初中"),
            
            GZ(2,"高中"),
             
            DX(3,"大学");
     
            private final Integer code;
     
            private final String desc;
    
            private TypeEnum(Integer code, String desc) {
                this.code = code;
                this.desc = desc;
            }
    
         public static TypeEnum getByValue(Integer value){
                for(TypeEnum typeEnum : values()){
                    if (typeEnum.getCode().equals(value) ) {
                        return typeEnum;
                    }
                }
                return null;
            }

使用代码：

public static void main(String[] args){
        Integer i = 0;
        switch (TypeEnum.getByValue(i)) {
            case XX:
                //TODO
                 System.out.println("小学");
                break;
             case CZ:
                //TODO
                 System.out.println("初中");
                break;
             case GZ:
                //TODO
                System.out.println("高中");
                break;
            default:
                System.out.println("大学");
        }
    }