本文将介绍如何使用 Java Stream API 将一个数据流分割成两个不同的集合。正如摘要所述，我们将利用 partitioningBy 方法，根据特定的条件将原始数据流分割成满足条件和不满足条件的两部分。这种方法相较于传统的循环方式，代码更加简洁高效。
使用 partitioningBy 方法分割数据流
Collectors.partitioningBy() 是 Java Stream API 提供的一个非常强大的收集器。它接收一个 Predicate 函数式接口作为参数，该接口定义了一个测试条件，用于判断流中的元素是否满足条件。partitioningBy() 方法会根据这个条件将流中的元素分成两部分：满足条件的元素会被收集到一个列表中，不满足条件的元素会被收集到另一个列表中。最终，partitioningBy() 方法会返回一个 map>，其中 Boolean 类型的键表示条件是否满足（true 表示满足，false 表示不满足），List 类型的值则分别存储了满足条件和不满足条件的元素列表。
示例代码
假设我们有一个 ID 列表 myIdList 和一个将 ID 映射到对象的 Map myObjectMap。我们需要根据 ID 是否存在于 myObjectMap 中，将 myIdList 分割成两个列表：一个包含存在于 myObjectMap 中的 ID 对应的对象，另一个包含 myObjectMap 中不存在的 ID。
以下是使用 partitioningBy 方法实现该功能的示例代码：
立即学习“Java免费学习笔记（深入）”；
import java.util.*; import java.util.stream.Collectors; public class StreamPartitioning { public static void main(String[] args) { List<String> myIdList = Arrays.asList("a", "b", "c"); Map<String, Object> myObjectMap = new HashMap<>(); myObjectMap.put("b", "B"); Map<Boolean, List<String>> partitioned = myIdList.stream() .collect(Collectors.partitioningBy(myObjectMap::containsKey)); List<Object> objectList = partitioned.get(true).stream() .map(myObjectMap::get).collect(Collectors.toList()); List<String> missingObjIds = partitioned.get(false); System.out.println("objectList=" + objectList); System.out.println("missingObjIds=" + missingObjIds); } }
代码解释：
- myIdList.stream(): 将 ID 列表转换为一个 Stream 对象。
- Collectors.partitioningBy(myObjectMap::containsKey): 使用 partitioningBy 方法，并传入 myObjectMap::containsKey 作为条件。myObjectMap::containsKey 是一个方法引用，等价于 id -> myObjectMap.containsKey(id)，用于判断 myObjectMap 中是否包含指定的 ID。
- partitioned.get(true): 获取 myObjectMap 中存在的 ID 列表。
- .stream().map(myObjectMap::get).collect(Collectors.toList()): 将存在的 ID 列表转换为 Stream 对象，然后使用 map 方法将每个 ID 映射到对应的对象，最后使用 collect 方法将映射后的对象收集到一个列表中。
- partitioned.get(false): 获取 myObjectMap 中不存在的 ID 列表。
输出结果：
objectList=[B] missingObjIds=[a, c]
注意事项
- partitioningBy 方法返回的 Map 始终包含 true 和 false 两个键，即使其中一个列表为空。
- 如果 Predicate 条件的计算成本较高，可能会影响性能。在这种情况下，可以考虑使用其他方法来分割数据流。
- myObjectMap::get 在stream的map操作中，如果map中不存在key值，则会返回NULL。
总结
通过使用 Java Stream API 的 partitioningBy 方法，我们可以简洁高效地将一个数据流分割成两个不同的集合。这种方法可以有效地减少代码的冗余，提高代码的可读性和可维护性。在处理需要根据条件分割数据流的场景时，partitioningBy 方法是一个非常有用的工具。
