Java8 利用函数式接口解决大量的if else

最近在写公司业务的时候发现存在大量的if else我就想着如何能够消除大量的if和else。第一反应是利用模式写大量的类去解决，但是那样我感觉也很烦，增加了那么多文件。对于平时搬砖得到业务系统来说，也不一定值当。恰好前段时间看了《on java 8》的函数式编程那一章，于是就想着能不能像js把函数引用保存起来，用Map去映射。实时证明这样是可以的。

先看看大量的if else

   public void test(String operator) {

        if (TestEnum.OPEN.getCode().equals(operator)) {

            // todo something
        } else if (TestEnum.CLOSE.getCode().equals(operator)) {
            // todo something
        }else if(TestEnum.RESTART.getCode().equals(operator)){
            // todo something
        } else if(){
            // ...
        }else if(){
            // ...
        }else{
            // ...
        }
    }

这里只是展示了部分代码，实际情况比这还多。

再看一下Enum类的代码，我们的改造从两方便入手：

public enum TestEnum {
    // 开
    OPEN("open", "开机"),

    // 关
    CLOSE("close", "关机"),

    // 重启
    RESTART("restart", "重启");

    private String code;
    private String title;

    TestEnum(String code, String title) {
        this.code = code;
        this.title = title;
    }

    TestEnum() {
    }

    public String getCode() {
        return code;
    }

    public String getTitle() {
        return title;
    }

    public void setCode(String code) {
        this.code = code;
    }

    public void setTitle(String title) {
        this.title = title;
    }
}

改造 if else

首先我们知道问题的根源是我们需要为不同的枚举值选不同的处理方法，来看代码:

@Configuration
public class ActionMap<T> {

    @Autowired
    private UserService userService;

    private final Map<TestEnum, BiFunction<String, String, Integer>> actionMap = new HashMap<>();

    @Bean
    public void initMap() {
        actionMap.put(TestEnum.OPEN, (var1, var2) -> {
            // todo something
            return userService.test(var1, var2);
        });
        actionMap.put(TestEnum.CLOSE, (var1, var2) -> {
            // todo something
            return userService.test1(var1, var2);
        });
        actionMap.put(TestEnum.RESTART, (var1, var2) -> {
            // todo something
            return userService.test2(var1, var2);
        });
    }


    public void helloWorld() {
        Integer flag = actionMap.get(TestEnum.RESTART).apply("hello", "world");
    }

}

initMap方法加了@Bean注解，这样自动装备的时候就是同一个map对象了。当然这里我直接在helloWorld方法里面直接调用了。但是到这里还有一个问题就是我调用的时候还是要判断Enum的类型才能通过map的get方法

获取到不同的方法引用。所以我们要接着改造枚举类：

改造枚举类

public enum TestEnum {
    // 开
    OPEN("open", "开机"),

    // 关
    CLOSE("close", "关机"),

    // 重启
    RESTART("restart", "重启"),

	// 添加错误处理方法
    // 错误处理
    ERROR_PROCESS("", "");

    private String code;
    private String title;

    TestEnum(String code, String title) {
        this.code = code;
        this.title = title;
    }

    TestEnum() {
    }

    public String getCode() {
        return code;
    }

    public String getTitle() {
        return title;
    }

    public void setCode(String code) {
        this.code = code;
    }

    public void setTitle(String title) {
        this.title = title;
    }

//添加根据字符串获取对应枚举的方法
    public static TestEnum getEnum(String s) {
        return Arrays.stream(TestEnum.values()).filter(item -> 						item.getCode().equals(s)).findAny().orElse(ERROR_PROCESS);
    }
}

利用java 8的流语法以及Optional判空达到一句话找到对应枚举类型。

然后在调用map的地方进行改造：

@Configuration
public class ActionMap<T> {

    @Autowired
    private UserService userService;

    private final Map<TestEnum, BiFunction<String, String, Integer>> actionMap = new HashMap<>();

    @Bean
    public void initMap() {
        actionMap.put(TestEnum.OPEN, (var1, var2) -> {
            // todo something
            return userService.test(var1, var2);
        });
        actionMap.put(TestEnum.CLOSE, (var1, var2) -> {
            // todo something
            return userService.test1(var1, var2);
        });
        actionMap.put(TestEnum.RESTART, (var1, var2) -> {
            // todo something
            return userService.test2(var1, var2);
        });

        actionMap.put(TestEnum.ERROR_PROCESS, (var1, var2) -> {
            // todo something
            System.out.println("error");
            return 0;
        });
    }

    public int helloWorld(String operator, String var1, String var2) {
        return actionMap.get(TestEnum.getEnum(operator)).apply(var1, var2);
    }
}

在看看helloWorld和之前的那么多的if else。是不是我们只在配置一个map就行，不需要写那么多的if else了。当然还是需要配置map。。。。。。也挺烦。。。。。

至于返回值一般业务代码都是和数据库打交道，返回个int值方便我写测试用例。