背景

之前栈长在公众号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:用于遍历,会中断流操作;

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

本文所有完整示例源代码已经上传:

https://github.com/javastacks/javastack

欢迎 Star 学习,后面 Java 示例都会在这上面提供!

好了,今天的分享就到这里了,后面栈长会分享更多好玩的 Java 技术和最新的技术资讯,关注公众号Java技术栈第一时间推送,我也将主流 Java 面试题和参考答案都整理好了,在公众号后台回复关键字 "面试" 进行刷题。

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注