讓 Java 中 if else 更優雅的幾個小技巧


對於一個高級 crud 工程師👨‍💻而言,if else 是寫代碼時使用頻率最高的關鍵詞之一,然而有時過多的 if else 會讓我們優雅的 crud 代碼顯得不那么優雅,並且感到腦殼疼🧠

不那么優雅的if-else

是不是很奔潰?雖然是偽代碼,並且看起來也很誇張,但在現實中,當我們無數次 review 別人代碼時,都會發現類似的場景,那么有沒有什么方法可以讓我們避免來寫這么多的 if else 呢?

1. 使用 return

使用 return 去掉多余的 else,實現代碼如下。

優化前代碼:

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

優化后代碼:

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

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

2. 使用 Map

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

優化前代碼:

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

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

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

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

type = typeMap.get(t);

3. 使用三元運算符

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

優化前代碼:

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

優化后代碼:

score = score > 80 ? 100 : 60;

4. 合並條件表達式

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

優化前代碼:

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

優化后代碼:

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

5. 使用枚舉

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

優化前代碼:

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

優化時,我們先來定義一個枚舉:

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 時對 Optional 類進行了改進,增加了 ifPresentOrElse() 方法,我們可以借助它,來消除 if else 的判斷,使用如下。

優化前代碼:

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

優化后代碼:

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

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

7. 梳理優化判斷邏輯

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

優化前代碼:

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

優化后代碼:

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

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

8. 使用多態


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

優化前代碼:

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

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

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

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

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

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

這個方法看似把一個簡單的判斷做得復雜了很多,但是這里提供的只是一個思路,普通的一個判斷確實犯不着這么做,但是如果遇到的比較負責的邏輯分類場景並且需要更好的拓展性,那么這種思想就很有用了。

9. 選擇性的使用 switch

很多人都搞不懂 switch 和 if else 的使用場景,但在兩者都能使用的情況下,可以盡量使用 switch,因為 switch 在常量分支選擇時,switch 可以直接定位到制定分支,不需要想 if else 那樣一個一個判斷,因此性能會比 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 代碼塊,實現代碼如下:

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

以上9個小技巧可以幫助我們盡量減少代碼中不那么優雅的 if else 判斷,比較常用的應該是1、3、7、9,但是重要的是要有讓自己的代碼更易讀的意識。

真正的高手不是能一眼看懂別人的代碼,而且能寫出讓別人一眼就看得懂的代碼。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM