Java优化(一) if-else的优化方案

---

前言

代码中如果if-else比较多，阅读起来比较困难，维护起来也比较困难，很容易出bug，接下来，针对代码中大量的 if/else ,优化的方案进行总结。

方案一：提前return，去除不必要的else

如果if-else代码块包含return语句，可以考虑通过提前return，把多余else干掉，使代码更加优雅。

优化前：

if(condition){
    //doSomething
}else{
    return ;
}

优化后：

if（!condition）{
    return ;
}

方案二：使用条件三目运算符

使用条件三目运算符可以简化某些if-else,使代码更加简洁，更具有可读性。

优化前：

int  weight;
if(condition){
    weight= 80;
}else{
    weight = 100;
}

优化后：

int weight= condition?80:100;

方案三：使用枚举

有些时候，使用枚举也可以优化if-else逻辑分支，按个人理解，它也可以看做一种表驱动方法。

优化前：

String roleName;
if(roleType==0){
    roleName="ordinary";
}else if(roleType==1){
     roleName="admin";
}
...

优化后：

定义枚举

public enum UserRoleEnum {
    ADMIN(1, "admin"),
    ORDINARY(0, "ordinary");

    private int roleType;
    private String roleName;

    UserRoleEnum(int roleType, String roleName) {
        this.roleType = roleType;
        this.roleName = roleName;
    }

    public int getRoleType() {
        return roleType;
    }

    public String getRoleName() {
        return roleName;
    }

    UserRoleEnum getRoleName(int roleType) {
        for (UserRoleEnum roleName : UserRoleEnum.values()) {
            if (roleName.getRoleType() == roleType) {
                return roleName;
            }
        }
        return null;
    }
}

有了枚举之后，以上if-else逻辑分支，可以优化为一行代码

String roleName = UserRoleEnum.getRoleName(roleType).getRoleName();

方案四：合并条件表达式

如果有一系列条件返回一样的结果，可以将它们合并为一个条件表达式，让逻辑更加清晰。

优化前：

 double getVipDiscount() {
        if(age < 18){
            return 0.8;
        }
        if("西安".equals(city)){
            return 0.8;
        }
        if(isStudent){
            return 0.8;
        }
    }

优化后：

 double getVipDiscount(){
        if(age < 18 || "西安".equals(city) || isStudent){
            return 0.8;
        }
    }

方案五：使用 Optional

优化前：

String str = "dmjxsy@126.com";
if (str != null) {
    System.out.println(str);
} else {
    System.out.println("Null");
}

优化后：

String str = "dmjxsy@126.com";
Optional<String> strOptional = Optional.of(str);
String name=Optional.ofNullable(str).orElse("Null");
System.out.println(name);

方案六：策略模式

优化前：

String strategy = "fast";
if ("fast".equals(strategy)){
 // 快速执行
    System.out.println("fast");
}else if ("normal".equals(strategy)){
 // 正常执行
    System.out.println("normal");
}else if ("smooth".equals(strategy)){
// 平滑执行
    System.out.println("smooth");
}else if ("slow".equals(strategy)){
 // 慢慢执行
    System.out.println("slow");
}

优化后：

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

public interface Strategy {
    void run() throws Exception;
}

• 我们根据每个逻辑条件，定义相对应的策略实现类，可得以下代码：

class FastStrategy implements Strategy {
    @Override
    public void run() throws Exception {
        // 快速执行逻辑
        System.out.println("快速执行逻辑");
    }
}

class NormalStrategy implements Strategy {
    @Override
    public void run() throws Exception {
        // 正常执行逻辑
        System.out.println("正常执行逻辑");
    }
}

class SmoothStrategy implements Strategy {
    @Override
    public void run() throws Exception {
        // 平滑执行逻辑
        System.out.println("平滑执行逻辑");
    }
}

class SlowStrategy implements Strategy {
    @Override
    public void run() throws Exception {
        // 慢速执行逻辑
        System.out.println("慢速执行逻辑");
    }
}

• 我们将具体策略对象存放在一个Map中，优化后的实现：

// 初始化map，将策略放入
private static HashMap<String,Strategy > map = new HashMap<String,Strategy >(){
    {
        put("FastStrategy",new FastStrategy());
        put("NormalStrategy",new NormalStrategy());
        put("SmoothStrategy",new SmoothStrategy());
        put("SlowStrategy",new SlowStrategy());
    }
};

// 构建方法，通过传入的key触发对应的策略
public static void testStrategy(String param) throws Exception {
    Strategy strategy  = map.get(param);
    strategy.run();
}

• 测试

public class Test {
	// 测试
	public static void main(String[] args) throws Exception {
    	testStrategy("FastStrategy");
	}
}

方案七：表驱动法

表驱动法，又称之为表驱动、表驱动方法。表驱动方法是一种使你可以在表中查找信息，而不必用很多的逻辑语句（if或Case）来把它们找出来的方法。以下的demo，把map抽象成表，在map中查找信息，而省去不必要的逻辑语句。

优化前：

if (param.equals(value1)) {
    doAction1(someParams);
} else if (param.equals(value2)) {
    doAction2(someParams);
} else if (param.equals(value3)) {
    doAction3(someParams);
}

优化后：

// 这里泛型 ? 是为方便演示，实际可替换为你需要的类型
Map<?, Function<?> action> actionMappings = new HashMap<>(); 

// 初始化
actionMappings.put(value1, (someParams) -> { doAction1(someParams)});
actionMappings.put(value2, (someParams) -> { doAction2(someParams)});
actionMappings.put(value3, (someParams) -> { doAction3(someParams)});

// 省略多余逻辑语句
actionMappings.get(param).apply(someParams);

方案八：优化逻辑结构，让正常流程走主干

优化前：

public double getAdjustedCapital(){
    if(_capital <= 0.0 ){
        return 0.0;
    }
    if(_intRate > 0 && _duration >0){
        return (_income / _duration) *ADJ_FACTOR;
    }
    return 0.0;
}

优化后：

public double getAdjustedCapital(){
    if(_capital <= 0.0 ){
        return 0.0;
    }
    if(_intRate <= 0 || _duration <= 0){
        return 0.0;
    }
 
    return (_income / _duration) *ADJ_FACTOR;
}

将条件反转使异常情况先退出，让正常流程维持在主干流程，可以让代码结构更加清晰。