面试官问：Stream 中的 map、peek、foreach 方法的区别？彻底懵了。。

背景

之前栈长在公众号Java技术栈分享了这篇文章：

[别再写一堆的 for 循环了！Java 8 中的 Stream 轻松遍历树形结构，是真的牛逼！]

原代码是这样的：

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 程序员必懂，这些题我也都整理到了Java面试库小程序中，欢迎前往小程序刷题。

peek

map 和 peek 都是 Stream 提供的流处理方法。

首先看 peek 的使用源码注释：

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 peekTest1() {
    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 和 peek 的方法参数：

可以看到，map 接收 Function 函数式接口参数（接收一个参数，返回一个参数），peek 接收 Consumer 函数式接口参数（接收一个参数，无返回）。

不理解的话来看下面的示例：

假如有以下 List：

private List<String> languageList = new ArrayList<String>() {{
    add("java");
    add("python");
    add("c++");
    add("php");
    add("go");
}};

peek 方法中的函数式接口参数不能有返回值：

意味着它不能像 map 一样处理流中的元素然后形成新流：

map 的详细用法就不介绍了，不清楚的可以看栈长分享的这篇：

[Java 8 map 和 flatMap 的区别？]

更多 Java 8 系列教程可以关注公众号Java技术栈，在公众号菜单中阅读，我都已经整理好了，希望对大家有帮助。

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 peekTest3() {
    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());

是不是优雅多了？

peek 和 foreach 有什么区别？

如 foreach 的源码：

和 peek 一样也是接收 Consumer 参数，不同是 foreach 没有返回参数，意味着 foreach 会中断流操作，只能用来遍历，不能再进行后续的流处理。

总结

根据文中的示例，大家应该都搞清楚了 map、peek、foreach 的区别和用法了，现在再来总结下吧！

• map：用于对流中的每个元素进行映射处理，然后再形成新的流；
• peek：用于 debug 调试流中间结果，不能形成新的流，但能修改引用类型字段的值；
• foreach：用于遍历，会中断流操作；

所以说，大家都搞清楚了吧？还有谁用错，把这篇文章发给他吧，让大家少走弯路，少写垃圾代码，共同进步。

本文由哈喽比特于1年以前收录，如有侵权请联系我们。
文章来源：https://mp.weixin.qq.com/s/iJDyjdXDkJx95MpqZmUT3g