巧用枚举替代if-阿里云开发者社区

巧用枚举替代if

2023-09-18 48

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 巧用枚举替代if

前言

亲爱的友友们，我们今天来看一下如何巧妙利用枚举来替代if语句

能实现功能的代码千篇一律，但优雅的代码万里挑一

业务背景

在coding中遇到一个需求，经过简化后就是：需要根据不同的code值，处理逻辑，然后返回对应的对象。

我就简答举个栗子哈💬

根据不同的code，返回不同的对象

传1 返回一个对象，包含属性：name、age ; 传2，返回一个对象，包含属性name ; 传3，返回一个对象，包含属性sex …

字段值默认为 test

思路

摇头版

public class TestEnum {
    public static void main(String[] args) {
        Integer code = 1;//这里为了简单，直接这么写的，实际情况一般是根据参数获取
        JSONObject jsonObject = new JSONObject();
        if(Objects.equals(0,code)){
            jsonObject.fluentPut("name", "test").fluentPut("age", "test");
            System.out.println("jsonObject = " + jsonObject);
        }
        if(Objects.equals(1, code)){
            jsonObject.fluentPut("name", "test");
            System.out.println("jsonObject = " + jsonObject);
        }
        if(Objects.equals(2,code)){
            jsonObject.fluentPut("sex", "test");
            System.out.println("jsonObject = " + jsonObject);
        }
    }
}

上面的代码在功能上是没有问题滴，但是要扩展的话就💘，比如当code为4时，ba la ba la，我们只有再去写一遍if语句，随着code的增加，if语句也会随之增加，后面的人接手你的代码时 💔

优雅版

我们首先定义一个枚举类，维护对应Code需要返回的字段

@Getter
@AllArgsConstructor
public enum DataEnum {
    /**
     * 枚举类
     */
    CODE1(1,new ArrayList<>(Arrays.asList("name","age"))),
    CODE2(2,new ArrayList<>(Arrays.asList("name"))),
    CODE3(3,new ArrayList<>(Arrays.asList("sex")))
    ;
    private Integer code;
    private List<String> fields;
  //传入code 即可获取对应的 fields
    public static List<String> getFieldsByCode(Integer code){
        DataEnum[] values = DataEnum.values();
        for (DataEnum value : values) {
            if(Objects.equals(code, value.getCode())) {
                return value.getFields();
            }
        }
        return null;
    }
}

客户端代码

public class TestEnum {
    public static void main(String[] args) {
        //优雅版
        JSONObject jsonObject = new JSONObject();
        //传入code，获取fields
        List<String> fieldsByCode = DataEnum.getFieldsByCode(1);
        assert fieldsByCode != null;
        fieldsByCode.forEach(x->{
           jsonObject.put(x,"test");
        });
        System.out.println(jsonObject);
    }
}