Java8 Stream之group

1 介绍

2 Collectors的groupingBy方法

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

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);