Java8 Stream這樣理解,事半功倍

之前看了許多介紹Java8 Stream的文章,但是初次接觸真的是難以理解(我悟性比較低),沒辦法只能"死記硬背",但是昨天我打王者榮耀(那一局我贏了,牛魔全場MVP)的時候,突然迸發了靈感,感覺之前沒有理解透徹的一下子就理解透徹了。所以決定用簡單的方式來回憶下我認爲的java8 Stream.

lambda表達式

語法

lambda表達式是Stream API的基石,所以想要學會Stream API的使用,必須先要理解lambda表達式,這裏對lambda做一個簡單回顧。

我們常常會看到這樣的代碼

Arrays.sort(new Integer[]{1, 8, 7, 4}, new Comparator<Integer>() {
   @Override
   public int compare(Integer first, Integer second) {
       return first.compareTo(second);
   }
});

上面這種寫法就是使用了匿名類,我們經常會使用匿名類的方式,因爲我們只運行一次,不想它一直存在。雖然說lambda表達式是爲了什麼所謂的函數式編程,也是大家在社區千呼萬喚纔出來的,但是在我看來就是爲了方(偷)便(懶)。

上面的代碼寫着麻煩,但是轉換成下面這樣的呢?

Arrays.sort(new Integer[]{1, 8, 7, 4},
	(first,second) -> first.compareTo(second));

這樣看着多清爽,而且把一些不必要的細節都屏蔽了。對於這種只包含一個抽象方法的接口,你可以通過lambda接口來創建該接口的對象,這種接口被稱爲函數式接口。

lambda表達式引入了一個新的操作符:->,它把lambda表達式分爲了2部分

(n) -> n*n

左側指定表達式所需的參數,如果不需要參數,也可以爲空。右側是lambda代碼塊,它指定lambda表達式的動作。

需要注意的是如果方法中只有一個返回的時候不用聲明,默認會返回。如果有分支返回的時候需要都進行聲明。

(n) -> {
	if( n <= 10) 
		return n*n;
	return n * 10;
}

方法引用以及構造器引用

方法引用

有些時候,先要傳遞給其他代碼的操作已經有實現的方法了。比如GUI中先要在按鈕被點擊時打印event對象,那麼可以這樣調用

button.setOnAction(event -> System.out.println(event));

這個時候我想偷懶,我不想寫event參數,因爲只有一個參數,jvm不能幫幫我嗎?下面是修改好的代碼


button.setOnAction(System.out::println);

表達式System.out::println是一個方法引用,等同於lambda表達式x -> System.out.println(x)。**:😗*操作符將方法名和對象或類的名字分割開來,以下是三種主要的使用情況:

  1. 對象::實例方法
  2. 類::靜態方法
  3. 類::實例方法

前兩種情況,方法引用等同於提供方法參數的lambda表達式。比如Math::pow ==== (x,y) -> Math.pow(x,y)

第三種情況,第一個參數會稱爲執行方法的對象。比如String::compareToIgnoreCase ==== (x,y) -> x.compareToIgnoreCase(y)

還有this::equals ==== x -> this.equals(x),super::equals ==== super.equals(x)

構造器引用

List<String> strList = Arrays.asList("1","2","3");
Stream<Integer> stream =  strList.stream().map(Integer::new);

上面代碼的Integer::new就是構造器引用,不同的是在構造器引用中方法名是new。如果存在多個構造器,編譯器會從上下文推斷並找出合適的那一個。

StreamAPI

Stream這個單詞翻譯過來就是流的意思,溪流的流,水流的流。

Stream

在我看來stream就像是上面的圖一樣,最開始的數據就是小水滴,它經過各種"攔截器"的處理之後,有的小水滴被丟棄,有的變大了,有的加上了顏色,有的變成了三角形。最後它們都變成了帶有顏色的圓。最後被我們放到結果集中。我們很多時候寫的代碼是這樣的:遍歷一個集合,然後對集合的元素進行判斷或者轉換,滿足條件的加入到新的集合裏面去,這種處理方式就和上面的圖是一樣的。先來看一段代碼

Map<String,Map<String,Integer>> resultMap = new HashMap<>();
Map<String,Integer> maleMap = new HashMap<>();
Map<String,Integer> femaleMap = new HashMap<>();

resultMap.put("male", maleMap);
resultMap.put("female",femaleMap);

for(int i = 0; i < list.size(); i++) {
    Person person = list.get(i);
    String gender = person.getGender();
    String level = person.getLevel();
    switch (gender) {
        case "male":
            Integer maleCount;
            if("gold".equals(level)) {
                maleCount = maleMap.get("gold");
                maleMap.put("gold", null != maleCount ? maleCount + 1 : 1);
            } else if("soliver".equals(level)){
                maleCount = maleMap.get("soliver");
                maleMap.put("soliver", null != maleCount ? maleCount + 1 : 1);
            }
            break;

        case "female":
            Integer femaleCount;
            if("gold".equals(level)) {
                femaleCount = femaleMap.get("gold");
                femaleMap.put("gold", null != femaleCount ? femaleCount + 1 : 1);
            } else if("soliver".equals(level)){
                femaleCount = femaleMap.get("soliver");
                femaleMap.put("soliver", null != femaleCount ? femaleCount + 1 : 1);
            }
            break;

    }
}

上面的代碼作用是統計不同性別的工程師職級的人數,在Java StreamAPI出來之前,這樣類似的業務代碼在系統中應該是隨處可見的,手打上面的代碼我大概花了兩分鐘,有了Stream之後,我偷了個懶

