if else 是我們寫代碼時(shí)，使用頻率最高的關(guān)鍵詞之一，然而有時(shí)過多的 if else 會(huì)讓我們感到腦殼疼。

[[323515]]

圖片來自 Pexels

例如下面這個(gè)偽代碼：

是不是很崩潰?雖然他是偽代碼，并且看起來也很夸張，但在現(xiàn)實(shí)中，當(dāng)我們無(wú)數(shù)次 Review 別人代碼時(shí)，都會(huì)發(fā)現(xiàn)類似的場(chǎng)景。

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

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

1.使用 return

我們使用 return 去掉多余的 else，實(shí)現(xiàn)代碼如下。

優(yōu)化前代碼：

if ("java".equals(str)) { 
    // 業(yè)務(wù)代碼...... 
} else { 
    return; 
} 

優(yōu)化后代碼：

if (!"java".equals(str)) { 
    return; 
} 
// 業(yè)務(wù)代碼...... 

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

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

2.使用 Map

使用 Map 數(shù)組，把相關(guān)的判斷信息，定義為元素信息可以直接避免 if else 判斷，實(shí)現(xiàn)代碼如下。

優(yōu)化前代碼：

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

我們先定義一個(gè) Map 數(shù)組，把相關(guān)判斷信息存儲(chǔ)起來：

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

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

type = typeMap.get(t); 

3.使用三元運(yùn)算符

三元運(yùn)算符也叫三元表達(dá)式或者三目運(yùn)算符/表達(dá)式，不過代表的都是一個(gè)意思，優(yōu)化代碼如下。

優(yōu)化前代碼：

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

優(yōu)化后代碼：

score = score > 80 ? 100 : 60; 

4.合并條件表達(dá)式

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

優(yōu)化前代碼：

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

優(yōu)化后代碼：

if ("西安".equals(city) || "029".equals(area) || "陜西".equals(province)){ 
    return "xi'an"; 
} 

5.使用枚舉

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

優(yōu)化前代碼：

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

優(yōu)化時(shí)，我們先來定義一個(gè)枚舉：

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

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

typeId = TypeEnum.valueOf("Name").typeId; 

6.使用 Optional

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

優(yōu)化前代碼：

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

優(yōu)化后代碼：

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

小貼士：注意運(yùn)行版本，必須是 JDK 9+ 才行。

7.梳理優(yōu)化判斷邏輯

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

優(yōu)化前代碼：

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

優(yōu)化后代碼：

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

我們需要盡量把表達(dá)式中的包含關(guān)系改為平行關(guān)系，這樣代碼可讀性更高，邏輯更清晰。

8.使用多態(tài)

繼承、封裝和多態(tài)是 OOP(面向?qū)ο缶幊?的重要思想，本文我們使用多態(tài)的思想，提供一種去除 if else 方法。

優(yōu)化前代碼：

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

使用多態(tài)，我們先定義一個(gè)接口，在接口中聲明一個(gè)公共返回 typeId 的方法，在添加三個(gè)子類分別實(shí)現(xiàn)這三個(gè)子類。

實(shí)現(xiàn)代碼如下：

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

注意：為了簡(jiǎn)便我們這里把類和接口放到了一個(gè)代碼塊中，在實(shí)際開發(fā)中應(yīng)該分別創(chuàng)建一個(gè)接口和三個(gè)類分別存儲(chǔ)。

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

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

有人可能會(huì)說，這樣反而讓代碼更加復(fù)雜了，此可謂“殺雞焉用宰牛刀”的典型范例了。

這里作者只是提供一種實(shí)現(xiàn)思路和提供了一些簡(jiǎn)易版的代碼，以供開發(fā)者在實(shí)際開發(fā)中，多一種思路和選擇，具體用不用需要根據(jù)實(shí)際情況來定了。靈活變通，舉一反三，才是開發(fā)的上乘心法。

9.選擇性的使用 switch

很多人都搞不懂 switch 和 if else 的使用場(chǎng)景，但在兩者都能使用的情況下，可以盡量使用 switch，因?yàn)?switch 在常量分支選擇時(shí)，switch 性能會(huì)比 if else 高。

if else 判斷代碼：

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

switch 代碼：

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

在 Java 14 可使用 switch 代碼塊，實(shí)現(xiàn)代碼如下：

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

10.總結(jié)

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

是啊，太簡(jiǎn)單的事情通常無(wú)法調(diào)動(dòng)起我們的興趣，而太難的工作又會(huì)讓我們喪失信心，只有那些看似很難但稍微努力一點(diǎn)就能完成的事情，才會(huì)給我們帶來巨大的快樂。

我想編程也是一樣，普通的方法每個(gè)人都會(huì)寫，然而優(yōu)雅一點(diǎn)的代碼不是所有人都能寫得出來，而本文恰恰是提供了寫出優(yōu)雅代碼的一些思路，希望可以幫助和啟發(fā)到你。

參考&鳴謝：

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

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

作者：王磊

編輯：陶家龍

出處：轉(zhuǎn)載自微信公眾號(hào)Java中文社群(ID：javacn666)