Stream中关于collect方法的介绍

最新推荐文章于 2025-03-09 14:05:31 发布

转载最新推荐文章于 2025-03-09 14:05:31 发布 · 3.8k 阅读

3 ·

CC 4.0 BY-SA版权

原文链接：https://blog.youkuaiyun.com/liujun03/article/details/80701999

java 专栏收录该内容

11 篇文章

订阅专栏

本文深入探讨Java Stream API中的collect方法，包括两种不同形式的用法。通过实例展示了如何使用Collectors进行数组转换、数据分组统计及lambda表达式在Stream中的应用。同时，详细解析了第二种collect方法的参数作用及其在复杂数据转换场景下的实践。

在Stream 的API中可以查到有两种collect方法，分别是：

第一种：

<R, A> R collect(Collector<? super T, A, R> collector);

第二种：

<R> R collect(Supplier<R> supplier, BiConsumer<R, ? super T> accumulator,
        BiConsumer<R, R> combiner);

对于第1个方法，我们主要是使用 Collectors（java.util.stream.Collectors）来进行各种 reduction 操作，下面举几个例子：

1.将数组组成字符串，注：Collectors.joining()有三个重载方法

String[] arr ={"aa","ccc","sss"};
System.out.println(Arrays.stream(arr).collect(joining()));
// aacccsss
System.out.println(Arrays.stream(arr).collect(joining("|")));
// aa|ccc|sss
System.out.println(Arrays.stream(arr).collect(joining(",","{","}")));
// {aa,ccc,sss}

2.将数组转为集合List

String[] arr ={"aa","ccc","sss"};
System.out.println(Arrays.stream(arr).collect(toList()));

// [aa, ccc, sss]

3.将list中的数据分组并统计数量

public static class Person{
    private long id;
    private int age;
       private String name;

//   get/set
｝

List<Person> list= Lists.newArrayList();
//假装list中已有许多条数据=-=
Map<Integer, Long> personGroups = list.stream().
               collect(Collectors.groupingBy(Person::getAge,counting()));
// 这样我们就得到了一个 以年龄为key,以这个年龄的人数为value的map了

上面应该是几个比较常用的对第一个collect方法的使用了。下面着重写一下对第二collect个方法的使用过程。

对于函数

<R> R collect(Supplier<R> supplier,
                  BiConsumer<R, ? super T> accumulator,
                  BiConsumer<R, R> combiner);

来说，参数supplier 是一个生成目标类型实例的方法，代表着目标容器是什么；accumulator是将操作的目标数据填充到supplier 生成的目标类型实例中去的方法，代表着如何将元素添加到容器中；而combiner是将多个supplier 生成的实例整合到一起的方法，代表着规约操作，将多个结果合并。

如上面的保存有多个Person实例的list来说，假如我们需要将这个list做一个转变，变为以id为key,value为person的Map的话，那就可以使用这个方法了：

Map<Long,Person> personMap =list.stream().collect(Maps::newHashMap,                                                                 
(map,p)>map.put(p.getId(),p),Map::putAll);

注：lambda表达式的应用在Stream中很重要的。

上面的双冒号运算符所在的表达式转化一下就可以得到：

Map<Long,Person> personMap = list.stream().collect(() -> new HashMap<>(), 
                                 (map ,p) ->map.put(p.getId(),p),(m ,n) -> m.putAll(n));