一、前言

如果要給 Java 所有異常弄個榜單，我會選擇將 NullPointerException 放在榜首。這個異常潛伏在程式碼中，就像個遙控炸彈，不知道什麼時候這個按鈕會被突然按下（傳入 null 物件）。

還記得剛入行程式設計師的時候，三天兩頭碰到空指標異常引發的 Bug，解決完一個，又在另一處碰到。那時候師兄就教我，不要相信任何『物件』，特別是別人給你的，這些地方都加上判斷。於是程式碼通常為會變成下面這樣：

if(obj!=null){
    // do something
}

有了這個防禦之後，雖然不用再擔心空指標異常，但是過多的判斷語句使得程式碼變得臃腫。

假設我們存在如下物件關係

原本為了獲取圖中的 name 屬性，原本一句程式碼就可以輕鬆完成。

Staff staff=..;
staff.getDepartment().getCompany().getName();

但是很不幸，為了程式碼的安全性，我們不得不加入空指標判斷程式碼。

Staff staff=..;
if (staff != null) {
    Department department = staff.getDepartment();
    if (department != null) {
        Company company = department.getCompany();
        if (company != null) {
            return company.getName();
        }
    }
}
return "Unknown";
 

當其中物件為 null 時，可以返回預設值，如上所示。也可以直接丟擲其他異常快速失敗。

雖然上面程式碼變得更加安全，但是過多巢狀 if 語句降低程式碼整體可讀性，提高複雜度。

所幸 Java 8 引入引入一個新類 Java.util.Optional<T>,依靠 Optional 類提供 API，我們可以寫出既安全又具有閱讀性的程式碼。

還在使用 JDK 6 ？那你也別急著關閉這篇文章。可以考慮使用 Guava Optional。不過需要注意的是，Guava Optional API 與 JDK 存在差異，以下以 JDK8 Optional 為例。

二、Optional API

2.1、Optional#of 與 Optional#ofNullable

Optional 本質是一個容器，需要我們將物件例項傳入該容器中。Optional 的構造方法為 private，無法直接使用 new 構建物件，只能使用 Optional 提供的靜態方法建立。

Optional 三個建立方法如下：

• Optional.of(obj),如果物件為 null，將會丟擲 NPE。
• Optional.ofNullable(obj),如果物件為 null,將會建立不包含值的 empty Optional 物件例項。
• Optional.empty() 等同於 Optional.ofNullable(null)
• 

只有在確定物件不會為 null 的情況使用 Optional#of，否則建議使用 Optional#ofNullable方法。

2.2、Optional#get 與 Optional#isPresent

物件例項存入 Optional 容器中之後，最後我們需要從中取出。Optional#get 方法用於取出內部物件例項，不過需要注意的是，如果是 empty Optional 例項，由於容器內沒有任何物件例項，使用 get 方法將會丟擲 NoSuchElementException 異常。

為了防止異常丟擲，可以使用 Optional#isPresent 。這個方法將會判斷內部是否存在物件例項，若存在則返回 true。

示例程式碼如下:

Optional<Company> optCompany = Optional.ofNullable(company);
// 與直接使用空指標判斷沒有任何區別
if (optCompany.isPresent()) {
    System.out.println(optCompany.get().getName());
}

仔細對比，可以發現上面用法與空指標檢查並無差別。剛接觸到 Optional ，看到很多文章介紹這個用法，那時候一直很疑惑，這個解決方案不是更加麻煩？

後來接觸到 Optional 其他 API，我才發現這個類真正有意義是下面這些 API。如果使用過 Java8 Stream 的 API，下面 Optional API 你將會很熟悉。

2.3、Optional#ifPresent

通常情況下，空指標檢查之後，如果物件不為空，將會進行下一步處理，比如列印該物件。

Company company = ...;
if(company!=null){
    System.out.println(company);
}

上面程式碼我們可以使用 Optional#ifPresent 代替，如下所示：

