面試官問:Stream 中的 map、peek、foreach 方法的區(qū)別?徹底懵了......
原代碼是這樣的:
List<Menu> children = all.stream().filter(...).map(
(m) -> {
m.setChildList(getChildrens(m, all));
return m;
}
).collect(Collectors.toList());
其中 stream 用的 map 映射,其實(shí)更建議把 map 修改為 peek。
你可能會(huì)有這些疑問:
- 為什么要把 map 換成 peek 呢?怎么改?
- map 和 peek 有什么區(qū)別?
- peek 和 foreach 有什么區(qū)別?
看到這,你是不是徹底懵了,沒問題,本篇棧長就來強(qiáng)勢分析下!
另外,這些問題是 Java 程序員面試過程中必問的,出場率賊高,Java 程序員必懂,這些題我也都整理到了小程序中,歡迎前往小程序刷題。
peek
map 和 peek 都是 Stream 提供的流處理方法。
This method exists mainly to support debugging, where you want to see the elements as they flow past a certain point in a pipeline:
翻譯:
這個(gè)方法主要用于支持 debug 調(diào)試,當(dāng)你想看處于某個(gè)特定點(diǎn)的流元素時(shí)
如:
@Test
public void () {
Stream.of("one", "two", "three", "four")
.filter(e -> e.length() > 3)
.peek(e -> System.out.println("Filtered value: " + e))
.map(String::toUpperCase)
.peek(e -> System.out.println("Mapped value: " + e))
.collect(Collectors.toList());
}
輸出結(jié)果:
Filtered value: three
Mapped value: THREE
Filtered value: four
Mapped value: FOUR
先后輸出 filter、map 之后的流元素,實(shí)際工作中如果想看某個(gè)過程的結(jié)果,可以派上用場。
可以看到,map 接收 Function 函數(shù)式接口參數(shù)(接收一個(gè)參數(shù),返回一個(gè)參數(shù)),peek 接收 Consumer 函數(shù)式接口參數(shù)(接收一個(gè)參數(shù),無返回)。
不理解的話來看下面的示例:
假如有以下 List:
private List<String> languageList = new ArrayList<String>() {{
add("java");
add("python");
add("c++");
add("php");
add("go");
}};
peek 方法中的函數(shù)式接口參數(shù)不能有返回值:
意味著它不能像 map 一樣處理流中的元素然后形成新流:
peek 不能修改流中的元素,只能對元素進(jìn)行打印輸出或者其他外部處理操作。
但流元素如果是引用類型,peek 卻可以達(dá)到 map 的效果:
private List<User> userList = new ArrayList<User>() {{
add(new User("張三"));
add(new User("李四"));
add(new User("王五"));
add(new User("趙六"));
}};
@Test
public void () {
userList.stream()
.peek(user -> user.setName("peek: " + user.getName()))
.forEach(System.out::println);
}
輸出結(jié)果:
SteamPeekTest.User(name=peek: 張三)
SteamPeekTest.User(name=peek: 李四)
SteamPeekTest.User(name=peek: 王五)
SteamPeekTest.User(name=peek: 趙六)
雖然不能有返回值形成新的流,但卻可以修改引用類型字段的值。
這也是粉絲建議的為什么要把 map 換成 peek 了,因?yàn)槭且妙愋?,使?peek 就沒必要 set 之后還要進(jìn)行 return 了。
List<Menu> children = all.stream().filter(...).map(
(m) -> {
m.setChildList(getChildrens(m, all));
return m;
}
).collect(Collectors.toList());
修改為:
List<Menu> children = all.stream().filter(...).peek(
m -> m.setChildList(getChildrens(m, all))
).collect(Collectors.toList());
是不是優(yōu)雅多了?
如 foreach 的源碼:
和 peek 一樣也是接收 Consumer 參數(shù),不同是 foreach 沒有返回參數(shù),意味著 foreach 會(huì)中斷流操作,只能用來遍歷,不能再進(jìn)行后續(xù)的流處理。
總結(jié)
根據(jù)文中的示例,大家應(yīng)該都搞清楚了 map、peek、foreach 的區(qū)別和用法了,現(xiàn)在再來總結(jié)下吧!
map:用于對流中的每個(gè)元素進(jìn)行映射處理,然后再形成新的流;
peek:用于 debug 調(diào)試流中間結(jié)果,不能形成新的流,但能修改引用類型字段的值;
foreach:用于遍歷,會(huì)中斷流操作;
本文所有完整示例源代碼已經(jīng)上傳:
https://github.com/javastacks/javastack
歡迎 Star 學(xué)習(xí),后面 Java 示例都會(huì)在這上面提供!