如何使用Java Streams提取指定数据生成HashMap统计主管下属人数

错误原因 #

你原有代码报错的核心问题是：提前通过map操作将Stream<Employee>转换为了仅存储主管字符串的Stream<String>，此时groupingBy没有指定正确的分组键规则。同时额外的map操作属于冗余步骤，完全可以在收集阶段直接完成字段提取，逻辑更简洁。

最优实现方案 #

不需要提前做map转换，直接在收集器中指定分组规则即可：

// 需提前导入相关工具类
import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;

// 通用实现，返回默认Map实现
Map<String, Long> supervisorEmpCount = streamWithEmployees
    .collect(Collectors.groupingBy(
        this::supervisorofEmployee, // 按员工的直属主管作为分组键
        Collectors.counting() // 统计每个分组的员工数量
    ));

// 若你明确要求返回HashMap类型，使用三参数的groupingBy重载
HashMap<String, Long> supervisorEmpCountHashMap = streamWithEmployees
    .collect(Collectors.groupingBy(
        this::supervisorofEmployee,
        HashMap::new, // 指定生成HashMap实例
        Collectors.counting()
    ));

兼容你原有写法的修正方案 #

如果你希望保留提前map提取主管的逻辑，只需要将groupingBy的分组键指定为元素本身即可：

import java.util.function.Function;

Map<String, Long> mp = streamWithEmployees
    .map(x -> supervisorofEmployee(x))
    .collect(Collectors.groupingBy(
        Function.identity(), // 以流中的主管字符串本身作为分组键
        Collectors.counting()
    ));

补充注意事项 #

1. 你定义的Employee记录中第三个字段类型写为小写string，Java中字符串类型为大写开头的String，需要修正为String imm_supervisor否则会编译失败。
2. Collectors.counting()返回的value类型为Long，如果你需要Integer类型的计数，可以将下游收集器替换为Collectors.summingInt(e -> 1)。

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