java Stream groupby 只取第一个

原创

mob649e816594b7 2024-10-03 07:06:11 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e816594b7的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java Stream GroupBy 只取第一个元素的使用

在Java 8引入的Stream API中，我们可以使用流的处理方式来简化集合的操作。在常见的需求中，我们可能需要对一个集合进行分组，并只取每个分组中的第一个元素。下面，我们将深入探讨这个主题，通过代码示例、序列图和甘特图来帮助你更好地理解。

基础概念

首先，我们先了解一下 Stream 和 Collectors.groupingBy 的基本用法。Stream 允许我们以声明式的方式处理集合，Collectors.groupingBy 则是一个非常强大的工具，用于对流中的元素进行分组。

示例场景

假设我们有一个员工类 Employee，该类包含员工的名字和部门。我们的目标是以部门为分组，获取每个部门中的第一个员工。

import java.util.*;
import java.util.stream.Collectors;

class Employee {
    String name;
    String department;

    Employee(String name, String department) {
        this.name = name;
        this.department = department;
    }

    public String getName() {
        return name;
    }

    public String getDepartment() {
        return department;
    }

    @Override
    public String toString() {
        return "Employee{name='" + name + "', department='" + department + "'}";
    }
}

public class GroupByExample {
    public static void main(String[] args) {
        List<Employee> employees = Arrays.asList(
            new Employee("Alice", "HR"),
            new Employee("Bob", "Engineering"),
            new Employee("Charlie", "HR"),
            new Employee("David", "Engineering"),
            new Employee("Eve", "Finance")
        );

        Map<String, Employee> firstInDepartments = employees.stream()
            .collect(Collectors.groupingBy(Employee::getDepartment,
                    Collectors.mapping(e -> e, Collectors.toList())))
            .entrySet()
            .stream()
            .map(entry -> entry.getValue().get(0)) // 只取第一个元素
            .collect(Collectors.toList())
            .stream()
            .collect(Collectors.toMap(Employee::getDepartment, e -> e));

        firstInDepartments.forEach((department, employee) ->
                System.out.println(department + " -> " + employee));
    }
}

解释

我们首先创建了一个员工列表。
使用 Collectors.groupingBy 方法对员工按部门进行分组，这会返回一个 Map，键为部门，值为对应的员工列表。
接着，我们从每个部门的员工列表中取出第一个员工。
使用 Collectors.toMap 将结果转化为需要的格式。

这段代码的输出为：

HR -> Employee{name='Alice', department='HR'}
Engineering -> Employee{name='Bob', department='Engineering'}
Finance -> Employee{name='Eve', department='Finance'}

序列图

为了更好地理解代码执行的顺序，我们可以使用序列图来表示流程。以下是基于上述示例的序列图。

sequenceDiagram
    participant EmployeeList
    participant StreamAPI
    participant Collectors
    participant Map

    EmployeeList->>StreamAPI: 创建员工列表
    StreamAPI->>Collectors: group by 部门
    Collectors-->>StreamAPI: 返回 Map
    StreamAPI->>Map: 取出每个部门的第一个员工
    Map-->>StreamAPI: 返回第一个员工列表
    StreamAPI->>EmployeeList: 打印员工信息

甘特图

在日常开发过程中，合理规划任务是至关重要的。下面是一个相关的甘特图，展示了根据部门取第一个员工的任务安排。

gantt
    title 处理员工数据任务
    dateFormat  YYYY-MM-DD
    section 解析数据
    创建员工列表       :a1, 2023-10-01, 1d
    section 分组数据
    按部门分组员工    :a2, 2023-10-02, 1d
    section 获取第一个
    提取第一个员工    :a3, 2023-10-03, 1d
    section 输出结果
    打印输出员工信息  :a4, 2023-10-04, 1d