使用 Java 8 Stream API 将 List 转换为 Map_技术教程

">" />

本文将指导你如何使用 Java 8 Stream API 将 List 转换为 Map>。这种转换在处理具有嵌套结构的数据时非常有用，例如当我们需要根据嵌套对象的属性对外部对象进行分组时。

使用 Stream API 进行转换

假设我们有以下两个领域类：Trip 和 Employee。

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Date;
import java.util.List;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Trip {
    private Date startTime;
    private Date endTime;
    List empList;
}

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Employee {
    private String name;
    private String empId;
}

我们的目标是将一个 List 转换为 Map>，其中 Map 的 Key 是 Employee 对象的 empId，Value 是包含该 Employee 参与的所有 Trip 对象的列表。

为了实现这个目标，我们需要使用 Stream API 中的 flatMap 和 groupingBy 操作。flatMap 用于将一个 Stream 中的每个元素转换为另一个 Stream，并将所有生成的 Stream 连接成一个 Stream。groupingBy 用于将 Stream 中的元素按照指定的条件进行分组。

首先，我们需要创建一个辅助类，用于保存 Employee 的 empId 和对应的

Trip 对象。在 Java 16 及以上版本，可以使用 record 来简化这个过程。

public record TripEmployee(String empId, Trip trip) {}

对于Java 8，可以使用如下的class来实现同样的功能：

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@NoArgsConstructor
@AllArgsConstructor
class TripEmployee {
    private String empId;
    private Trip trip;
}

接下来，我们可以使用以下代码将 List 转换为 Map>：

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

public class TripConverter {

    public static Map> convertTripsToEmployeeMap(List trips) {
        return trips.stream()
                .flatMap(trip -> trip.getEmpList().stream()
                        .map(emp -> new TripEmployee(emp.getEmpId(), trip))
                )
                .collect(Collectors.groupingBy(
                        TripEmployee::empId,
                        Collectors.mapping(TripEmployee::trip,
                                Collectors.toList())
                ));
    }

    public static void main(String[] args) {
        // 创建一些示例数据
        List employees1 = new ArrayList<>();
        employees1.add(new Employee("Alice", "1"));
        employees1.add(new Employee("Bob", "2"));

        List employees2 = new ArrayList<>();
        employees2.add(new Employee("Charlie", "3"));
        employees2.add(new Employee("Alice", "1"));

        Trip trip1 = new Trip(new java.util.Date(), new java.util.Date(), employees1);
        Trip trip2 = new Trip(new java.util.Date(), new java.util.Date(), employees2);

        List trips = new ArrayList<>();
        trips.add(trip1);
        trips.add(trip2);

        // 进行转换
        Map> employeeTripMap = convertTripsToEmployeeMap(trips);

        // 打印结果
        employeeTripMap.forEach((empId, tripList) -> {
            System.out.println("Employee ID: " + empId);
            tripList.forEach(trip -> System.out.println("  Trip: " + trip));
        });
    }
}

这段代码首先使用 flatMap 将每个 Trip 对象转换为一个 Stream，其中每个 TripEmployee 对象包含一个 Employee 的 empId 和对应的 Trip 对象。然后，使用 groupingBy 将 TripEmployee 对象按照 empId 进行分组，并将每个 empId 对应的 Trip 对象收集到一个列表中。

代码解释

trips.stream(): 将 List 转换为一个 Stream。
flatMap(trip -> trip.getEmpList().stream().map(emp -> new TripEmployee(emp.getEmpId(), trip))): 对于 Stream 中的每个 Trip 对象，获取其 empList，将其转换为一个 Stream，并将每个 Employee 对象映射为一个 TripEmployee 对象。flatMap 将所有生成的 Stream 连接成一个 Stream。
collect(Collectors.groupingBy(TripEmployee::empId, Collectors.mapping(TripEmployee::trip, Collectors.toList()))): 使用 groupingBy 将 TripEmployee 对象按照 empId 进行分组。Collectors.mapping 用于将每个 empId 对应的 TripEmployee 对象映射为 Trip 对象，并使用 Collectors.toList() 将所有 Trip 对象收集到一个列表中。

注意事项

确保 Trip 类的 getEmpList() 方法返回的 List 不为 null，否则可能会导致 NullPointerException。
如果 Employee 类的 getEmpId() 方法返回 null 或空字符串，可能会导致 groupingBy 的结果出现问题。建议在代码中进行空值或空字符串的检查。
record 是 Java 16 引入的特性，如果你的项目使用的是 Java 8，你需要使用 class 来代替 record。

总结

本文介绍了如何使用 Java 8 Stream API 将一个 Trip 对象的列表转换为一个 Map>，其中 Map 的 Key 是员工 ID，Value 是包含该员工参与的所有 Trip 对象的列表。通过使用 flatMap 和 groupingBy 组合的 Stream 操作，我们可以简洁而高效地实现这种转换。这种方法在处理具有嵌套结构的数据时非常有用，可以帮助我们更好地组织和分析数据。