首页　>　文章列表　>　Java 8中的Stream API：如何使用collect()方法将集合收集为Map对象

Java 8中的Stream API：如何使用collect()方法将集合收集为Map对象

java map StreamAPI collect()
472 2023-08-01

Java 8 中引入了新的 Stream API，它提供了一种更加高效、简洁的方式来处理集合数据。Stream API 提供了各种方法来对数据进行处理和转换，其中 collect() 方法是一个非常重要且常用的方法之一。本文将介绍如何使用 collect() 方法将集合收集为 Map 对象，并提供相应的代码示例。

在 Java 8 之前，如果我们想将一个集合转换为 Map 对象，需要使用繁琐的遍历和添加操作。而在 Java 8 中使用 Stream API 的 collect() 方法可以更加方便地实现这个目标。

collect() 方法是 Stream API 的终止操作之一，它接收一个 Collector 参数，用于指定收集的方式。在收集为 Map 对象时，我们可以使用 Collectors.toMap() 方法来进行收集。

下面是使用 collect() 方法将集合收集为 Map 对象的示例代码：

import java.util.*;
import java.util.stream.Collectors;

public class StreamCollectExample {
    public static void main(String[] args) {
        List<String> fruits = Arrays.asList("apple", "banana", "orange");

        Map<String, Integer> fruitLengthMap = fruits.stream()
                .collect(Collectors.toMap(
                        fruit -> fruit,                   // Key 映射函数
                        fruit -> fruit.length()           // Value 映射函数
                ));

        System.out.println(fruitLengthMap);
    }
}

上述代码中，我们首先创建了一个包含三个水果的集合 fruits，然后通过 stream() 方法将其转换为一个流。接着使用 collect() 方法并传入 Collectors.toMap() 方法作为参数，该方法接收两个 lambda 表达式参数，用于指定 Key 和 Value 的映射函数。

在我们的示例中，Key 映射函数是 fruit -> fruit，即将水果作为 Key；Value 映射函数是 fruit -> fruit.length()，即将水果的长度作为 Value。最后，collect() 方法将流中的元素按照指定的映射函数进行处理，并返回一个 Map 对象。

输出结果如下：

{orange=6, banana=6, apple=5}

可以看到，最终我们获得了一个包含水果及其长度的 Map 对象。

除了基本的收集功能，Collectors.toMap() 方法还提供了一些其他的参数。例如，我们可以指定当存在重复的 Key 时应该如何处理，通过传入一个合并函数来解决冲突。

下面是一个带有 Key 冲突处理的示例代码：

import java.util.*;
import java.util.stream.Collectors;

public class StreamCollectExample {
    public static void main(String[] args) {
        List<String> fruits = Arrays.asList("apple", "banana", "orange", "apple");

        Map<String, Integer> fruitLengthMap = fruits.stream()
                .collect(Collectors.toMap(
                        fruit -> fruit,                   // Key 映射函数
                        fruit -> fruit.length(),          // Value 映射函数
                        (length1, length2) -> length1    // Key 冲突处理函数
                ));

        System.out.println(fruitLengthMap);
    }
}

在上述代码中，我们在 toMap() 方法的第三个参数位置上传入了一个合并函数 (length1, length2) -> length1。该函数会在遇到重复的 Key 时选择保留第一个 Key，并忽略后续的 Key。

输出结果如下：