Java封神之路-stream流-入門
單從“Stream”這個單詞上來看,它似乎和 java.io 包下的 InputStream 和 OutputStream 有些關係。實際上呢,沒毛關係。Java 8 新增的 Stream 是為了解放程式設計師操作集合(Collection)時的生產力,之所以能解放,很大一部分原因可以歸功於同時出現的 Lambda 表示式——極大的提高了程式設計效率和程式可讀性。
Stream 究竟是什麼呢?
Stream 就好像一個高階的迭代器,但只能遍歷一次,就好像一江春水向東流;在流的過程中,對流中的元素執行一些操作,比如“過濾掉長度大於 10 的字串”、“獲取每個字串的首字母”等。
要想操作流,首先需要有一個數據源,可以是陣列或者集合。每次操作都會返回一個新的流物件,方便進行鏈式操作,但原有的流物件會保持不變。
流的操作可以分為兩種型別:
1)中間操作,可以有多個,每次返回一個新的流,可進行鏈式操作。
2)終端操作,只能有一個,每次執行完,這個流也就用光光了,無法執行下一個操作,因此只能放在最後。
來舉個例子。
List<String> list = new ArrayList<>();
list.add("武漢加油");
list.add("中國加油");
list.add("世界加油");
list.add("世界加油");
long count = list.stream().distinct().count();
System.out.println(count);
distinct()
方法是一箇中間操作(去重),它會返回一個新的流(沒有共同元素)。
Stream<T> distinct();
count()
方法是一個終端操作,返回流中的元素個數。
long count();
中間操作不會立即執行,只有等到終端操作的時候,流才開始真正地遍歷,用於對映、過濾等。通俗點說,就是一次遍歷執行多個操作,效能就大大提高了。
理論部分就扯這麼多,下面直接進入實戰部分。
01、建立流
如果是陣列的話,可以使用Arrays.stream()
或者Stream.of()
建立流;如果是集合的話,可以直接使用stream()
方法建立流,因為該方法已經新增到 Collection 介面中。
public class CreateStreamDemo {
public static void main(String[] args) {
String[] arr = new String[]{"武漢加油", "中國加油", "世界加油"};
Stream<String> stream = Arrays.stream(arr);
stream = Stream.of("武漢加油", "中國加油", "世界加油");
List<String> list = new ArrayList<>();
list.add("武漢加油");
list.add("中國加油");
list.add("世界加油");
stream = list.stream();
}
}
檢視 Stream 原始碼的話,你會發現of()
方法內部其實呼叫了Arrays.stream()
方法。
public static<T> Stream<T> of(T... values) {
return Arrays.stream(values);
}
另外,集合還可以呼叫parallelStream()
方法建立併發流,預設使用的是ForkJoinPool.commonPool()
執行緒池。
List<Long> aList = new ArrayList<>();
Stream<Long> parallelStream = aList.parallelStream();
02、操作流
Stream 類提供了很多有用的操作流的方法,我來挑一些常用的給你介紹一下。
1)過濾
通過filter()
方法可以從流中篩選出我們想要的元素。
public class FilterStreamDemo {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("周杰倫");
list.add("王力巨集");
list.add("陶喆");
list.add("林俊杰");
Stream<String> stream = list.stream().filter(element -> element.contains("王"));
stream.forEach(System.out::println);
}
}
filter()
方法接收的是一個 Predicate(Java 8 新增的一個函式式介面,接受一個輸入引數返回一個布林值結果)型別的引數,因此,我們可以直接將一個 Lambda 表示式傳遞給該方法,比如說element -> element.contains("王")
就是篩選出帶有“王”的字串。
forEach()
方法接收的是一個 Consumer(Java 8 新增的一個函式式介面,接受一個輸入引數並且無返回的操作)型別的引數,類名 :: 方法名
是 Java 8 引入的新語法,System.out
返回 PrintStream 類,println 方法你應該知道是列印的。
stream.forEach(System.out::println);
相當於在 for 迴圈中列印,類似於下面的程式碼:
for (String s : strs) {
System.out.println(s);
}
很明顯,一行程式碼看起來更簡潔一些。來看一下程式的輸出結果:
王力巨集
2)對映
如果想通過某種操作把一個流中的元素轉化成新的流中的元素,可以使用map()
方法。
public class MapStreamDemo {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("周杰倫");
list.add("王力巨集");
list.add("陶喆");
list.add("林俊杰");
Stream<Integer> stream = list.stream().map(String::length);
stream.forEach(System.out::println);
}
}
map()
方法接收的是一個 Function(Java 8 新增的一個函式式介面,接受一個輸入引數 T,返回一個結果 R)型別的引數,此時引數 為 String 類的 length 方法,也就是把Stream<String>
的流轉成一個Stream<Integer>
的流。
程式輸出的結果如下所示:
3
3
2
3
3)匹配
Stream 類提供了三個方法可供進行元素匹配,它們分別是:
-
anyMatch()
,只要有一個元素匹配傳入的條件,就返回 true。 -
allMatch()
,只有有一個元素不匹配傳入的條件,就返回 false;如果全部匹配,則返回 true。 -
noneMatch()
,只要有一個元素匹配傳入的條件,就返回 false;如果全部匹配,則返回 true。
public class MatchStreamDemo {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("周杰倫");
list.add("王力巨集");
list.add("陶喆");
list.add("林俊杰");
boolean anyMatchFlag = list.stream().anyMatch(element -> element.contains("王"));
boolean allMatchFlag = list.stream().allMatch(element -> element.length() > 1);
boolean noneMatchFlag = list.stream().noneMatch(element -> element.endsWith("沉"));
System.out.println(anyMatchFlag);
System.out.println(allMatchFlag);
System.out.println(noneMatchFlag);
}
}
因為“王力巨集”以“王”字開頭,所以 anyMatchFlag 應該為 true;因為“周杰倫”、“王力巨集”、“陶喆”、“林俊杰”的字串長度都大於 1,所以 allMatchFlag 為 true;因為 4 個字串結尾都不是“沉”,所以 noneMatchFlag 為 true。
程式輸出的結果如下所示:
true
true
true
4)組合
reduce()
方法的主要作用是把 Stream 中的元素組合起來,它有兩種用法:
Optional<T> reduce(BinaryOperator<T> accumulator)
沒有起始值,只有一個引數,就是運算規則,此時返回Optional。
T reduce(T identity, BinaryOperator<T> accumulator)
有起始值,有運算規則,兩個引數,此時返回的型別和起始值型別一致。
來看下面這個例子。
public class ReduceStreamDemo {
public static void main(String[] args) {
Integer[] ints = {0, 1, 2, 3};
List<Integer> list = Arrays.asList(ints);
Optional<Integer> optional = list.stream().reduce((a, b) -> a + b);
Optional<Integer> optional1 = list.stream().reduce(Integer::sum);
System.out.println(optional.orElse(0));
System.out.println(optional1.orElse(0));
int reduce = list.stream().reduce(6, (a, b) -> a + b);
System.out.println(reduce);
int reduce1 = list.stream().reduce(6, Integer::sum);
System.out.println(reduce1);
}
}
運算規則可以是Lambda 表示式(比如(a, b) -> a + b
),也可以是類名::方法名(比如Integer::sum
)。
程式執行的結果如下所示:
0、1、2、3 在沒有起始值相加的時候結果為 6;有起始值 6 的時候結果為 12。
03、轉換流
既然可以把集合或者陣列轉成流,那麼也應該有對應的方法,將流轉換回去——collect()
方法就滿足了這種需求。
public class CollectStreamDemo {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("周杰倫");
list.add("王力巨集");
list.add("陶喆");
list.add("林俊杰");
String[] strArray = list.stream().toArray(String[]::new);
System.out.println(Arrays.toString(strArray));
List<Integer> list1 = list.stream().map(String::length).collect(Collectors.toList());
List<String> list2 = list.stream().collect(Collectors.toCollection(ArrayList::new));
System.out.println(list1);
System.out.println(list2);
String str = list.stream().collect(Collectors.joining(", ")).toString();
System.out.println(str);
}
}
toArray()
方法可以將流轉換成陣列,你可能比較好奇的是String[]::new
,它是什麼東東呢?來看一下toArray()
方法的原始碼。
<A> A[] toArray(IntFunction<A[]> generator);
也就是說String[]::new
是一個 IntFunction,一個可以產生所需的新陣列的函式,可以通過反編譯位元組碼看看它到底是什麼:
String[] strArray = (String[])list.stream().toArray((x$0) -> {
return new String[x$0];
});
System.out.println(Arrays.toString(strArray));
也就是相當於返回了一個指定長度的字串陣列。
當我們需要把一個集合按照某種規則轉成另外一個集合的時候,就可以配套使用map()
方法和collect()
方法。
List<Integer> list1 = list.stream().map(String::length).collect(Collectors.toList());
- 1
通過stream()
方法建立集合的流後,再通過map(String:length)
將其對映為字串長度的一個新流,最後通過collect()
方法將其轉換成新的集合。
Collectors 是一個收集器的工具類,內建了一系列收集器實現,比如說toList()
方法將元素收集到一個新的java.util.List
中;比如說toCollection()
方法將元素收集到一個新的java.util.ArrayList
中;比如說joining()
方法將元素收集到一個可以用分隔符指定的字串中。
來看一下程式的輸出結果:
[周杰倫, 王力巨集, 陶喆, 林俊杰]
[3, 3, 2, 3]
[周杰倫, 王力巨集, 陶喆, 林俊杰]
周杰倫, 王力巨集, 陶喆, 林俊杰