Java使用Optional與Stream來取代if判空邏輯（JDK8以上）

Java使用Optional與Stream來取代if判空邏輯（JDK8以上）

經過本文你能夠用很是簡短的代碼替代業務邏輯中的判null校驗，而且很容易的在出現空指針的時候進行打日誌或其餘操做。

注：若是對Java8新特性中的lambda表達式與Stream不熟悉的能夠去補一下基礎，瞭解概念。

首先下面代碼中的List放入了不少Person對象，其中有的對象是null的，若是不加校驗調用Person的getXXX()方法確定會報空指針錯誤，通常咱們採起的方案就是加上if判斷：

public class DemoUtils {

    public static void main(String[] args) {
        List<Person> personList = new ArrayList<>();
        personList.add(new Person());
        personList.add(null);
        personList.add(new Person("小明",10));
        personList.add(new Person("小紅",12));
        
        for (Person person : personList) {
        //if判空邏輯
            if (person != null) {
                System.out.println(person.getName());
                System.out.println(person.getAge());
            }
        }
    }

    static class Person {
        private String name;
        private int age;

        public Person() {
        }

        public Person(String name, int age) {
            this.name = name;
            this.age = age;
        }

        public String getName() {
            return name;
        }
        public void setName(String name) {
            this.name = name;
        }
        public int getAge() {
            return age;
        }
        public void setAge(int age) {
            this.age = age;
        }
    }
}

其實，Java新特性Stream API 與 Optional 提供了更加優雅的方法：

利用Stream API 中的 filter將隊列中的空對象過濾掉，filter(Objects::nonNull)的意思是，list中的每一個元素執行Objects的nonNull()方法，返回false的元素被過濾掉，保留返回true的元素。

public static void main(String[] args) {
        List<Person> personList = new ArrayList<>();
        personList.add(new Person());
        personList.add(null);
        personList.add(new Person("小明",10));
        personList.add(new Person("小紅",12));

        personList.stream().filter(Objects::nonNull).forEach(person->{
            System.out.println(person.getName());
            System.out.println(person.getAge());
        });
    }

示例中的personList自己也可能會null，若是業務邏輯中要求personList爲null時打日誌報警，能夠用Optional優雅的實現:

public static void main(String[] args) {
    List<Person> personList = new ArrayList<>();
    personList.add(new Person());
    personList.add(null);
    personList.add(new Person("小明", 10));
    personList.add(new Person("小紅", 12));

    Optional.ofNullable(personList).orElseGet(() -> {
        System.out.println("personList爲null！");
        return new ArrayList<>();
    }).stream().filter(Objects::nonNull).forEach(person -> {
        System.out.println(person.getName());
        System.out.println(person.getAge());
    });
}

代碼中的

orElseGet(() -> {
    //代替log
    System.out.println("personList爲null！");
    return new ArrayList<>();
})

表示若是personList爲null，則執行這2句代碼，返回一個不含元素的List，這樣當personList爲null的時候不會報空指針錯誤，而且還打了日誌。