徹底消滅if-else，你需要這8種方案！

代碼中如果 if-else 比較多，閱讀起來比較困難，維護起來也比較困難，很容易出 Bug，接下來，本文將介紹優化 if-else 代碼的八種方案。

---

---

---

優化方案 1：提前 return，去除不必要的 else

 

如果 if-else 代碼塊包含 return 語句，可以考慮通過提前 return，把多餘 else 幹掉，使代碼更加優雅。

 

優化前：

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

 

優化後：

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

 

優化方案 2：使用條件三目運算符

 

使用條件三目運算符可以簡化某些 if-else，使代碼更加簡潔，更具有可讀性。

 

優化前：

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

 

優化後：

int price = condition?80:100;

 

優化方案 3：使用枚舉

 

在某些時候，使用枚舉也可以優化 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();

 

優化方案 4：合併條件表達式

 

如果有一系列條件返回一樣的結果，可以將它們合併爲一個條件表達式，讓邏輯更加清晰。

 

優化前：

 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
    }

 

優化方案 5：使用 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"));

 

優化方案 6：表驅動法

 

表驅動法，又稱之爲表驅動、表驅動方法。表驅動方法是一種使你可以在表中查找信息，而不必用很多的邏輯語句（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);

 

優化方案 7：優化邏輯結構，讓正常流程走主幹

 

優化前：

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;
}

 

將條件反轉使異常情況先退出，讓正常流程維持在主幹流程，可以讓代碼結構更加清晰。

 

優化方案 8：策略模式+工廠方法消除 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();
    }
}