Stream 很好，Map 很酷，但答应我别用 toMap()

IT_陈寒

发布于 2025-06-01 13:10:05

14900

代码可运行

运行总次数：0

代码可运行

在现代 Java 编程中，Stream API 提供了一种简洁且强大的方式来处理集合数据。结合 Lambda 表达式，Stream API 可以让代码变得更加简洁和易读。Map 作为一种常用的数据结构，其与 Stream 的结合更是如虎添翼。然而，尽管 Stream 提供了强大的 Collectors.toMap() 方法，但在某些场景下，直接使用 toMap() 可能会带来一些潜在的问题和复杂性。

在本文中，我们将探讨 toMap() 的常见问题，并介绍一些替代方案，使你的代码更加健壮和易于维护。

`Collectors.toMap()` 的常见问题

重复键问题

使用 toMap() 时，如果源数据中存在重复键，会抛出 IllegalStateException：

List<String> names = Arrays.asList("John", "Jane", "John");
Map<String, Integer> nameLengthMap = names.stream()
    .collect(Collectors.toMap(name -> name, name -> name.length()));
// 抛出 IllegalStateException: Duplicate key John

处理空值

如果源数据中有空值（null），会抛出 NullPointerException：

List<String> names = Arrays.asList("John", "Jane", null);
Map<String, Integer> nameLengthMap = names.stream()
    .collect(Collectors.toMap(name -> name, name -> name.length()));
// 抛出 NullPointerException

性能问题

在处理大型数据集时，使用 toMap() 可能会导致性能问题，尤其是在需要合并重复键或处理复杂映射逻辑时。

替代方案

为了避免上述问题，我们可以采用以下几种替代方案：

1. 使用 `Collectors.groupingBy()`

对于可能有重复键的场景，可以使用 Collectors.groupingBy() 将值收集到 List 中：

List<String> names = Arrays.asList("John", "Jane", "John");
Map<String, List<Integer>> nameLengthMap = names.stream()
    .collect(Collectors.groupingBy(name -> name, Collectors.mapping(String::length, Collectors.toList())));

System.out.println(nameLengthMap);
// 输出: {John=[4, 4], Jane=[4]}

2. 处理空值

在映射过程中处理空值，避免 NullPointerException：

List<String> names = Arrays.asList("John", "Jane", null);
Map<String, Integer> nameLengthMap = names.stream()
    .filter(Objects::nonNull)
    .collect(Collectors.toMap(name -> name, name -> name.length()));

System.out.println(nameLengthMap);
// 输出: {John=4, Jane=4}

3. 自定义合并逻辑

通过 toMap() 的第三个参数自定义合并逻辑，处理重复键问题：

List<String> names = Arrays.asList("John", "Jane", "John");
Map<String, Integer> nameLengthMap = names.stream()
    .collect(Collectors.toMap(name -> name, name -> name.length(), (existing, replacement) -> existing));

System.out.println(nameLengthMap);
// 输出: {John=4, Jane=4}