優化if-else程式碼的八種方案
阿新 • • 發佈:2021-11-09
前言
程式碼中如果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() { returnindex; } 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(); } }
參考:https://mp.weixin.qq.com/s/tK0r0jNbYpvezNqWft5ydA