腦殼疼！代碼中那麼多“煩人”的if else

if else 是我們寫代碼時，使用頻率最高的關鍵詞之一，然而有時過多的 if else 會讓我們感到腦殼疼。

圖片來自 Pexels

例如下面這個偽代碼：

是不是很崩潰?雖然他是偽代碼，並且看起來也很誇張，但在現實中，當我們無數次 Review 別人代碼時，都會發現類似的場景。

那麼我們本文就來詳細聊聊，有沒有什麼方法可以讓我們避免來寫這麼多的 if else 呢?

我們本文提供了 9 種方法來解決掉那些“煩人”的 if else，一起來看吧。

1.使用 return

我們使用 return 去掉多餘的 else，實現代碼如下。

優化前代碼：

<code>if ("java".equals(str)) { 
    // 業務代碼...... 
} else { 
    return; 
} /<code>

優化後代碼：

<code>if (!"java".equals(str)) { 
    return; 
} 
// 業務代碼...... /<code>

這樣看起來就會舒服很多，雖然相差只有一行代碼，但真正的高手和普通人之間的差距就是從這一行行代碼中體現出來的。

“勿以善小而不為，勿以惡小而為之”，“千里之堤，潰於蟻穴”，說的都是同樣的道理。

2.使用 Map

使用 Map 數組，把相關的判斷信息，定義為元素信息可以直接避免 if else 判斷，實現代碼如下。

優化前代碼：

<code>if (t == 1) { 
    type = "name"; 
} else if (t == 2) { 
    type = "id"; 
} else if (t == 3) { 
    type = "mobile"; 
} /<code>

我們先定義一個 Map 數組，把相關判斷信息存儲起來：

<code>Map typeMap = new HashMap<>(); 
typeMap.put(1, "name"); 
typeMap.put(2, "id"); 
typeMap.put(3, "mobile"); /<code>

之前的判斷語句可以使用以下一行代碼代替了：

<code>type = typeMap.get(t);/<code>

3.使用三元運算符

三元運算符也叫三元表達式或者三目運算符/表達式，不過代表的都是一個意思，優化代碼如下。

優化前代碼：

<code>Integer score = 81;
if (score > 80) {
    score = 100;
} else {
    score = 60;
}/<code>

優化後代碼：

<code>score = score > 80 ? 100 : 60;/<code>

4.合併條件表達式

在項目中有些邏輯判斷是可以通過梳理和歸納，變更為更簡單易懂的邏輯判斷代碼，如下所示。

優化前代碼：

<code>String city = "西安";
String area = "029";
String province = "陝西";
if ("西安".equals(city)) {
    return "xi'an";
}
if ("029".equals(area)) {
    return "xi'an";
}
if ("陝西".equals(province)){
    return "xi'an";
}/<code>

優化後代碼：

<code>if ("西安".equals(city) || "029".equals(area) || "陝西".equals(province)){
    return "xi'an";
}/<code>

5.使用枚舉

JDK 1.5 中引入了新的類型——枚舉(enum)，我們使用它可以完成很多功能，例如下面這個。

優化前代碼：

<code>Integer typeId = 0;
String type = "Name";
if ("Name".equals(type)) {
    typeId = 1;
} else if ("Age".equals(type)) {
    typeId = 2;
} else if ("Address".equals(type)) {
    typeId = 3;
}/<code>

優化時，我們先來定義一個枚舉：

<code>public enum TypeEnum {
    Name(1), Age(2), Address(3);
    public Integer typeId;
    TypeEnum(Integer typeId) {
        this.typeId = typeId;
    }
}
/<code>

之前的 if else 判斷就可以被如下一行代碼所替代了：

<code>typeId = TypeEnum.valueOf("Name").typeId;/<code>

6.使用 Optional

從 JDK 1.8 開始引入 Optional 類，在 JDK 9 時對 Optional 類進行了改進，增加了 ifPresentOrElse() 方法，我們可以藉助它，來消除 if else 的判斷，使用如下。

優化前代碼：

<code>String str = "java";
if (str == null) {
    System.out.println("Null");
} else {
    System.out.println(str);
}/<code>

優化後代碼：

<code>Optional opt = Optional.of("java");
opt.ifPresentOrElse(v -> 
    System.out.println(v), () -> System.out.println("Null"));/<code>

小貼士：注意運行版本，必須是 JDK 9+ 才行。

7.梳理優化判斷邏輯

和第 4 點比較類似，我們可以通過分析 if else 的邏輯判斷語義，寫出更加易懂的代碼，例如以下這個嵌套判斷的優化。

