Java 17中用Stream API实现分组聚合判断活跃状态

假设你的实体类如下（用Java17的record简化，普通类同理）：

// 实体类：包含分组名和活跃标记
record PropItem(String propGroupName, boolean propValActive) {}

短路聚合的Stream实现 #

要实现组内只要有一个propValActive为true就返回true，全为false才返回false的短路逻辑，需要自定义Collector——因为默认收集器（比如reducing）会遍历组内所有元素，无法做到短路。

核心代码如下：

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Collector;

public class GroupShortCircuitExample {
    public static void main(String[] args) {
        List<PropItem> items = List.of(
            new PropItem("groupA", false),
            new PropItem("groupA", true),
            new PropItem("groupA", false),
            new PropItem("groupB", false),
            new PropItem("groupB", false)
        );

        Map<String, Boolean> groupedResult = items.stream()
            .collect(Collectors.groupingBy(
                PropItem::propGroupName,
                // 自定义短路Collector
                Collector.of(
                    // 1. 供应容器：用boolean数组存结果（基本类型数组支持修改）
                    () -> new boolean[]{false},
                    // 2. 累加器：仅当当前结果为false时才更新，遇到true后不再处理后续元素
                    (container, item) -> {
                        if (!container[0]) {
                            container[0] = item.propValActive();
                        }
                    },
                    // 3. 合并器：并行流场景下，只要任一子结果为true，合并后结果为true
                    (leftContainer, rightContainer) -> {
                        leftContainer[0] = leftContainer[0] || rightContainer[0];
                        return leftContainer;
                    },
                    // 4. 收尾器：将容器中的boolean转为Boolean
                    container -> container[0]
                )
            ));

        // 输出结果：{groupA=true, groupB=false}
        System.out.println(groupedResult);
    }
}

自定义Collector的短路逻辑说明 #

• Supplier：创建boolean[]作为容器，因为基本类型boolean不可变，数组作为引用类型可在累加器中修改值。
• Accumulator：仅当当前容器结果为false时，才判断当前元素的propValActive。一旦容器值变为true，后续元素不会触发任何更新，实现串行流的短路。
• Combiner：处理并行流的分片合并逻辑，只要任意分片结果为true，合并后的结果就是true。
• Finisher：将数组中的原始boolean自动装箱为Boolean，适配Map的泛型要求。

非短路实现对比（参考） #

如果你之前用的是遍历所有元素的非短路实现，比如reducing，代码如下：

Map<String, Boolean> nonShortCircuitResult = items.stream()
    .collect(Collectors.groupingBy(
        PropItem::propGroupName,
        Collectors.reducing(false, PropItem::propValActive, Boolean::logicalOr)
    ));

这个实现会遍历组内所有元素，哪怕中途已经遇到true，性能略低于短路实现，尤其是组内元素数量较多时。

内容的提问来源于stack exchange，提问作者Nzall