Java 16 中新增的 Stream 接口的一些思考

这里先提一个题外话，如果想看 JDK 不同版本之间有何差异，增加或者删除了哪些 API，可以通过下面这个链接查看：

• https://javaalmanac.io/jdk/17...

路径中的两个版本就是要对比的两个版本，其界面如下：

同时，我们也可以通过 JDK 内置 jdeps 工具查找过期以及废弃API以及对应的替换

jdeps --jdk-internals -R --class-path 'libs/*' $project

libs是你的所有依赖的目录，$project是你的项目jar包，示例输出：

...
JDK Internal API                         Suggested Replacement
----------------                         ---------------------
sun.misc.BASE64Encoder                   Use java.util.Base64 @since 1.8
sun.reflect.Reflection                   Use java.lang.StackWalker @since 9

Java 16 中针对 Stream API 有两个更新：

增加了 mapMulti 和 toList 这两个 API。

mapMulti

mapMulti 其实主要是对现有的 flatMap 在某些场景使用起来不够合适的补充。flatMap 是将一个对象映射为多个对象之后继续 Stream，例如将 List<List<Integer>> 里面的每一个数字取出，转换成一个新的 List<Integer>：

integerLists.stream().flatMap(integers -> integers.stream()).collect(Collectors.toList());

这对于每个元素本身就是集合类型的场景来说，非常适用。我们再来看一个例子，假设有邮件这个 Record 类，包含 id，以及发送到的邮箱和抄送到的邮箱：

record Mail(int id, Set<String> sendTo, Set<String> cc) {}

我们想找到一批邮件的所有不同的联系人，最后放到一个 List 中，可能会这么写：

Set<String> collect = mails.stream().flatMap(mail -> {
    Set<String> result = new HashSet<>();
    result.addAll(mail.sendTo());
    result.addAll(mail.cc());
    return result.stream();
}).collect(Collectors.toSet());

但是，这样写显然很不优雅，首先是对于每一个 Mail 都创建了额外的 Set 和对应的 Stream，并且，对于每个 mail 的 sendTo 还有 cc 都遍历了两遍（addAll 一遍，后续 Stream 又一遍）。其实我们的目前只是将 mail 中的 cc 以及 sendTo 取出来，用于参与后续的 Stream。在这种场景下，就非常适合用 mapMulti：

Set<String> collect = mails.stream().<String>mapMulti((mail, consumer) -> {
    mail.cc().forEach(consumer::accept);
    mail.sendTo().forEach(consumer::accept);
}).collect(Collectors.toSet());

可以看出：

• mapMulti 的入参是一个 BiConsumer，其实就是使用其参数中的 consumer 接收参与 Stream 后续的对象
• mapMulti 的思路就是将参数中的需要参与后续 Stream 的对象传入 consumer 来继续 Stream
• consumer 没有限制对象类型，想要限制必须加上形参 <String> 否则最后返回的是 Set<Object> 而不是 Set<String>

假设 mail 的 sendTo 还有 cc 都需要去其他地方获取，使用 mapMulti 还可以实现：

Set<String> collect = mailIds.stream().<String>mapMulti((mailId, consumer) -> {
    mailService.getCCById(mailId).forEach(consumer::accept);
    mailService.getSendToById(mailId).forEach(consumer::accept);
}).collect(Collectors.toSet());

还有一些比较有意思的用法，例如混合类型的 List 转换成统一类型：

class C {
     static void expandIterable(Object e, Consumer<Object> c) {
         if (e instanceof Iterable<?> elements) {
             for (Object ie : elements) {
                 expandIterable(ie, c);
             }
         } else if (e != null) {
             c.accept(e);
         }
     }
    
     public static void main(String[] args) {
         var nestedList = List.of(1, List.of(2, List.of(3, 4)), 5);
         Stream<Object> expandedStream = nestedList.stream().mapMulti(C::expandIterable);
     }
}

活用 Optional.ifPresent(Consumer<? super T> action) 方法：

Stream.of(Optional.of("0"), Optional.of("1"), Optional.empty())
    .mapMulti(Optional::ifPresent)
    .forEach(System.out::print);

toList

对于 Stream 增加了 toList 直接转换成 List，由于不涉及 collect 里面的截断操作，所以比 collect 占用的内存更小，需要的操作更少并且更快。

之前转换成 List，需要 collect(Collectors.toList())，生成的 List 是 ArrayList，是可变的

但是这次新加的 Api，toList 生成的是 UnmodifiableList，是不可变的。

所以这两个 API 不能直接互相替换，需要做一些检查确认没有更改才能替换。

微信搜索“我的编程喵”关注公众号，每日一刷，轻松提升技术，斩获各种offer：