” />
本文将指导你如何使用 Java 8 Stream API 将 List
使用 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<Employee> empList; }
import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; @Data @NoArgsConstructor @AllArgsConstructor public class Employee { private String name; private String empId; }
我们的目标是将一个 List
为了实现这个目标，我们需要使用 Stream API 中的 flatMap 和 groupingBy 操作。flatMap 用于将一个 Stream 中的每个元素转换为另一个 Stream，并将所有生成的 Stream 连接成一个 Stream。groupingBy 用于将 Stream 中的元素按照指定的条件进行分组。
立即学习“Java免费学习笔记（深入）”；
首先，我们需要创建一个辅助类，用于保存 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
import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.stream.Collectors; public class TripConverter { public static Map<String, List<Trip>> convertTripsToEmployeeMap(List<Trip> 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<Employee> employees1 = new ArrayList<>(); employees1.add(new Employee("Alice", "1")); employees1.add(new Employee("Bob", "2")); List<Employee> 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<Trip> trips = new ArrayList<>(); trips.add(trip1); trips.add(trip2); // 进行转换 Map<String, List<Trip>> employeeTripMap = convertTripsToEmployeeMap(trips); // 打印结果 employeeTripMap.forEach((empId, tripList) -> { System.out.println("Employee ID: " + empId); tripList.forEach(trip -> System.out.println(" Trip: " + trip)); }); } }
这段代码首先使用 flatMap 将每个 Trip 对象转换为一个 Stream
代码解释
- 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
