在java开发中，我们经常会遇到需要对复杂数据结构进行转换的场景。例如，当有一个map
import java.util.HashMap;
import java.util.Map;
public class MapFlatteningExample {
public static void main(String[] args) {
Map> myMap = new HashMap<>();
myMap.put("categoryA", new HashMap() {{
put("item1", "valueA1");
put("item2", "valueA2");
}});
myMap.put("categoryB", new HashMap() {{
put("item3", "valueB3");
put("item4", "valueB4");
}});
Map result = new HashMap<>();
myMap.forEach((k, v) -> {
result.putAll(v);
});
System.out.println("传统方法结果: " + result);
}
} 虽然上述方法能够解决问题，但Java 8引入的Stream API提供了更具函数式编程风格和表达力的解决方案，尤其在处理大规模数据或需要链式操作时，Stream API的优势更为明显。
使用Stream API扁平化Map
使用Stream API扁平化嵌套Map的核心在于flatMap操作和Collectors.toMap收集器。flatMap能够将流中的每个元素（这里是内部Map）转换成一个流，然后将这些独立的流合并成一个扁平化的流。
场景一：内部Map之间没有重复键
如果可以确定所有内部Map的键都是唯一的，即扁平化后不会出现键冲突，那么可以直接使用Collectors.toMap的两个参数重载形式：
import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;
public class MapFlatteningStreamUniqueKeys {
public static void main(String[] args) {
Map> myMap = new HashMap<>();
myMap.put("categoryA", new HashMap() {{
put("item1", "valueA1");
put("item2", "valueA2");
}});
myMap.put("categoryB", new HashMap() {{
put("item3", "valueB3");
put("item4", "valueB4");
}});
Map res = myMap.values() // 获取所有内部Map的集合
.stream() // 将内部Map的集合转换为流
.flatMap(value -> value.entrySet().stream()) // 将每个内部Map的entrySet转换为流，并扁平化为一个Entry流
.collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)); // 收集Entry流到新的Map中
System.out.println("Stream API (无重复键) 结果: " + res);
}
} 代码解析：
立即学习“Java免费学习笔记（深入）”；
- myMap.values()：获取外部Map中所有内部Map的集合（Collection
- .stream()：将这个集合转换为一个Stream
- .flatMap(value -> value.entrySet().stream())：这是关键一步。对于流中的每一个内部Map (value)，我们调用其entrySet().stream()方法，这会产生一个Stream
>。flatMap的作用是将这些由各个内部Map生成的Entry流“展平”成一个单一的Stream >。 - .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue))：将扁平化的Entry流收集到一个新的Map中。Map.Entry::getKey作为键映射函数，Map.Entry::getValue作为值映射函数。
场景二：内部Map之间可能存在重复键
如果不同的内部Map之间可能存在相同的键，那么在扁平化时会发生键冲突。Collectors.toMap的默认行为是在遇到重复键时抛出IllegalStateException。为了解决这个问题，我们需要提供一个合并函数（merge function）来指定如何处理冲突。
一个常见的合并策略是保留后遇到的值：
import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;
public class MapFlatteningStreamDuplicateKeys {
public static void main(String[] args) {
Map> myMapWithDuplicates = new HashMap<>();
myMapWithDuplicates.put("categoryA", new HashMap() {{
put("item1", "valueA1");
put("item2", "valueA2");
}});
myMapWithDuplicates.put("categoryB", new HashMap() {{
put("item2", "valueB2_duplicate"); // 重复键
put("item3", "valueB3");
}});
Map resWithMerge = myMapWithDuplicates.values()
.stream()
.flatMap(value -> value.entrySet().stream())
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
(v1, v2) -> v2 // 合并函数：当键冲突时，保留第二个值 (v2)
));
System.out.println("Stream API (有重复键，保留后者) 结果: " + resWithMerge);
// 也可以选择保留第一个值
Map resKeepFirst = myMapWithDuplicates.values()
.stream()
.flatMap(value -> value.entrySet().stream())
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
(v1, v2) -> v1 // 合并函数：当键冲突时，保留第一个值 (v1)
));
System.out.println("Stream API (有重复键，保留前者) 结果: " + resKeepFirst);
// 或者抛出异常 (默认行为，但此处显式声明)
try {
Map resThrowError = myMapWithDuplicates.values()
.stream()
.flatMap(value -> value.entrySet().stream())
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
(v1, v2) -> { throw new IllegalStateException(String.format("Duplicate key %s", Map.Entry::getKey)); }
));
System.out.println("Stream API (有重复键，抛出异常) 结果: " + resThrowError);
} catch (IllegalStateException e) {
System.out.println("Stream API (有重复键，抛出异常) 捕获到错误: " + e.getMessage());
}
}
} 合并函数 (v1, v2) -> v2 解析：Collectors.toMap的第三个参数是一个BinaryOperator
- (v1, v2) -> v2：表示当遇到重复键时，总是使用新遇到的值（v2）覆盖旧值（v1）。
- (v1, v2) -> v1：表示当遇到重复键时，总是保留旧值（v1），忽略新值（v2）。
- 你也可以实现更复杂的合并逻辑，例如将两个值拼接起来，或者进行数值相加等，这取决于具体的业务需求。
注意事项与总结
- 性能考量： 对于非常小的数据集，传统的forEach循环可能在性能上与Stream API不相上下，甚至可能略快。但对于大规模数据集，Stream API的表达力、可读性以及潜在的并行处理能力（通过.parallelStream()）使其成为更优的选择。
- 可读性： Stream API链式操作使得数据转换逻辑更加清晰、简洁，避免了嵌套循环和临时变量。
- 键冲突处理： 在扁平化Map时，务必考虑是否存在键冲突的可能性。如果存在，必须提供一个合适的合并函数来处理冲突，否则程序可能会抛出运行时异常。
- flatMap的作用： 理解flatMap是理解此类转换的关键。它将一个包含多个内部流的流转换为一个扁平化的单一流，非常适合处理嵌套集合的展开操作。
通过Stream API，我们可以以一种声明式、高效且富有表现力的方式，将复杂的嵌套Map结构扁平化为所需的简单Map。掌握flatMap和Collectors.toMap的不同用法，将极大地提升Java数据处理的效率和代码质量。
