前言

《阿里巴巴开发手册》中,有关于多 if-else 分支和嵌套的建议和解决方案,如下:

Java 多层内嵌属性 java多层if嵌套怎么优化_状态模式

那么本文介绍几种优化方案,给大家提供一些思路

卫语句 

《重构--改善既有代码的设计》 书籍中有有如下描述:

如果某个条件极其罕见,就应该单独检查该条件,并在条件为真时立即从函数中返回。这样的单独检查常常被称为 “卫语句”。

Java 多层内嵌属性 java多层if嵌套怎么优化_ide_02

Java 多层内嵌属性 java多层if嵌套怎么优化_百度_03

使用卫语句,我们可以对上面的示例修改为:

public String getFactoryName(int type) {

        // 条件检查
        if (type == 0) {
            throw new IllegalArgumentException("参数有误");
        }

        if (type == 1) {
            return "京东";
        }

        if (type == 2) {
            return "百度";
        }
        
        return "阿里巴巴";
    }

先进行条件检查,然后将 if-else 逻辑转成对应的卫语句格式。

枚举

《Effective Java 中文版》 中第 30 条 :用 enum 代替 int 常量 小节 有描述:使用枚举,来替代分支语句,虽然失去了简洁性,但是更加安全和灵活。

通过在枚举内部定义抽象函数,每个枚举常量重写该函数,这样根据枚举值获取枚举常量后调用该函数即可获得期待的计算结果。

示例代码如下:

public enum FactoryEnum {

    JD(1) {
        @Override
        String getTitle() {
            return "京东";
        }
    },
    BaiDu(2){
        @Override
        String getTitle() {
            return "百度";
        }
    },
    TenXun(3){
        @Override
        String getTitle() {
            return "腾讯";
        }
    },
    AliBaBa(4){
        @Override
        String getTitle() {
            return "阿里巴巴";
        }
    };

    private final int type;

    FactoryEnum(int type){
        this.type = type;
    }

    /**
     * 获取枚举名称
     * @return
     */
    abstract String getTitle();

    /**
     * 根据类型获取枚举类
     * @param type
     * @return
     */
    public static FactoryEnum valueOf(int type){
        for (FactoryEnum factoryEnum : FactoryEnum.values()) {
            if (factoryEnum.type == type){
                return factoryEnum;
            }
        }
        return null;
    }
}

状态模式

《设计模式之禅》 状态模式章节中讲到:

Java 多层内嵌属性 java多层if嵌套怎么优化_java_04

状态模式的其中一个优点就是 “结构清晰”。状态模式体现了开闭原则和单一职责原则,易于拓展和维护。

所谓的结构清晰就是避免了过多的 switch-case 或者 if-else 语句的使用,避免了程序的复杂性,提高了程序的可维护性。

《设计模式之禅》书提供的案例和文章上面给出的非常类似,根据当前状态来执行不同的行为,有兴趣可以翻出看看

同样优化下上面的案例代码:

定义一个状态行为接口

public interface FactoryState {
    String getFactory();
}

定义多个具体的状态实现

static class JD_State implements FactoryState{
        @Override
        public String getFactory() {
            return "京东";
        }
    }

    static  class BaiDu_State implements FactoryState{
        @Override
        public String getFactory() {
            return "百度";
        }
    }

定义一个状态使用的client

public class State_Client {
    private FactoryState factoryState;

    public void setFactoryState(FactoryState factoryState) {
        this.factoryState = factoryState;
    }

    public FactoryState getFactoryState() {
        return factoryState;
    }
    public void getFactory(){
        factoryState.getFactory();
    }
}

外部使用

public class TestState {
    public static void main(String[] args) {
        State_Client client = new State_Client();
        client.setFactoryState(new State.JD_State());
        client.getFactory();
    }
}

策略+工厂模式

首先,我们把每个条件逻辑代码块,抽象成一个公共的接口,可以得出以下代码

总结

本文主要讲了如何解决 if-else 语句拓展性和多层嵌套问题。可以通过卫语句、枚举、状态模式、策略+工厂模式等方式解决,如果大家有更好的方案,可以和我交流

希望大家能够在实际开发中尝试使用这些方法来编写更加优雅的代码。