Java Stream：通过布尔谓词分为两个列表

Question

我有一个employees 的列表。他们有 isActive 布尔字段。我想将employees 分成两个列表：activeEmployees 和formerEmployees。是否可以使用 Stream API？最复杂的方法是什么？

Answer 1

Collectors.partitioningBy:

Map<Boolean, List<Employee>> partitioned = 
    listOfEmployees.stream().collect(
        Collectors.partitioningBy(Employee::isActive));

生成的映射包含两个列表，分别对应谓词是否匹配：

List<Employee> activeEmployees = partitioned.get(true);
List<Employee> formerEmployees = partitioned.get(false);

---

使用partitioningBy 而非groupingBy 有几个原因（正如Juan Carlos Mendoza 所建议的那样）：

首先，groupingBy 的参数是Function<Employee, Boolean>（在这种情况下），因此有可能向它传递一个可以返回 null 的函数，意味着如果这样，就会有第三个分区函数为任何员工返回 null。 partitioningBy 使用 Predicate<Employee>，因此它只能返回 2 个分区。 这将导致收集器抛出 NullPointerException：虽然没有明确记录，但会为空键明确抛出异常，大概是因为 Map.computeIfAbsent 的行为，即“如果函数返回 null，则不记录任何映射”，否则意味着元素将被默默地从输出中删除。 （感谢lczapski 指出这一点）。

其次，您会在结果映射中获得两个列表 (*)，partitioningBy；使用groupingBy，您只能获得元素映射到给定键的键/值对：

System.out.println(
    Stream.empty().collect(Collectors.partitioningBy(a -> false)));
// Output: {false=[], true=[]}

System.out.println(
    Stream.empty().collect(Collectors.groupingBy(a -> false)));
// Output: {}

---

(*) 此行为未记录在 Java 8 Javadoc 中，但已为 Java 9 添加。

Answer 2

在这种情况下，您也可以使用 groupingBy，因为有 2 个组的可能性（活跃和不活跃的员工）：

Map<Boolean, List<Employee>> grouped = employees.stream()
                .collect(Collectors.groupingBy(Employee::isActive));

List<Employee> activeEmployees = grouped.get(true);
List<Employee> formerEmployees = grouped.get(false);

Answer 3

最复杂的方法是什么？

Java 12 当然还有新的Collectors::teeing

List<List<Employee>> divided = employees.stream().collect(
      Collectors.teeing(
              Collectors.filtering(Employee::isActive, Collectors.toList()),
              Collectors.filtering(Predicate.not(Employee::isActive), Collectors.toList()),
              List::of
      ));

System.out.println(divided.get(0));  //active
System.out.println(divided.get(1));  //inactive

Answer 4

如果您愿意使用第三方库，则可以使用来自Eclipse Collections 的Collectors2.partition。

PartitionMutableList<Employee> partition =
        employees.stream().collect(
                Collectors2.partition(Employee::isActive, PartitionFastList::new));

List<Employee> activeEmployees = partition.getSelected();
List<Employee> formerEmployees = partition.getRejected();

您还可以使用ListIterate 来简化事情。

PartitionMutableList<Employee> partition =
        ListIterate.partition(employees, Employee::isActive);

List<Employee> activeEmployees = partition.getSelected();
List<Employee> formerEmployees = partition.getRejected();

PartitionMutableList 是从PartitionIterable 扩展而来的类型。 PartitionIterable 的每个子类型都有一个阳性结果 getSelected() 和阴性结果 getRejected() 的集合。

注意：我是 Eclipse Collections 的提交者。