1. 程式人生 > 其它 >優化if-else程式碼的八種方案

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

前言

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

}

參考:https://mp.weixin.qq.com/s/tK0r0jNbYpvezNqWft5ydA