Java8 Stream之group

2019-01-06  本文已影响0人  大哥你先走

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)
static <T,K,A,D> Collector<T,?,Map<K,D>> 
     groupingBy(Function<? super T,? extends K> classifier, Collector<? super T,A,D> downstream)
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));
上一篇 下一篇

猜你喜欢

热点阅读