Optional<Company> optCompany = ...;
optCompany.ifPresent(System.out::println);

使用 ifPresent 方法，我們不用再顯示的進行檢查，如果 Optional 為空，上面例子將不再輸出。

2.4、Optional#filter

有時候我們需要某些屬性滿足一定條件，才進行下一步動作。這裡假設我們當 Company name 屬性為 Apple，列印輸出 ok。

if (company != null && "Apple".equals(company.getName())) {
    System.out.println("ok");
}

下面使用 Optional#filter 結合 Optional#ifPresent 重寫上面的程式碼，如下所示：

Optional<Company> companyOpt=...;
companyOpt
        .filter(company -> "Apple".equals(company.getName()))
        .ifPresent(company -> System.out.println("ok"));

filter 方法將會判斷物件是否符合條件。如果不符合條件，將會返回一個空的 Optional 。

2.5、Optional#orElse 與 Optional#orElseThrow

當一個物件為 null 時，業務上通常可以設定一個預設值，從而使流程繼續下去。

String name = company != null ? company.getName() : "Unknown";

或者丟擲一個內部異常，記錄失敗原因，快速失敗。

if (company.getName() == null) {
    throw new RuntimeException();
}

Optional 類提供兩個方法 orElse 與 orElseThrow ，可以方便完成上面轉化。

// 設定預設值
String name=companyOpt.orElse(new Company("Unknown")).getName();

// 丟擲異常
String name=companyOpt.orElseThrow(RuntimeException::new).getName();

如果 Optional 為空，提供預設值或丟擲異常。

2.6、Optional#map 與 Optional#flatMap

熟悉 Java8 Stream 同學的應該瞭解，Stream#map 方法可以將當前物件轉化為另外一個物件， Optional#map 方法也與之類似。

Optional<Company> optCompany = ...;
Optional<String> nameopt = optCompany.map(Company::getName);

map 方法可以將原先 Optional<Company> 轉變成 Optional<String> ，此時 Optional 內部物件變成 String 型別。如果轉化之前 Optional 物件為空，則什麼也不會發生。

另外 Optional 還有一個 flatMap 方法，兩者區別見下圖。

Department#getCompany返回物件為 Optional<Company>

三、程式碼重構

上面我們學習了 Optional 類主要 API ，下面我們使用 Optional 重構文章剛開頭的程式碼。為了方便讀者對比，將上面的程式碼複製了下來。

程式碼重構前：

if (staff != null) {
    Department department = staff.getDepartment();
    if (department != null) {
        Company company = department.getCompany();
        if (company != null) {
            return company.getName();
        }
    }
}
return "Unknown";

首先我們需要將 Staff，Department 修改 getter 方法返回結果型別改成 Optional。

public class Staff {
    private Department department;
    public Optional<Department> getDepartment() {
        return Optional.ofNullable(department);
    }
    ...
}
public class Department {

    private Company company;
    public Optional<Company> getCompany() {
        return Optional.ofNullable(company);
    }
    ...
}

public class Company {
    private String name;
    public String getName() {
        return name;
    }
    ...
}

然後綜合使用 Optional API 重構程式碼如下：

Optional<Staff> staffOpt =...;
staffOpt
        .flatMap(Staff::getDepartment)
        .flatMap(Department::getCompany)
        .map(Company::getName)
        .orElse("Unknown");

可以看到重構之後程式碼利用 Optional 的 Fluent Interface，以及 lambda 表示式，使程式碼更加流暢連貫，並且提高程式碼整體可讀性。

四、幫助文章

1、Tired of Null Pointer Exceptions? Consider Using Java SE 8's Optional!
3、Optionals: Patterns and Good Practices
3、Java8 in Action

歡迎關注我的公眾號：程式通事，獲得日常乾貨推送。如果您對我的專題內容感興趣，也可以關注我的部落格：studyidea.cn

相關推薦