Java8新特性之Collectors

蔚落 2022-01-26 00:11 442阅读 0赞

### 1. 前言 ###

`collect`是一个将数据流缩减为一个值的归约操作。这个值可以是集合、映射，或者一个值对象。你可以使用`collect`达到以下目的：

1.  **将数据流缩减为一个单一值**：一个流执行后的结果能够被缩减为一个单一的值。单一的值可以是一个`Collection`，或者像`int`、`double`等的数值，再或者是一个用户自定义的值对象。
2.  **将一个数据流中的元素进行分组**：根据任务类型将流中所有的任务进行分组。这将产生一个`Map<TaskType, List<Task>>`的结果，其中每个实体包含一个任务类型以及与它相关的任务。你也可以使用除了列表以外的任何其他的集合。如果你不需要与一任务类型相关的所有的任务，你可以选择产生一个`Map<TaskType, Task>`。这是一个能够根据任务类型对任务进行分类并获取每类任务中第一个任务的例子。
3.  **分割一个流中的元素**：你可以将一个流分割为两组——比如将任务分割为要做和已经做完的任务。

### 2. Collector实际应用 ###

为了感受到`Collector`的威力，让我们来看一下我们要根据任务类型来对任务进行分类的例子。在`Java8`中，我们可以通过编写如下的代码达到将任务根据类型分组的目的。

private static Map<TaskType, List<Task>> groupTasksByType(List<Task> tasks) {
        return tasks.stream().collect(Collectors.groupingBy(task -> task.getType()));
    }

上面的代码使用了定义在辅助类`Collectors`中的`groupingBy`收集器。它创建了一个映射，其中`TaskType`是它的键，而包含了所有拥有相同`TaskType`的任务的列表是它的值。为了在`Java7`中达到相同的效果，你需要编写如下的代码。

public static void main(String[] args) {
        List<Task> tasks = getTasks();
        Map<TaskType, List<Task>> allTasksByType = new HashMap<>();
        for (Task task : tasks) {
            List<Task> existingTasksByType = allTasksByType.get(task.getType());
            if (existingTasksByType == null) {
                List<Task> tasksByType = new ArrayList<>();
                tasksByType.add(task);
                allTasksByType.put(task.getType(), tasksByType);
            } else {
                existingTasksByType.add(task);
            }
        }
        for (Map.Entry<TaskType, List<Task>> entry : allTasksByType.entrySet()) {
            System.out.println(String.format("%s =>> %s", entry.getKey(), entry.getValue()));
        }
    }

### 3. 收集器：常用的规约操作 ###

`Collectors`辅助类提供了大量的静态辅助方法来创建收集器为常见的使用场景服务，像将元素收集到一个集合中、分组和分割元素，或者根据不同的标准来概述元素。我们将在这篇博文中涵盖大部分常见的`Collector`。

#### 3.1 缩减为一个值 ####

正如上面讨论的，收集器可以被用来收集流的输出到一个集合，或者产生一个单一的值。

**将数据收集进一个列表**  
让我们编写我们的第一个测试用例——给定一个任务列表，我们想将他们的标题收集进一个列表。

import static java.util.stream.Collectors.toList;
    
    public class Example2_ReduceValue {
        public List<String> allTitles(List<Task> tasks) {
            return tasks.stream().map(Task::getTitle).collect(toList());
        }
    }

`toList`收集器使用了列表的`add`方法来向结果列表中添加元素。`toList`收集器使用了`ArrayList`作为列表的实现。

**将数据收集进一个集合**  
如果我们想要确保返回的标题都是唯一的，并且我们不在乎元素的顺序，那么我们可以使用`toSet`收集器。

import static java.util.stream.Collectors.toSet;
    
    public Set<String> uniqueTitles(List<Task> tasks) {
        return tasks.stream().map(Task::getTitle).collect(toSet());
    }

toSet方法使用了HashSet作为集合的实现来存储结果集。

**将数据收集进一个映射**  
你可以使用`toMap`收集器将一个流转换为一个映射。`toMap`收集器需要两个映射方法来获得映射的键和值。在下面展示的代码中，`Task::getTitle`是接收一个任务并产生一个只包含该任务标题的键的`Function`。`task -> task`是一个用来返回任务本身的lambda表达式。

private static Map<String, Task> taskMap(List<Task> tasks) {
      return tasks.stream().collect(toMap(Task::getTitle, task -> task));
    }

我们可以通过使用`Function`接口中的默认方法`identity`来改进上面展示的代码，如下所示，这样可以让代码更加简洁，并更好地传达开发者的意图。

import static java.util.function.Function.identity;
    
    private static Map<String, Task> taskMap(List<Task> tasks) {
      return tasks.stream().collect(toMap(Task::getTitle, identity()));
    }

从一个流中创建映射的代码会在存在重复的键时抛出异常。你将会得到一个类似下面的错误。

Exception in thread "main" java.lang.IllegalStateException: Duplicate key Task{title='Read Version Control with Git book', type=READING}
    at java.util.stream.Collectors.lambda$throwingMerger$105(Collectors.java:133)

你可以通过使用`toMap`方法的另一个变体来处理重复问题，它允许我们指定一个合并方法。这个合并方法允许用户他们指定想如何处理多个值关联到同一个键的冲突。在下面展示的代码中，我们只是使用了新的值，当然你也可以编写一个智能的算法来处理冲突。

private static Map<String, Task> taskMap_duplicates(List<Task> tasks) {
      return tasks.stream().collect(toMap(Task::getTitle, identity(), (t1, t2) -> t2));
    }

**以下报错：**  
你可以通过使用`toMap`方法的第三个变体来指定其他的映射实现。这需要你指定将用来存储结果的`Map`和`Supplier`。

public Map<String, Task> collectToMap(List<Task> tasks) {
        return tasks.stream().collect(toMap(Task::getTitle, identity(), (t1, t2) -> t2, LinkedHashMap::new));
    }

类似于`toMap`收集器，也有`toConcurrentMap`收集器，它产生一个`ConcurrentMap`而不是`HashMap`。

--------------------

##### 转自：[https://blog.csdn.net/u013291394/article/details/52662761][https_blog.csdn.net_u013291394_article_details_52662761] #####

[https_blog.csdn.net_u013291394_article_details_52662761]: https://blog.csdn.net/u013291394/article/details/52662761