使用 Java Stream API 分割数据流并获取不同的集合

本文将介绍如何使用 Java Stream API 有效地分割数据流，根据特定条件将元素划分到不同的集合中。通过 partitioningBy 收集器，我们可以将数据流分割成两个列表，分别包含满足条件和不满足条件的元素，从而避免使用传统的循环结构，提高代码的可读性和简洁性。

在实际开发中，我们经常需要根据某些条件将数据进行分类处理。例如，我们有一个 ID 列表和一个 ID 到对象的映射，需要将列表中存在于映射中的 ID 对应的对象放入一个列表，而将不存在于映射中的 ID 放入另一个列表。传统的做法是使用循环遍历列表，然后逐个判断，代码较为冗长。Java Stream API 提供了一种更简洁、更高效的方式来实现这个需求。

使用 partitioningBy 收集器

Collectors.partitioningBy() 是一个非常有用的收集器，它可以根据给定的 Predicate（谓词）将数据流分割成两个列表。Predicate 是一个函数式接口，用于判断元素是否满足某个条件。partitioningBy 收集器返回一个 Map<Boolean, List>，其中 key 为 true 的列表包含满足条件的元素，key 为 false 的列表包含不满足条件的元素。

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

示例代码

以下代码演示了如何使用 partitioningBy 收集器来分割 ID 列表，并将对应的对象和缺失的 ID 分别放入不同的列表中：

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