Map<String,Map<String,Integer>> result = list.stream().collect(
    Collectors.toMap(
	    person -> person.getGender(),
	    person -> Collections.singletonMap(person.getLevel(), 1),
	    (existValue,newValue) -> {
	        HashMap<String,Integer> newMap = new HashMap<>(existValue);
	        newValue.forEach((key,value) ->{
	            if(newMap.containsKey(key)) {
	                newMap.put(key, newMap.get(key) + 1);
	            } else {
	                newMap.put(key, value);
	            }
	        });
	        return newMap;
	    })
);

或者改成這樣的代碼

Map<String,Map<String,Integer>> result =  stream.collect(
    Collectors.groupingBy(
        Person::getGender, 
        Collectors.toMap(
            person->person.getLevel(), 
            person -> 1,
            (existValue,newValue) -> existValue + newValue
        )
    )
);

不僅代碼塊減少了許多,甚至邏輯也更清晰了。真的是用stream一時爽,一直用一直爽呀。

Stream作爲流,它可以是有限的可以是無限的,當然我們用得最多的還是有限的流(for循環就是有限的流),如上面那張圖一樣,我們可以對流中的元素做各種各樣常見的處理。比如求和,過濾,分組,最大值,最小值等常見處理,所以現在就開始使用Stream吧

Stream的特性

  1. Stream自己不會存儲元素,元素可能被存儲在底層集合中,或者被生產出來。
  2. Stream操作符不會改變源對象,相反,他們會返回一個持有新對象的stream
  3. Stream操作符是延遲執行的,可能會等到需要結果的時候纔去執行。

Stream API

函數式接口 參數類型 返回類型 抽象方法名 描述 其他方法
Runnable void run 執行一個沒有參數和返回值的操作
Supplier<T> T get 提供一個T類型的值
Counsumer<T> T void accept 處理一個T類型的值 chain
BiConsumer<T,U> T,U void accept 處理T類型和U類型的值 chain
Function<T,R> T R apply 一個參數類型爲T的函數 compose,andThen,identity
BiFunction<T,U,R> T,U R apply 一個參數類型爲T和U的函數 andThen
UnaryOperator<T> T T apply 對類型T進行的一元操作 compose,andThen,identity
BinaryOperator<T> T,T T apply 對類型T進行二元操作 andThen
Predicate<T> T boolean test 一個計算boolean值的函數 And,or,negate,isEqual
BiPredicate<T,U> T,U boolean test 一個含有兩個參數,計算boolean值的函數 and,or,negate

map()和flatMap()的區別

使用map方法的時候,相當於對每個元素應用一個函數,並將返回的值收集到新的Stream中。

Stream<String[]>	-> flatMap ->	Stream<String>
Stream<Set<String>>	-> flatMap ->	Stream<String>
Stream<List<String>>	-> flatMap ->	Stream<String>
Stream<List<Object>>	-> flatMap ->	Stream<Object>

{{1,2}, {3,4}, {5,6} } -> flatMap -> {1,2,3,4,5,6}

中間操作以及結束操作

Stream上的所有操作分爲兩類:中間操作和結束操作,中間操作只是一種標記(調用到這類方法,並沒有真正開始流的遍歷。),只有結束操作纔會觸發實際計算。簡單的說就是API返回值仍然是Stream的就是中間操作,否則就是結束操作。

如何debug

  1. 請使用代碼段,比如IntStream.of(1,2,3,4,5).fiter(i -> {return i%2 == 0;})將斷點打在代碼段上即可。
  2. 引用方法也可以進行調試,在isDouble中打上斷點比如IntStream.of(1,2,3,4,5).fiter(MyMath::isDouble)

那些不好理解的API

  1. reduce()
    我們以前做累加是如何完成的呢?

int sum = 0;
for(int value in values) {
	sum = sum + value;
}

現在改成stream的方式來實現

values.stream().reduce(Integer::sum);

這個reduce()方法就是一個二元函數:從流的前兩個元素開始,不斷將它應用到流中的其他元素上。

如何寫好Stream代碼

stream API就是爲了方便而設計的,在sql層面並不方便處理的數據可以通過stream來實現分組,聚合,最大值,最小值,排序,求和等等操作。所以不要把它想得太複雜,只管寫就好了。總有那麼一天你熟練了就可以寫出簡潔得代碼。或者從現在開始把你項目中的大量for循環改造成stream方式。

代碼示例

本來想寫大段代碼來樣式到stream API的轉換,但是想了想完全沒有必要,github上找了hutool工具類的部分代碼來完成轉換示例。(可以通過這種方式來提高stream api的能力)

  1. 計算每個元素出現的次數(請先想象下jdk7怎麼實現)
代碼效果:[a,b,c,c,c]  -> a:1,b:1,c:3

Arrays.asList("a","b","c","c","c").stream().collect(Collectors.groupingBy(str->str, Collectors.counting()));
  1. 以特定分隔符將集合轉換爲字符串,並添加前綴和後綴(請先想象下jdk7怎麼實現)
List<String> myList = Arrays.asList("a","b","c","c","c");
myList.stream().collect(Collectors.joining(",","{","}"));

  1. 判斷列表不全爲空(請先想象下jdk7怎麼實現)
myList.stream().anyMatch(s -> !s.isEmpty());

點贊關注不迷路

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章