Java9中新增的Collector收集器

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

Filtering Collector

Collectors.filtering类似于流过滤器Stream filter()；它用于过滤输入元素，但用于不同的场景。流的过滤器在流链中使用，而过滤是一个收集器，设计用于与groupingBy一起使用。

使用流的过滤器，首先过滤值，然后对其进行分组。这样，被过滤掉的值就消失了，没有任何痕迹。如果我们需要一个跟踪，那么我们需要先分组，然后应用过滤，这实际上是Collectors.filtering做的。

Collectors.filtering功能用于过滤输入元素，收集器用于收集过滤后的元素：

@Test
public void givenList_whenSatifyPredicate_thenMapValueWithOccurences() {
    List<Integer> numbers = List.of(1, 2, 3, 5, 5);

    Map<Integer, Long> result = numbers.stream()
      .filter(val -> val > 3)
      .collect(Collectors.groupingBy(i -> i, Collectors.counting()));

    assertEquals(1, result.size());

    result = numbers.stream()
      .collect(Collectors.groupingBy(i -> i,
        Collectors.filtering(val -> val > 3, Collectors.counting())));

    assertEquals(4, result.size());
}

FlatMapping Collector

Collectors.flatMapping类似于Collectors.mapping。但有一个更细粒度的目标。两个采集器都接受一个函数和一个采集器，其中收集元素，但flatMapping函数接受元素流，然后由采集器累积。

让我们看看下面的模型类：

class Blog {
    private String authorName;
    private List<String> comments;
      
    // constructor and getters
}

Collectors.flatMapping允许我们跳过中间集合，直接写入映射到收集器定义的组的单个容器。

分组方式：

@Test
public void givenListOfBlogs_whenAuthorName_thenMapAuthorWithComments() {
    Blog blog1 = new Blog("1", "Nice", "Very Nice");
    Blog blog2 = new Blog("2", "Disappointing", "Ok", "Could be better");
    List<Blog> blogs = List.of(blog1, blog2);
        
    Map<String,  List<List<String>>> authorComments1 = blogs.stream()
     .collect(Collectors.groupingBy(Blog::getAuthorName, 
       Collectors.mapping(Blog::getComments, Collectors.toList())));
       
    assertEquals(2, authorComments1.size());
    assertEquals(2, authorComments1.get("1").get(0).size());
    assertEquals(3, authorComments1.get("2").get(0).size());

    Map<String, List<String>> authorComments2 = blogs.stream()
      .collect(Collectors.groupingBy(Blog::getAuthorName, 
        Collectors.flatMapping(blog -> blog.getComments().stream(), 
        Collectors.toList())));

    assertEquals(2, authorComments2.size());
    assertEquals(2, authorComments2.get("1").size());
    assertEquals(3, authorComments2.get("2").size());
}

Collectors.mapping将所有分组作者的注释映射到收集器的容器（即列表），而此中间集合将使用flatMapping删除，因为它提供了要映射到收集器容器的注释列表的直接流。