在 java 编程中，处理不可变集合（immutable set）是常见的需求，尤其是在函数式编程范式中，它有助于提升代码的健壮性和可预测性。当我们需要基于一个已有的不可变 set，并添加一些新的元素来创建一个全新的不可变 set 时，直接使用 set.of() 方法可能会导致意想不到的结果。例如，尝试 set t = set.of(s, "d") 会将 s 作为一个整体元素而不是其内部的元素进行处理，从而创建一个包含 set
为了解决这个问题，我们可以利用 Java 8 引入的 Stream API，特别是 flatMap 操作符，它能够将一个流中的每个元素转换为一个流，然后将这些新的流连接成一个单一的流。
解决方案一：合并集合流
这种方法的核心思想是将现有的不可变 Set 和包含新元素的 Set 都视为独立的集合，然后创建一个包含这些集合的流，再通过 flatMap 将它们扁平化为单个元素的流。
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class ImmutableSetExtension {
public static void main(String[] args) {
// 原始不可变集合
Set<String> s = Set.of("a", "b", "c");
// 创建包含原始集合和新元素集合的流，然后扁平化并收集
Set<String> t = Stream.of(s, Set.of("d", "e")) // Stream.of(Set<String>, Set<String>)
.flatMap(Set::stream) // 将每个Set<String>转换为Stream<String>并扁平化
.collect(Collectors.toUnmodifiableSet()); // 收集为新的不可变Set
System.out.println(t);
// 预期输出：[d, e, c, b, a] (顺序可能不同，因为Set无序)
}
}代码解析：
- Set
s = Set.of("a", "b", "c");：初始化一个包含 "a", "b", "c" 的不可变 Set。 - Stream.of(s, Set.of("d", "e"))：这里创建了一个 Stream
>，它包含了两个 Set 对象：s 和一个临时创建的 Set.of("d", "e")。 - .flatMap(Set::stream)：这是关键步骤。对于流中的每个 Set 对象（即 s 和 Set.of("d", "e")），Set::stream 方法会将其转换为一个 Stream
。flatMap 会将这些独立的 Stream 连接成一个单一的 Stream ，其中包含了所有原始集合和新集合的元素。 - .collect(Collectors.toUnmodifiableSet())：最后，使用 Collectors.toUnmodifiableSet() 将扁平化后的元素流收集到一个新的不可变 Set 中。这个方法确保了返回的 Set 是不可修改的。
解决方案二：合并元素流
第二种方法更为直接，它创建了一个包含现有集合的元素流和新元素的流的流，然后通过 flatMap 进行扁平化。
立即学习“Java免费学习笔记（深入）”；
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class ImmutableSetExtensionAlternative {
public static void main(String[] args) {
// 原始不可变集合
Set<String> s = Set.of("a", "b", "c");
// 创建包含原始集合元素流和新元素流的流，然后扁平化并收集
Set<String> t = Stream.of(s.stream(), Stream.of("d", "e")) // Stream.of(Stream<String>, Stream<String>)
.flatMap(Function.identity()) // 扁平化Stream<Stream<String>>为Stream<String>
.collect(Collectors.toUnmodifiableSet()); // 收集为新的不可变Set
System.out.println(t);
// 预期输出：[d, e, c, b, a] (顺序可能不同，因为Set无序)
}
}代码解析：
- Set
s = Set.of("a", "b", "c");：同样初始化原始不可变 Set。 - Stream.of(s.stream(), Stream.of("d", "e"))：这里创建了一个 Stream
>。第一个元素是 s 的元素流 (s.stream())，第二个元素是直接由新元素 "d" 和 "e" 构成的流 (Stream.of("d", "e"))。 - .flatMap(Function.identity())：Function.identity() 是一个返回其输入参数的函数。在这种情况下，flatMap 会接收一个 Stream
，并直接将其作为结果流的一部分。由于我们已经有一个 Stream >，flatMap 将把内部的 Stream 提取出来并连接成一个单一的 Stream 。 - .collect(Collectors.toUnmodifiableSet())：同样，将扁平化后的元素流收集到一个新的不可变 Set 中。
注意事项与总结
- 不可变性保证： 两种解决方案都通过 Collectors.toUnmodifiableSet() 确保了最终生成的 Set 是不可修改的。这意味着一旦创建，就不能添加、删除或修改其元素。
- 类型安全： Stream API 的类型推断和泛型使用确保了在整个操作过程中，集合的元素类型 (String 在本例中) 始终保持一致，避免了 Set.of(s, "d") 带来的类型混淆问题。
- 元素顺序： Set 是一个无序集合，因此打印出的元素顺序可能与添加顺序不同，甚至在多次运行时也可能不同。如果需要保持顺序，应考虑使用 List 或 LinkedHashSet。
- 性能考量： 对于少量元素的集合，Stream API 的开销可以忽略不计。对于大量元素的集合，Stream API 通常是高效的，因为它利用了惰性求值和并行化潜力。
- 选择哪种方法： 两种方法在功能上是等价的。第一种方法可能在概念上更直观，因为它处理的是集合的流；第二种方法直接处理元素的流，可能在某些场景下显得更简洁。根据个人偏好和代码可读性选择即可。
通过上述方法，我们可以优雅且安全地从一个现有不可变 Set 中创建并扩展一个新的不可变 Set，这在处理需要保持数据完整性和避免副作用的场景中尤为有用。
