優化if-else程式碼的八種方案!

  • 2020 年 3 月 13 日
  • 筆記

前言

程式碼中如果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 ="訂單未支付";  }else if(OrderStatus==1){      OrderStatusDes ="訂單已支付";  }else if(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();  }

我們根據每個邏輯條件,定義相對應的策略實現類,可得以下程式碼:

//守護勳章策略實現類  public class GuardMedalServiceImpl implements IMedalService {      @Override      public void showMedal() {          System.out.println("展示守護勳章");      }  }  //嘉賓勳章策略實現類  public class GuestMedalServiceImpl implements IMedalService {      @Override      public void showMedal() {          System.out.println("嘉賓勳章");      }  }  //VIP勳章策略實現類  public class VipMedalServiceImpl implements IMedalService {      @Override      public void showMedal() {          System.out.println("會員勳章");      }  }

接下來,我們再定義策略工廠類,用來管理這些勳章實現策略類,如下:

//勳章服務工產類  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();      }  }