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

[[318084]]

圖片來自 Pexels

優化方案 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(); 
    } 
} 

參考與感謝：

• 6 個實例詳解如何把 if-else 代碼重構成高質量代碼
• 如何 “干掉” if...else