如何在应用嵌套分组时保留所有子组通过收集器

Question

我正在尝试按性别和部门对员工列表进行分组。

如何确保所有部门都包含在每个性别的排序顺序中，即使相关性别计数为零嗯>？

目前，我有以下代码和输出

employeeRepository.findAll().stream()
            .collect(Collectors.groupingBy(Employee::getGender, 
                        Collectors.groupingBy(Employee::getDepartment, 
                                              Collectors.counting())));

//output
//{MALE={HR=1, IT=1}, FEMALE={MGMT=1}}

首选输出是：

{MALE={HR=1, IT=1, MGMT=0}, FEMALE={HR=0, IT=0, MGMT=1}}

Answer 1

为此，首先必须按部门分组，然后才按性别分组，而不是相反。

第一个收集器 groupingBy(Employee::getDepartment, _downstream_ ) 将根据部门将数据集拆分为组。由于下游收集器 partitioningBy(employee -> employee.getGender() == Employee.Gender.MALE, _downstream_ ) 将被应用，它将映射到每个部门的数据划分为 根据员工性别分为两部分。最后，作为下游应用的 Collectors.counting() 将提供每个部门<的每个性别员工的总数 /强>。

因此，由 collect() 操作生成的中间 map 类型将为 Map> - < em>每个部门按性别划分的员工人数（布尔值）（为了简单起见，部门是一个纯字符串< /em>）。

下一步将此映射转换为 Map> - 每个部门的员工人数 性别。

我的方法是在条目集上创建一个流，并将每个条目替换为一个新条目，该新条目将性别作为其键，并且为了保留有关部门的信息，其值依次将是一个以部门为键和一个带有部门的条目>按部门计算作为其值。

然后通过条目键使用groupingBy收集条目流。应用映射作为下游收集器来提取嵌套条目。然后应用Collectors.toMap()将Map.Entry类型的条目收集到地图中。

所有部门都包含在排序顺序中

为了确保嵌套地图中的顺序（按计数划分）NavigableMap 应该使用。

为此，需要使用需要 mapFactory 的 toMap() 风格（它还需要一个 mergeFunction这对于此任务来说并不是真正有用，因为不会有重复项，但也必须提供它）。

public static void main(String[] args) {
    List<Employee> employeeRepository = 
            List.of(new Employee("IT", Employee.Gender.MALE),
                    new Employee("HR", Employee.Gender.MALE),
                    new Employee("MGMT", Employee.Gender.FEMALE));

    Map<Employee.Gender, NavigableMap<String, Long>> departmentCountByGender = employeeRepository
            .stream()
            .collect(Collectors.groupingBy(Employee::getDepartment, // Map<String, Map<Boolean, Long>> - department to *employee count* by gender
                        Collectors.partitioningBy(employee -> employee.getGender() == Employee.Gender.MALE,
                                                  Collectors.counting())))
            .entrySet().stream()
            .flatMap(entryDep -> entryDep.getValue().entrySet().stream()
                    .map(entryGen -> Map.entry(entryGen.getKey() ? Employee.Gender.MALE : Employee.Gender.FEMALE,
                                               Map.entry(entryDep.getKey(), entryGen.getValue()))))
            .collect(Collectors.groupingBy(Map.Entry::getKey,
                        Collectors.mapping(Map.Entry::getValue,
                                Collectors.toMap(Map.Entry::getKey,
                                                 Map.Entry::getValue,
                                                 (v1, v2) -> v1,
                                                 TreeMap::new))));

    System.out.println(departmentCountByGender);
}

用于演示目的的虚拟 Employee class：

class Employee {
    enum Gender {FEMALE, MALE};

    private String department;
    private Gender gender;
    // etc.
    
    // constructor, getters
}

输出

{FEMALE={HR=0, IT=0, MGMT=1}, MALE={HR=1, IT=1, MGMT=0}}

Answer 2

您可以继续处理代码的结果：

List<String> deptList = employees.stream().map(Employee::getDepartment).sorted().toList();

Map<Gender, Map<String, Long>> tmpResult = employees.stream()
        .collect(Collectors.groupingBy(Employee::getGender, Collectors.groupingBy(Employee::getDepartment, Collectors.counting())));

Map<Gender, Map<String, Long>> finalResult = new HashMap<>();

for (Map.Entry<Gender, Map<String, Long>> entry : tmpResult.entrySet()) {
    Map<String, Long> val = new LinkedHashMap<>();
    for (String dept : deptList) {
        val.put(dept, entry.getValue().getOrDefault(dept, 0L));
    }

    finalResult.put(entry.getKey(), val);
}

System.out.print(finalResult);

如果您想用一行代码获得结果，则代码的可读性或可维护性可能不会很好。

但是，如果您不介意使用第三方库，还有一种选择： abacus-common

Map<Gender, Map<String, Integer>> result = Stream.of(employees)
        .groupByToEntry(Employee::getGender, MoreCollectors.countingIntBy(Employee::getDepartment)) // step 1) group by gender 
        .mapValue(it -> Maps.newMap(deptList, Fn.identity(), dept -> it.getOrDefault(dept, 0), IntFunctions.ofLinkedHashMap())) // step 2) process the value.
        .toMap();

声明：我是 abacus-common 的开发者