原代码是这样的:
List<Menu> children = all.stream().filter(...).map(
(m) -> {
m.setChildList(getChildrens(m, all));
return m;
}
).collect(Collectors.toList());
其中 stream 用的 map 映射,其实更建议把 map 修改为 peek。
你可能会有这些疑问:
- 为什么要把 map 换成 peek 呢?怎么改?
- map 和 peek 有什么区别?
- peek 和 foreach 有什么区别?
看到这,你是不是彻底懵了,没问题,本篇栈长就来强势分析下!
另外,这些问题是 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:
翻译:
这个方法主要用于支持 debug 调试,当你想看处于某个特定点的流元素时
如:
@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());
}
输出结果:
Filtered value: three
Mapped value: THREE
Filtered value: four
Mapped value: FOUR
先后输出 filter、map 之后的流元素,实际工作中如果想看某个过程的结果,可以派上用场。
可以看到,map 接收 Function 函数式接口参数(接收一个参数,返回一个参数),peek 接收 Consumer 函数式接口参数(接收一个参数,无返回)。
不理解的话来看下面的示例:
假如有以下 List:
private List<String> languageList = new ArrayList<String>() {{
add("java");
add("python");
add("c++");
add("php");
add("go");
}};
peek 方法中的函数式接口参数不能有返回值:
意味着它不能像 map 一样处理流中的元素然后形成新流:
peek 不能修改流中的元素,只能对元素进行打印输出或者其他外部处理操作。
但流元素如果是引用类型,peek 却可以达到 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);
}
输出结果:
SteamPeekTest.User(name=peek: 张三)
SteamPeekTest.User(name=peek: 李四)
SteamPeekTest.User(name=peek: 王五)
SteamPeekTest.User(name=peek: 赵六)
虽然不能有返回值形成新的流,但却可以修改引用类型字段的值。
这也是粉丝建议的为什么要把 map 换成 peek 了,因为是引用类型,使用 peek 就没必要 set 之后还要进行 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());
是不是优雅多了?
如 foreach 的源码:
和 peek 一样也是接收 Consumer 参数,不同是 foreach 没有返回参数,意味着 foreach 会中断流操作,只能用来遍历,不能再进行后续的流处理。
总结
根据文中的示例,大家应该都搞清楚了 map、peek、foreach 的区别和用法了,现在再来总结下吧!
map:用于对流中的每个元素进行映射处理,然后再形成新的流;
peek:用于 debug 调试流中间结果,不能形成新的流,但能修改引用类型字段的值;
foreach:用于遍历,会中断流操作;
本文所有完整示例源代码已经上传:
https://github.com/javastacks/javastack
欢迎 Star 学习,后面 Java 示例都会在这上面提供!