Java中collect方法怎么用 掌握流结果收集的各种方式

java的collect方法是stream api中的终结操作,用于将流元素收集到指定数据结构中。其核心在于supplier、accumulator和combiner三个参数:1.supplier创建结果容器;2.accumulator将元素添加到容器;3

.combiner合并多个容器结果。常用collectors类简化操作,如tolist收集到list,toset去重收集到set,tomap构建映射,groupingby按条件分组(可嵌套下游收集器),joining连接字符串(支持分隔符、前缀、后缀),reducing执行聚合计算(如求和)。掌握这些用法可高效处理集合数据。

Java的collect方法是Stream API中的一个终结操作,它允许你将流中的元素收集到一个集合、列表、映射或者任何其他你想要的数据结构中。简单来说,它就是把流水线上的东西打包带走,装进你指定的容器里。

解决方案

collect方法的核心在于它的三个参数:Supplier, Accumulator, 和 Combiner

  • Supplier: 一个函数,用来创建结果容器的初始实例。 可以理解为你要装东西的那个“空箱子”。
  • Accumulator: 一个函数,将流中的元素添加到结果容器中。 这就是把流水线上每个产品放到箱子里的操作。
  • Combiner: 一个函数,用于合并多个部分结果容器(在并行流中)。 如果流水线分成了几条线同时生产,最后需要把几个箱子里的东西合并成一个。

但大多数时候,我们不需要自己手动实现这三个参数,而是使用Collectors类提供的静态方法,它们已经封装好了常用的收集逻辑。

例如,将流收集到一个List:

List names = Stream.of("Alice", "Bob", "Charlie")
                           .collect(Collectors.toList());
System.out.println(names); // 输出: [Alice, Bob, Charlie]

收集到一个Set:

Set uniqueNames = Stream.of("Alice", "Bob", "Alice", "Charlie")
                               .collect(Collectors.toSet());
System.out.println(uniqueNames); // 输出: [Alice, Bob, Charlie] (顺序可能不同)

收集到一个Map(需要指定key和value的映射):

Map nameLengths = Stream.of("Alice", "Bob", "Charlie")
                                         .collect(Collectors.toMap(
                                                 name -> name,  // key
                                                 name -> name.length() // value
                                         ));
System.out.println(nameLengths); // 输出: {Alice=5, Bob=3, Charlie=7}

如何使用groupingBy进行分组?

groupingByCollectors中非常强大的一个方法,它允许你根据某个条件将流中的元素分组到不同的集合中。 可以理解为把流水线上的产品按照型号分别装到不同的箱子里。

例如,按照字符串长度分组:

Map> namesByLength = Stream.of("Alice", "Bob", "Charlie", "Dave")
                                                 .collect(Collectors.groupingBy(String::length));
System.out.println(namesByLength);
// 输出: {3=[Bob, Dave], 5=[Alice], 7=[Charlie]}

如果还想对每个分组的结果进行进一步的处理,可以使用groupingBy的第二个参数,它接受另一个Collector作为下游收集器。 比如,计算每个长度的字符串的数量:

Map nameCountByLength = Stream.of("Alice", "Bob", "Charlie", "Dave")
                                               .collect(Collectors.groupingBy(
                                                       String::length,
                                                       Collectors.counting()
                                               ));
System.out.println(nameCountByLength); // 输出: {3=2, 5=1, 7=1}

joining方法有什么用?

joining方法用于将流中的字符串元素连接成一个字符串。 它接受三个可选参数:分隔符、前缀和后缀。

String joinedNames = Stream.of("Alice", "Bob", "Charlie")
                             .collect(Collectors.joining(", "));
System.out.println(joinedNames); // 输出: Alice, Bob, Charlie

加入前缀和后缀:

String joinedNamesWithPrefixSuffix = Stream.of("Alice", "Bob", "Charlie")
                                             .collect(Collectors.joining(", ", "[", "]"));
System.out.println(joinedNamesWithPrefixSuffix); // 输出: [Alice, Bob, Charlie]

如何使用reducing进行聚合计算?

reducing方法用于对流中的元素进行聚合计算,比如求和、求平均值等。 它接受三个参数:初始值、累加器和组合器。

例如,计算所有字符串的长度之和:

Integer totalLength = Stream.of("Alice", "Bob", "Charlie")
                            .collect(Collectors.reducing(
                                    0,  // 初始值
                                    String::length, // 累加器 (将字符串转换为长度)
                                    Integer::sum // 组合器 (将两个长度相加)
                            ));
System.out.println(totalLength); // 输出: 15

reducing还有更简洁的重载形式,可以直接传入一个二元操作符:

Optional sum = Stream.of(1, 2, 3, 4, 5)
                               .collect(Collectors.reducing(Integer::sum));

System.out.println(sum.orElse(0)); // 输出: 15

注意这里返回的是Optional,因为如果流为空,则无法计算总和。

collect方法是Java Stream API中一个非常灵活和强大的工具,掌握它可以让你更高效地处理集合数据。虽然一开始可能觉得参数有点多,但多用几次就熟练了。