Java8 Stream 的核心秘密

news2024/9/20 17:35:49

小伙伴们好呀，我是 4ye，今天来分享下 Java8 Stream 的源码

核心回顾

stream 是一次性的，不是数据结构，不存储数据，不改变源数据.。
API 分为终端和中间操作，中间操作是惰性的，碰到终端才去执行。
中间操作有无状态和有状态之分，有状态需要更改上一步操作获得的所有元素，才可以进行下一步操作，比如排序 sorted，去重 distinct，跳过 skip，限制 limit 这四个，需要多迭代一次。
终端操作有短路与否之分，短路操作有 anyMatch, allMatch, noneMatch, findFirst, findAny

整体概览

这里列出一些重要的类，是看源码过程中必须了解的。

比如：

Sink 接口是数据实际操作的地方，核心方法： begin，accept，end
AbstractPipeline 抽象类是核心的数据结构，双链表

demo代码

这里沿用上文的例子

        Student aStud = new Student(1, "a");
        Student bStud = new Student(2, "b");
        Student cStud = new Student(3, null);

//         集合的创建 一
        List<Student> collect1 = Stream.of(aStud, bStud, cStud).collect(Collectors.toList());
        collect1.forEach(System.out::println);

        List<String> studNameList = studentList.stream()
                .map(Student::getName)
                .filter(Objects::nonNull)
                .map(String::toUpperCase)
                .sorted()
                .map(e -> e + "c")
                .collect(Collectors.toList());
复制代码

步骤解析

都在这里了 👇

这里步骤太多了，就不一一放出来了，列下核心

wrapSink, 创建 Sink 链，将管道的 Sink 操作连接在一起
copyInto , 处理数据

wrapSink()

开始套娃，从 ReducingSink 往前套

opWarpSink 方法调用的是每一步 中间操作 中的方法

通过 单链表 的形式将他们联系在一起

Sink 链创建结果

copyInto()

这里判断是不是 短路操作 ，然后就去执行 Sink 的 begin，accept，end 方法。

通过 forEachRemaining 进行内部迭代，这个是 Spliterator 的方法。

map 链节点，直接调用传进来的方法，

filter 链节点，多一步判断

sorted 节点，添加到 list 中。

注意，此时没有继续调用 downstream.accept 方法！

意味着，我们代码中的 5 个中间步骤只执行了前 3 个。

不过别担心， sorted 链节点中它重写了这个 end，并开启对新数据的新一轮遍历！

这就是我们提到的，有状态中间操作多一次迭代的原因

最后呢，是来到终止操作 TerminalOp 中的 accept，这里执行的是 list 的 add 方法（我们调用 Collectors.toList() 中构建的），至此，数据添加到 state 中

获取数据，ReducingSink 继承了 Box 这个抽象类，最后 get 方法得到结果。

总结

代码对应的执行流程👇

先创建流，出现了 Head 节点
创建中间管道 Pipeline
调用终端操作后有三步 👇

（一）将中间管道的 Sink 操作连接在一起（wrapSink）（二）处理数据（copyInto），主要调用 Sink 中的 begin，accept（核心），end 操作（三）返回结果，ReducingSink 中的 get 方法

主要记住这个 wrapSink 方法和 copyInto 方法。

一个套娃，一个调用 begin，accept，end 等方法。

那么，这个 stream 的原理机制就出来了：

利用 wrapSink 方法将各个中间操作中的 Sink 嵌套在一起，然后来到 copyInto 方法，调用 begin 通知各个 Sink 做好准备，接着进行内部迭代调用 accept 方法，再调用 end 方法完成数据的操作，最后通过 get 方法，获取新容器中的数据，便是结果了。

此外，源码的 链式调用API 写法 ，设计模式 的使用以及泛型，四大函数式接口 组合构建的高度抽象，封装写法，对我们的编码能力，源码阅读能力也有很大的帮助！

比如这个 Consumer+Function 接口的组合，配合泛型上下限的使用