java lambada groupby 多个字段

原创

mob64ca12e20c7d 2024-09-07 05:13:12 ©著作权

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

使用Java Lambda对多个字段进行分组

在Java中使用Lambda表达式来对数据集合进行分组是非常常见的操作，尤其是在处理与数据库相关的数据时。今天，我将带你一步一步地学习如何在Java中使用Lambda表达式按照多个字段进行分组。以下是整个操作的流程：

整体流程表

步骤	描述
1	定义数据类
2	创建数据集合
3	使用`Stream`和`Collectors`进行分组

下面，我们将详细介绍每一步所需的代码和注释。

1. 定义数据类

首先，我们需要定义一个数据类，用于表示我们要处理的数据。假设我们在处理一个“用户”类，其中包含用户的姓名、年龄和城市。

public class User {
    private String name; // 用户名
    private int age;     // 用户年龄
    private String city; // 用户所在城市

    // 构造器
    public User(String name, int age, String city) {
        this.name = name;
        this.age = age;
        this.city = city;
    }

    // Getter方法
    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }

    public String getCity() {
        return city;
    }
}

解析

定义了一个名为User的类，包含用户名、年龄和城市属性。
提供了构造器以及相应的getter方法。

2. 创建数据集合

接下来，我们需要创建一个用户集合，模拟实际数据的来源。

import java.util.ArrayList;
import java.util.List;

List<User> users = new ArrayList<>();
users.add(new User("Alice", 30, "New York"));
users.add(new User("Bob", 25, "Los Angeles"));
users.add(new User("Charlie", 30, "New York"));
users.add(new User("David", 25, "Los Angeles"));
users.add(new User("Eve", 30, "Chicago"));

解析

使用ArrayList来存储多个User对象。
向users集合中添加了几位用户的信息。

3. 使用Stream和Collectors进行分组

我们将需要使用Stream API及Collectors.groupingBy()方法来按照年龄和城市对用户进行分组。

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

Map<String, Map<Integer, List<User>>> groupedUsers = users.stream()
    .collect(Collectors.groupingBy(User::getCity, 
            Collectors.groupingBy(User::getAge)));

解析

users.stream()：将用户集合转换为流。
Collectors.groupingBy()：用于接收一个键提取函数，并根据返回的键对流中的元素进行分组。
本例中，我们先按照城市分组，再按照年龄进行分组，形成嵌套的Map。

一张关系图

erDiagram
    USER {
        string name
        int age
        string city
    }
    USER }|--|{ CITY : contains

解析

这里展示了用户（User）和城市（City）的关系，一个用户属于一个城市。

用例

后面我们可以使用groupedUsers来输出结果。

groupedUsers.forEach((city, ageGroup) -> {
    System.out.println("City: " + city);
    ageGroup.forEach((age, userList) -> {
        System.out.println("  Age: " + age + " users: " + userList.size());
    });
});

解析

forEach用于迭代groupedUsers的每个条目，输出每个城市下的年龄和用户数量。

一张序列图

sequenceDiagram
    participant User as 用户
    participant Group as 分组处理

    User->>Group: 提交用户数据
    Group-->>User: 返回分组结果

解析

这里展示了用户提交数据至分组处理的简单流程。

完整代码示例

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

public class Main {
    public static void main(String[] args) {
        // Step 1: 创建用户集合
        List<User> users = new ArrayList<>();
        users.add(new User("Alice", 30, "New York"));
        users.add(new User("Bob", 25, "Los Angeles"));
        users.add(new User("Charlie", 30, "New York"));
        users.add(new User("David", 25, "Los Angeles"));
        users.add(new User("Eve", 30, "Chicago"));

        // Step 2: 分组
        Map<String, Map<Integer, List<User>>> groupedUsers = users.stream()
            .collect(Collectors.groupingBy(User::getCity, 
                     Collectors.groupingBy(User::getAge)));

        // Step 3: 打印分组结果
        groupedUsers.forEach((city, ageGroup) -> {
            System.out.println("City: " + city);
            ageGroup.forEach((age, userList) -> {
                System.out.println("  Age: " + age + " users: " + userList.size());
            });
        });
    }
}