在java 8 stream表達式中實現if/else邏輯

目錄

• 簡介
• 傳統寫法
• 使用filter
• 總結

簡介

在Stream處理中，我們通常會遇到if/else的判斷情況，對於這樣的問題我們怎么處理呢？

還記得我們在上一篇文章lambda最佳實踐中提到，lambda表達式應該越簡潔越好，不要在其中寫臃腫的業務邏輯。

接下來我們看一個具體的例子。

傳統寫法

假如我們有一個1 to 10的list，我們想要分別挑選出奇數和偶數出來，傳統的寫法，我們會這樣使用：

    public void inForEach(){
        List<Integer> ints = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);

        ints.stream()
                .forEach(i -> {
                    if (i.intValue() % 2 == 0) {
                        System.out.println("i is even");
                    } else {
                        System.out.println("i is old");
                    }
                });
    }

上面的例子中，我們把if/else的邏輯放到了forEach中，雖然沒有任何問題，但是代碼顯得非常臃腫。

接下來看看怎么對其進行改寫。

使用filter

我們可以把if/else的邏輯改寫為兩個filter：

List<Integer> ints = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);

        Stream<Integer> evenIntegers = ints.stream()
                .filter(i -> i.intValue() % 2 == 0);
        Stream<Integer> oddIntegers = ints.stream()
                .filter(i -> i.intValue() % 2 != 0);

有了這兩個filter，再在filter過后的stream中使用for each：

        evenIntegers.forEach(i -> System.out.println("i is even"));
        oddIntegers.forEach(i -> System.out.println("i is old"));

怎么樣，代碼是不是非常簡潔明了。

總結

lambda表達式需要盡可能的簡潔，我們可以用stream的filter來替代if/else業務邏輯。

本文的例子https://github.com/ddean2009/learn-java-streams/tree/master/lambda-if-else

歡迎關注我的公眾號:程序那些事，更多精彩等着您！
更多內容請訪問 www.flydean.com