使用 Java Stream API 分割数据流并生成两个不同的集合

本文将介绍如何使用 Java Stream API 将一个数据流分割成两个不同的集合。正如摘要所述，我们将利用 partitioningBy 方法，根据特定的条件将原始数据流分割成满足条件和不满足条件的两部分。这种方法相较于传统的循环方式，代码更加简洁高效。

使用 partitioningBy 方法分割数据流

Collectors.partitioningBy() 是 Java Stream API 提供的一个非常强大的收集器。它接收一个 Predicate 函数式接口作为参数，该接口定义了一个测试条件，用于判断流中的元素是否满足条件。partitioningBy() 方法会根据这个条件将流中的元素分成两部分：满足条件的元素会被收集到一个列表中，不满足条件的元素会被收集到另一个列表中。最终，partitioningBy() 方法会返回一个 Map<Boolean, List>，其中 Boolean 类型的键表示条件是否满足（true 表示满足，false 表示不满足），List 类型的值则分别存储了满足条件和不满足条件的元素列表。

示例代码

假设我们有一个 ID 列表 myIdList 和一个将 ID 映射到对象的 Map myObjectMap。我们需要根据 ID 是否存在于 myObjectMap 中，将 myIdList 分割成两个列表：一个包含存在于 myObjectMap 中的 ID 对应的对象，另一个包含 myObjectMap 中不存在的 ID。

以下是使用 partitioningBy 方法实现该功能的示例代码：

立即学习“Java免费学习笔记（深入）”；

SpeakingPass-打造你的专属雅思口语语料

使用chatGPT帮你快速备考雅思口语，提升分数

25 查看详情

import java.util.*;import java.util.stream.Collectors;public class StreamPartitioning {    public static void main(String[] args) {        List myIdList = Arrays.asList("a", "b", "c");        Map myObjectMap = new HashMap();        myObjectMap.put("b", "B");        Map<Boolean, List> partitioned = myIdList.stream()                .collect(Collectors.partitioningBy(myObjectMap::containsKey));        List