優化前代碼：

<code>// 年齡大於 18
if (age > 18) {
    // 工資大於 5000
    if (salary > 5000) {
        // 是否漂亮
        if (pretty == true) {
            return true;
        }
    }
}
return false;/<code>

優化後代碼：

<code>if (age < 18) {
    return false;
}
if (salary < 5000) {
    return false;
}
return pretty; /<code>

我們需要儘量把表達式中的包含關係改為平行關係，這樣代碼可讀性更高，邏輯更清晰。

8.使用多態

繼承、封裝和多態是 OOP(面向對象編程)的重要思想，本文我們使用多態的思想，提供一種去除 if else 方法。

優化前代碼：

<code>Integer typeId = 0;
String type = "Name";
if ("Name".equals(type)) {
    typeId = 1;
} else if ("Age".equals(type)) {
    typeId = 2;
} else if ("Address".equals(type)) {
    typeId = 3;
}/<code>

使用多態，我們先定義一個接口，在接口中聲明一個公共返回 typeId 的方法，在添加三個子類分別實現這三個子類。

實現代碼如下：

<code>public interface IType {
    public Integer getType();
}

public class Name implements IType {
    @Override
    public Integer getType() {
        return 1;
    }
}

public class Age implements IType {
    @Override
    public Integer getType() {
        return 2;
    }
}

public class Address implements IType {
    @Override
    public Integer getType() {
        return 3;
    }
}/<code>

注意：為了簡便我們這裡把類和接口放到了一個代碼塊中，在實際開發中應該分別創建一個接口和三個類分別存儲。

此時，我們之前的 if else 判斷就可以改為如下代碼：

<code>IType itype = (IType) Class.forName("com.example." + type).newInstance();
Integer typeId = itype.getType();/<code>

有人可能會說，這樣反而讓代碼更加複雜了，此可謂“殺雞焉用宰牛刀”的典型範例了。

這裡作者只是提供一種實現思路和提供了一些簡易版的代碼，以供開發者在實際開發中，多一種思路和選擇，具體用不用需要根據實際情況來定了。靈活變通，舉一反三，才是開發的上乘心法。

9.選擇性的使用 switch

很多人都搞不懂 switch 和 if else 的使用場景，但在兩者都能使用的情況下，可以儘量使用 switch，因為 switch 在常量分支選擇時，switch 性能會比 if else 高。

if else 判斷代碼：

<code>if ("add".equals(cmd)) {
    result = n1 + n2;
} else if ("subtract".equals(cmd)) {
    result = n1 - n2;
} else if ("multiply".equals(cmd)) {
    result = n1 * n2;
} else if ("divide".equals(cmd)) {
    result = n1 / n2;
} else if ("modulo".equals(cmd)) {
    result = n1 % n2;
}/<code>

switch 代碼：

<code>switch (cmd) {
    case "add":
        result = n1 + n2;
        break;
    case "subtract":
        result = n1 - n2;
        break;
    case "multiply":
        result = n1 * n2;
        break;
    case "divide":
        result = n1 / n2;
        break;
    case "modulo":
        result = n1 % n2;
        break;
}/<code>

在 Java 14 可使用 switch 代碼塊，實現代碼如下：

<code>// java 14
switch (cmd) {
    case "add" -> {
        result = n1 + n2;
    }
    case "subtract" -> {
        result = n1 - n2;
    }
    case "multiply" -> {
        result = n1 * n2;
    }
    case "divide" -> {
        result = n1 / n2;
    }
    case "modulo" -> {
        result = n1 % n2;
    }
}
/<code>

10.總結

業精於勤荒於嬉，行成於思毀於隨。編程是一門手藝，更是一種樂趣，哈佛最受歡迎的幸福課《幸福的方法》一書中寫到“讓我們能感到快樂和幸福的方法，無非是全身心的投入到自己稍微努力一下就能完成的工作中去”!

是啊，太簡單的事情通常無法調動起我們的興趣，而太難的工作又會讓我們喪失信心，只有那些看似很難但稍微努力一點就能完成的事情，才會給我們帶來巨大的快樂。

我想編程也是一樣，普通的方法每個人都會寫，然而優雅一點的代碼不是所有人都能寫得出來，而本文恰恰是提供了寫出優雅代碼的一些思路，希望可以幫助和啟發到你。

參考&鳴謝：

https://www.tuicool.com/wx/2euqQvZ

http://www.blogjava.net/xzclog/archive/2006/10/16/75399.html

作者：王磊

編輯：陶家龍

出處：轉載自微信公眾號Java中文社群(ID：javacn666)