8種優化if-else代碼的方案請拿走

  • 2020 年 3 月 12 日
  • 筆記

前言

代碼中如果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();    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();    }}