优化if-else代码的八种方案

 [[433207]]

前言

代码中如果if-else比较多，阅读起来比较困难，维护起来也比较困难，很容易出bug，接下来，本文将介绍优化if-else代码的八种方案。

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

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

优化前： 

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

优化后： 

if（!condition）{  
    return;  
}  
//doSomething  

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

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

优化前： 

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

优化后： 

int price = condition ? 80 : 100;  

优化方案三：使用枚举

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

优化前： 

String OrderStatusDes;  
if (orderStatus == 0) {  
    OrderStatusDes = "订单未支付";  
} elseif ( OrderStatus == 1 ) {  
    OrderStatusDes = "订单已支付";  
} elseif ( OrderStatus == 2 ) {  
    OrderStatusDes = "已发货";  
}  
... 

优化后：

先定义一个枚举 

public enum OrderStatusEnum {  
      UN_PAID(0, "订单未支付"),  
      PAIDED(1, "订单已支付"),  
      SENDED(2, "已发货"), 
      ;  
      private int index;  
      private String desc;  
      public int getIndex() {  
          return index;  
      }  
      public String getDesc() {  
          return desc;  
      }  
      OrderStatusEnum(int index, String desc) {  
          this.index = index;  
          this.desc = desc;  
      }  
      OrderStatusEnum of(int orderStatus) {  
          for (OrderStatusEnum temp : OrderStatusEnum.values()) {  
              if (temp.getIndex() == orderStatus) {  
                  return temp;  
              }  
          }  
          return null;  
      }  
  } 

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

String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc();  

优化方案四：合并条件表达式

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

优化前 

double getVipDiscount () {  
  if (age < 18) {  
      return 0.8;  
  } 
  if ("深圳".equals(city)) {  
      return 0.8;  
  }  
  if (isStudent) {  
      return 0.8;  
  }  
  //do somethig  
} 

优化后 

double getVipDiscount(){  
    if (age< 18 || "深圳".equals(city)||isStudent){  
        return 0.8;  
    }  
    //doSomthing  
}  

优化方案五：使用 Optional

有时候if-else比较多，是因为非空判断导致的，这时候你可以使用java8的Optional进行优化。

优化前： 

String str = "jay@huaxiao";  
if (str != null) {  
    System.out.println(str);  
} else {  
    System.out.println("Null");  
} 

优化后： 

Optional<String> strOptional = Optional.of("jay@huaxiao");  
strOptional.ifPresentOrElse(System.out::println, () -> System.out.println("Null")); 

优化方案六：表驱动法

表驱动法，又称之为表驱动、表驱动方法。表驱动方法是一种使你可以在表中查找信息，而不必用很多的逻辑语句（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;  
} 

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

优化方案八：策略模式+工厂方法消除if else

假设需求为，根据不同勋章类型，处理相对应的勋章服务，优化前有以下代码： 

String medalType = "guest";  
if ("guest".equals(medalType)) {  
    System.out.println("嘉宾勋章");  
} else if ("vip".equals(medalType)) {  
    System.out.println("会员勋章");  
} else if ("guard".equals(medalType)) {  
    System.out.println("展示守护勋章");  
}  
... 

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

//勋章接口  
public interface IMedalService {  
    void showMedal ();  
    String getMedalType ();  
} 

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

//守护勋章策略实现类  
public class  GuardMedalServiceImpl implements IMedalService {  
    @Override  
    public void showMedal() {  
        System.out.println("展示守护勋章");  
    }  
    @Override  
    public String getMedalType() {  
        return "guard";  
    }  
}  
//嘉宾勋章策略实现类  
public class GuestMedalServiceImpl implements IMedalService {  
    @Override  
    public void showMedal() {  
        System.out.println("嘉宾勋章");  
    }  
    @Override  
    public String getMedalType() {  
        return "guest";  
    }  
}  
//VIP勋章策略实现类  
public class VipMedalServiceImpl implements IMedalService {  
    @Override  
    public void showMedal() {  
        System.out.println("会员勋章");  
    }  
    @Override  
    public String getMedalType() {  
        return "vip";  
    }  
} 

接下来，我们再定义策略工厂类，用来管理这些勋章实现策略类，如下： 

//勋章服务工产类  
public class MedalServicesFactory {  
    private static final Map<String, IMedalService> map = new HashMap<>();  
    static {  
        map.put("guard", new GuardMedalServiceImpl());  
        map.put("vip", new VipMedalServiceImpl());  
        map.put("guest", new GuestMedalServiceImpl());  
    }  
    public static IMedalService getMedalService(String medalType) {  
        return map.get(medalType);  
    }  
}  
... 

使用了策略+工厂模式之后，代码变得简洁多了，如下： 

public class Test {  
    public static void main(String[] args) {  
        String medalType = "guest";  
        IMedalService medalService = MedalServicesFactory.getMedalService(medalType);  
        medalService.showMedal();  
    }  
}