Java8 Stream之group

1 介绍

2 Collectors的groupingBy方法

使用Java 8 提供的Stream API 可以以声明式的方法来处理集合中的数据。静态工厂方法 Collectors.groupingBy()Collectors.groupingByConcurrent()提供类似SQL语言中的‘GROUP BY’的功能。两个工厂方法用于根据某些属性对集合中的元素分组并将分组结果存储在MapConcurrentMap)中。

groupingBy方法的三个重载版本:

  • 参数为分类函数
static <T,K> Collector<T,?,Map<K,List<T>>> groupingBy(Function<? super T,? extends K> classifier)
  • 参数为分类函数、Collector对象(执行reduce操作)
static <T,K,A,D> Collector<T,?,Map<K,D>> 
     groupingBy(Function<? super T,? extends K> classifier, Collector<? super T,A,D> downstream)
  • 参数为分类函数、Supplier对象(提供Map接口的实现,用于分类结果的存储)和Collector对象(执行reduce操作)
static <T,K,D,A,M extends Map<K,D>> Collector<T,?,M>
  groupingBy(Function<? super T,? extends K> classifier, 
    Supplier<M> mapFactory, Collector<? super T,A,D> downstream)

2.1 示例代码设置

  • 文章类型
public enum  ArticleType {
    NEWS,
    REVIEW,
    GUIDE
}
  • 文章,包括文章标题、作者、文章类型和文章获得的喜欢数。
public class Article {
    String title;
    String author;
    ArticleType type;
    int likes;
}
  • 待处理的文章列表
List<Article> articles = Arrays.asList(...);

2.2 基于单属性分组

单属性分组只需要一个分类函数作为参数,分类函数会作用于流的所有元素,分类函数的返回值将作为分组结果的键。

基于作者姓名的分组如下:

        Map<String, List<Article>> byAuthor = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor));

        Map<ArticleType, List<Article>> byType = articles.stream()
            .collect(Collectors.groupingBy(article -> article.getType()));

2.3 分组结果的键为复杂类型

分类函数不仅限于返回一个纯量或字符串,分类结果的键可以是任何类型,但是要求实现了必要的equalshashcode方法。

基于作者姓名和文章类型的分组如下:

Map<Tuple2<String, ArticleType>, List<Article>> complexKeMap = articles.stream()
            .collect(Collectors.groupingBy(new Function<Article, Tuple2<String, ArticleType>>() {
                @Override
                public Tuple2<String, ArticleType> apply(Article article) {
                    return new Tuple2<>(article.getAuthor(), article.getType());
                }
            }));

2.4 修改分组结果的值类型

默认分组结果的值类型为List,可以通过提供第二个参数来修改返回的类型。返回Set类型的示例代码如下:

        // modify the return type Map<String,List> to Map<String,Set>
        Map<String, Set<Article>> authorSet = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.toSet()));

2.5 多条件分组

下面的例子,首先按照作业分组,然后对分组的结果再按照文章类型分组:

Map<String, Map<ArticleType, List<Article>>> chainGroup = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.groupingBy(Article::getType)));

2.6 对分组结果执行Reduce操作

一般分组结果的值类型为List,可以根据需要对List执行Reduce操作,比如获取List中某个属性值最大最小值的元素。

计算作者文章的获得的喜欢数的平均值:

Map<String, Double> averageLike = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.averagingInt(Article::getLikes)));

计算作者文章得到的喜欢数的总和:

        Map<String, Integer> sumLike = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.summingInt(Article::getLikes)));

计算每位作者最受欢迎的文章:

        Map<String, Optional<Article>> maxLikeByAuthor = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.maxBy(Comparator.comparingInt(Article::getLikes))));
        System.out.println(maxLikeByAuthor);

计算每位作者最不受欢迎的文章:

        Map<String, Optional<Article>> minLikeByAuthor = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.minBy(Comparator.comparingInt(Article::getLikes))));

2.7 获取分组结果的统计数据

Stream专门提供了计算分组结果统计值的接口,统计值包括:最大值、最小值、平均值、总数、总和五个指标。

        Map<String, IntSummaryStatistics> summaryStatistics = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.summarizingInt(Article::getLikes)));

2.8 修改分组结果值的类型

        Map<String, String> modifyGroupType = articles.stream()
            .collect(Collectors.groupingBy(Article::getAuthor, Collectors.mapping(Article::getTitle, joining("||"))));
        System.out.println(modifyGroupType);

2.9 返回支持并发操作的Map

        ConcurrentMap<String,List<Article>> concurrentGroup = articles.parallelStream()
            .collect(Collectors.groupingByConcurrent(Article::getAuthor));
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Int Double Long 设置特定的stream类型, 提高性能,增加特定的函数 无存储。stream不是一...
    patrick002阅读 1,291评论 0 0
  • Java8 in action 没有共享的可变数据,将方法和函数即代码传递给其他方法的能力就是我们平常所说的函数式...
    铁牛很铁阅读 1,288评论 1 2
  • 小鲜肉李易峰同款手链,为啥对李易峰总是喜欢不起来,好像没啥特别的,不过这款手链确实漂亮。 上一张成品图 怎么样,漂...
    木子_a24c阅读 2,251评论 0 0
  • 聽汀阅读 138评论 0 0
  • 逃离 用一场急雨的时间 仰头看 太阳明媚如斯 嘲笑我的怯懦 我的不甘 写在濡湿的地面 渗入那龟裂的混泥土中 消失不...
    田乐淘阅读 345评论 0 5