Java Stream API进阶篇
qq678770
8年前
<p>本节我们将仍然以 <em>Stream</em> 为例，介绍流的规约操作。</p> <p>规约操作（ <em>reduction operation</em> ）又被称作折叠操作（ <em>fold</em> ），是通过某个连接动作将所有元素汇总成一个汇总结果的过程。元素求和、求最大值或最小值、求出元素总个数、将所有元素转换成一个列表或集合，都属于规约操作。 <em>Stream</em> 类库有两个通用的规约操作 reduce() 和 collect() ，也有一些为简化书写而设计的专用规约操作，比如 sum() 、 max() 、 min() 、 count() 等。</p> <p>最大或最小值这类规约操作很好理解（至少方法语义上是这样），我们着重介绍 reduce() 和 collect() ，这是比较有魔法的地方。</p> <h2>多面手reduce()</h2> <p><em>reduce</em> 操作可以实现从一组元素中生成一个值， sum() 、 max() 、 min() 、 count() 等都是 <em>reduce</em> 操作，将他们单独设为函数只是因为常用。 reduce() 的方法定义有三种重写形式：</p> <ul> <li>Optional<T> reduce(BinaryOperator<T> accumulator)</li> <li>T reduce(T identity, BinaryOperator<T> accumulator)</li> <li><U> U reduce(U identity, BiFunction<U,? super T,U> accumulator, BinaryOperator<U> combiner)</li> </ul> <p>虽然函数定义越来越长，但语义不曾改变，多的参数只是为了指明初始值（参数 <em>identity</em> ），或者是指定并行执行时多个部分结果的合并方式（参数 <em>combiner</em> ）。 reduce() 最常用的场景就是从一堆值中生成一个值。用这么复杂的函数去求一个最大或最小值，你是不是觉得设计者有病。其实不然，因为“大”和“小”或者“求和"有时会有不同的语义。</p> <p>需求： <em>从一组单词中找出最长的单词</em> 。这里“大”的含义就是“长”。</p> <pre> <code class="language-java">// 找出最长的单词 Stream<String> stream = Stream.of("I", "love", "you", "too"); Optional<String> longest = stream.reduce((s1, s2) -> s1.length()>=s2.length() ? s1 : s2); //Optional<String> longest = stream.max((s1, s2) -> s1.length()-s2.length()); System.out.println(longest.get());</code></pre> <p>上述代码会选出最长的单词 <em>love</em> ，其中 <em>Optional</em> 是（一个）值的容器，使用它可以避免 <em>null</em> 值的麻烦。当然可以使用 Stream.max(Comparator<? super T> comparator) 方法来达到同等效果，但 reduce() 自有其存在的理由。</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/cff1c8d9391ed03c1dbfd50305872731.png"></p> <p>需求： <em>求出一组单词的长度之和</em> 。这是个“求和”操作，操作对象输入类型是 <em>String</em> ，而结果类型是 <em>Integer</em> 。</p> <pre> <code class="language-java">// 求单词长度之和 Stream<String> stream = Stream.of("I", "love", "you", "too"); Integer lengthSum = stream.reduce(0, // 初始值 // (1) (sum, str) -> sum+str.length(), // 累加器 // (2) (a, b) -> a+b); // 部分和拼接器，并行执行时才会用到 // (3) // int lengthSum = stream.mapToInt(str -> str.length()).sum(); System.out.println(lengthSum);</code></pre> <p>上述代码标号(2)处将i. 字符串映射成长度，ii. 并和当前累加和相加。这显然是两步操作，使用 reduce() 函数将这两步合二为一，更有助于提升性能。如果想要使用 map() 和 sum() 组合来达到上述目的，也是可以的。</p> <p>reduce() 擅长的是生成一个值，如果想要从 <em>Stream</em> 生成一个集合或者 <em>Map</em> 等复杂的对象该怎么办呢？终极武器 collect() 横空出世！</p> <h2>>>> 终极武器collect() <<<</h2> <p>不夸张的讲，如果你发现某个功能在 <em>Stream</em> 接口中没找到，十有八九可以通过 collect() 方法实现。 collect() 是 <em>Stream</em> 接口方法中最灵活的一个，学会它才算真正入门Java函数式编程。先看几个热身的小例子：</p> <pre> <code class="language-java">// 将Stream转换成容器或Map Stream<String> stream = Stream.of("I", "love", "you", "too"); List<String> list = stream.collect(Collectors.toList()); // (1) // Set<String> set = stream.collect(Collectors.toSet()); // (2) // Map<String, Integer> map = stream.collect(Collectors.toMap(Function.identity(), String::length)); // (3)</code></pre> <p>上述代码分别列举了如何将 <em>Stream</em> 转换成 <em>List</em> 、 <em>Set</em> 和 <em>Map</em> 。虽然代码语义很明确，可是我们仍然会有几个疑问：</p> <ol> <li>Function.identity() 是干什么的？</li> <li>String::length 是什么意思？</li> <li><em>Collectors</em> 是个什么东西？</li> </ol> <h2>接口的静态方法和默认方法</h2> <p><em>Function</em> 是一个接口，那么 Function.identity() 是什么意思呢？这要从两方面解释：</p> <ol> <li>Java 8允许在接口中加入具体方法。接口中的具体方法有两种， <em>default</em> 方法和 <em>static</em> 方法， identity() 就是 <em>Function</em> 接口的一个静态方法。</li> <li>Function.identity() 返回一个输出跟输入一样的Lambda表达式对象，等价于形如 t -> t 形式的Lambda表达式。</li> </ol> <p>上面的解释是不是让你疑问更多？不要问我为什么接口中可以有具体方法，也不要告诉我你觉得 t -> t 比 identity() 方法更直观。我会告诉你接口中的 <em>default</em> 方法是一个无奈之举，在Java 7及之前要想在定义好的接口中加入新的抽象方法是很困难甚至不可能的，因为所有实现了该接口的类都要重新实现。试想在 <em>Collection</em> 接口中加入一个 stream() 抽象方法会怎样？ <em>default</em> 方法就是用来解决这个尴尬问题的，直接在接口中实现新加入的方法。既然已经引入了 <em>default</em> 方法，为何不再加入 <em>static</em> 方法来避免专门的工具类呢！</p> <h2>方法引用</h2> <p>诸如 String::length 的语法形式叫做方法引用（ <em>method references</em> ），这种语法用来替代某些特定形式Lambda表达式。如果Lambda表达式的全部内容就是调用一个已有的方法，那么可以用方法引用来替代Lambda表达式。方法引用可以细分为四类：</p> <table> <thead> <tr> <th>方法引用类别</th> <th>举例</th> </tr> </thead> <tbody> <tr> <td>引用静态方法</td> <td>Integer::sum</td> </tr> <tr> <td>引用某个对象的方法</td> <td>list::add</td> </tr> <tr> <td>引用某个类的方法</td> <td>String::length</td> </tr> <tr> <td>引用构造方法</td> <td>HashMap::new</td> </tr> </tbody> </table> <p>我们会在后面的例子中使用方法引用。</p> <h2>收集器</h2> <p>相信前面繁琐的内容已彻底打消了你学习Java函数式编程的热情，不过很遗憾，下面的内容更繁琐。</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/c1e0d5bad68806ce6aa9de9ce738e4cd.png"></p> <p>收集器（ <em>Collector</em> ）是为 Stream.collect() 方法量身打造的工具接口（类）。考虑一下将一个 <em>Stream</em> 转换成一个容器（或者 <em>Map</em> ）需要做哪些工作？我们至少需要两样东西：</p> <ol> <li>目标容器是什么？是 <em>ArrayList</em> 还是 <em>HashSet</em> ，或者是个 <em>TreeMap</em> 。</li> <li>新元素如何添加到容器中？是 List.add() 还是 Map.put() 。</li> </ol> <p>如果并行的进行规约，还需要告诉 <em>collect()</em> 3. 多个部分结果如何合并成一个。</p> <p>结合以上分析， <em>collect()</em> 方法定义为 <R> R collect(Supplier<R> supplier, BiConsumer<R,? super T> accumulator, BiConsumer<R,R> combiner) ，三个参数依次对应上述三条分析。不过每次调用 <em>collect()</em> 都要传入这三个参数太麻烦，收集器 <em>Collector</em> 就是对这三个参数的简单封装,所以 <em>collect()</em> 的另一定义为 <R,A> R collect(Collector<? super T,A,R> collector) 。 <em>Collectors</em> 工具类可通过静态方法生成各种常用的 <em>Collector</em> 。举例来说，如果要将 <em>Stream</em> 规约成 <em>List</em> 可以通过如下两种方式实现：</p> <pre> <code class="language-java">// 将Stream规约成List Stream<String> stream = Stream.of("I", "love", "you", "too"); List<String> list = stream.collect(ArrayList::new, ArrayList::add, ArrayList::addAll);// 方式１ //List<String> list = stream.collect(Collectors.toList());// 方式2 System.out.println(list);</code></pre> <p>通常情况下我们不需要手动指定 <em>collect()</em> 的三个参数，而是调用 collect(Collector<? super T,A,R> collector) 方法，并且参数中的 <em>Collector</em> 对象大都是直接通过 <em>Collectors</em> 工具类获得。实际上传入的 <strong> 收集器的行为决定了 collect() 的行为 </strong> 。</p> <h2>使用collect()生成Collection</h2> <p>前面已经提到通过 collect() 方法将 <em>Stream</em> 转换成容器的方法，这里再汇总一下。将 <em>Stream</em> 转换成 <em>List</em> 或 <em>Set</em> 是比较常见的操作，所以 <em>Collectors</em> 工具已经为我们提供了对应的收集器，通过如下代码即可完成：</p> <pre> <code class="language-java">// 将Stream转换成List或Set Stream<String> stream = Stream.of("I", "love", "you", "too"); List<String> list = stream.collect(Collectors.toList()); // (1) Set<String> set = stream.collect(Collectors.toSet()); // (2)</code></pre> <p>上述代码能够满足大部分需求，但由于返回结果是接口类型，我们并不知道类库实际选择的容器类型是什么，有时候我们可能会想要人为指定容器的实际类型，这个需求可通过 Collectors.toCollection(Supplier<C> collectionFactory) 方法完成。</p> <pre> <code class="language-java">// 使用toCollection()指定规约容器的类型 ArrayList<String> arrayList = stream.collect(Collectors.toCollection(ArrayList::new));// (3) HashSet<String> hashSet = stream.collect(Collectors.toCollection(HashSet::new));// (4)</code></pre> <p>上述代码(3)处指定规约结果是 <em>ArrayList</em> ，而(4)处指定规约结果为 <em>HashSet</em> 。一切如你所愿。</p> <h2>使用collect()生成Map</h2> <p>前面已经说过 <em>Stream</em> 背后依赖于某种数据源，数据源可以是数组、容器等，但不能是 <em>Map</em> 。反过来从 <em>Stream</em> 生成 <em>Map</em> 是可以的，但我们要想清楚 <em>Map</em> 的 <em>key</em> 和 <em>value</em> 分别代表什么，根本原因是我们要想清楚要干什么。通常在三种情况下 collect() 的结果会是 <em>Map</em> ：</p> <ol> <li>使用 Collectors.toMap() 生成的收集器，用户需要指定如何生成 <em>Map</em> 的 <em>key</em> 和 <em>value</em> 。</li> <li>使用 Collectors.partitioningBy() 生成的收集器，对元素进行二分区操作时用到。</li> <li>使用 Collectors.groupingBy() 生成的收集器，对元素做 <em>group</em> 操作时用到。</li> </ol> <p>情况1：使用 toMap() 生成的收集器，这种情况是最直接的，前面例子中已提到，这是和 Collectors.toCollection() 并列的方法。如下代码展示将学生列表转换成由 <学生 gpa=""> 组成的 <em>Map</em> 。非常直观，无需多言。 <!--学生--></p> <pre> <code class="language-java">// 使用toMap()统计学生GPA Map<Student, Double> studentToGPA = students.stream().collect(Collectors.toMap(Functions.identity(),// 如何生成key student -> computeGPA(student)));// 如何生成value</code></pre> <p>情况2：使用 partitioningBy() 生成的收集器，这种情况适用于将 Stream 中的元素依据某个二值逻辑（满足条件，或不满足）分成互补相交的两部分，比如男女性别、成绩及格与否等。下列代码展示将学生分成成绩及格或不及格的两部分。</p> <pre> <code class="language-java">// Partition students into passing and failing Map<Boolean, List<Student>> passingFailing = students.stream() .collect(Collectors.partitioningBy(s -> s.getGrade() >= PASS_THRESHOLD));</code></pre> <p>情况3：使用 groupingBy() 生成的收集器，这是比较灵活的一种情况。跟SQL中的 <em>group by</em> 语句类似，这里的 <em>groupingBy()</em> 也是按照某个属性对数据进行分组，属性相同的元素会被对应到 <em>Map</em> 的同一个 <em>key</em> 上。下列代码展示将员工按照部门进行分组：</p> <pre> <code class="language-java">// Group employees by department Map<Department, List<Employee>> byDept = employees.stream() .collect(Collectors.groupingBy(Employee::getDepartment));</code></pre> <p>以上只是分组的最基本用法，有些时候仅仅分组是不够的。在SQL中使用 <em>group by</em> 是为了协助其他查询，比如 <em>1. 先将员工按照部门分组，2. 然后统计每个部门员工的人数</em> 。Java类库设计者也考虑到了这种情况，增强版的 groupingBy() 能够满足这种需求。增强版的 groupingBy() 允许我们对元素分组之后再执行某种运算，比如求和、计数、平均值、类型转换等。这种先将元素分组的收集器叫做 <strong>上游收集器</strong> ，之后执行其他运算的收集器叫做 <strong>下游收集器</strong> ( <em>downstream Collector</em> )。</p> <pre> <code class="language-java">// 使用下游收集器统计每个部门的人数 Map<Department, Integer> totalByDept = employees.stream() .collect(Collectors.groupingBy(Employee::getDepartment, Collectors.counting()));// 下游收集器</code></pre> <p>上面代码的逻辑是不是越看越像SQL？高度非结构化。还有更狠的，下游收集器还可以包含更下游的收集器，这绝不是为了炫技而增加的把戏，而是实际场景需要。考虑将员工按照部门分组的场景，如果 <em>我们想得到每个员工的名字（字符串），而不是一个个</em> Employee <em>对象</em> ，可通过如下方式做到：</p> <pre> <code class="language-java">// 按照部门对员工分布组，并只保留员工的名字 Map<Department, List<String>> byDept = employees.stream() .collect(Collectors.groupingBy(Employee::getDepartment, Collectors.mapping(Employee::getName,// 下游收集器 Collectors.toList())));// 更下游的收集器</code></pre> <p>如果看到这里你还没有对Java函数式编程失去信心，恭喜你，你已经顺利成为Java函数式编程大师了。</p> <h2>使用collect()做字符串join</h2> <p>这个肯定是大家喜闻乐见的功能，字符串拼接时使用 Collectors.joining() 生成的收集器，从此告别 <em>for</em> 循环。 Collectors.joining() 方法有三种重写形式，分别对应三种不同的拼接方式。无需多言，代码过目难忘。</p> <pre> <code class="language-java">// 使用Collectors.joining()拼接字符串 Stream<String> stream = Stream.of("I", "love", "you"); //String joined = stream.collect(Collectors.joining());// "Iloveyou" //String joined = stream.collect(Collectors.joining(","));// "I,love,you" String joined = stream.collect(Collectors.joining(",", "{", "}"));// "{I,love,you}"</code></pre> <h2>collect()还可以做更多</h2> <p>除了可以使用 <em>Collectors</em> 工具类已经封装好的收集器，我们还可以自定义收集器，或者直接调用 collect(Supplier<R> supplier, BiConsumer<R,? super T> accumulator, BiConsumer<R,R> combiner) 方法， <strong>收集任何形式你想要的信息</strong> 。不过 <em>Collectors</em> 工具类应该能满足我们的绝大部分需求，手动实现之间请先看看文档。</p> <p> </p> <h2>参考文献</h2> <ol> <li><a href="/misc/goto?guid=4959741539207964577" rel="nofollow,noindex">https://docs.oracle.com/javase/8/docs/api/java/util/stream/package-summary.html#package.description</a></li> <li><a href="/misc/goto?guid=4958970912940962098" rel="nofollow,noindex">https://docs.oracle.com/javase/tutorial/java/javaOO/methodreferences.html</a></li> <li><a href="/misc/goto?guid=4959741539322008417" rel="nofollow,noindex">https://docs.oracle.com/javase/8/docs/api/java/util/stream/Collector.html</a></li> <li><a href="/misc/goto?guid=4959664587425097235" rel="nofollow,noindex">https://docs.oracle.com/javase/8/docs/api/java/util/stream/Stream.html</a></li> <li><a href="/misc/goto?guid=4959669235400570996" rel="nofollow,noindex">https://docs.oracle.com/javase/8/docs/api/java/util/stream/Collectors.html</a></li> </ol> <p> </p> <p> </p>