” />
本文介紹了如何使用Java 8 Stream API將一個包含Trip對象的List轉換為map
在實際開發中，經常會遇到需要根據嵌套對象的屬性對集合進行分組的需求。例如，我們有一個包含Trip對象的列表，每個Trip對象包含一個Employee列表，我們希望將所有Trip對象按照Employee的empId進行分組，生成一個Map
Java 8 Stream API提供了強大的數據處理能力，可以簡潔高效地實現這一需求。
實體類定義
首先，我們定義兩個實體類：Trip 和 Employee。
立即學習“Java免費學習筆記（深入）”；
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; } @Data @NoArgsConstructor @AllArgsConstructor public class Employee { private String name; private String empId; }
使用Stream API進行轉換
核心在于使用 flatMap 和 groupingBy Collectors。flatMap 用于將 Stream
如果使用的是 Java 16 或更高版本，可以使用 record 來簡化 TripEmployee 的定義：
public record TripEmployee(String empId, Trip trip) {}
如果使用的是 Java 8，則需要使用普通的 class 來定義 TripEmployee。
完整的代碼示例如下：
import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; import java.util.stream.Collectors; public class TripGrouping { public record TripEmployee(String empId, Trip trip) {} public static void main(String[] args) { // 示例數據 List<Trip> trips = new ArrayList<>(); List<Employee> empList1 = List.of(new Employee("Alice", "101"), new Employee("Bob", "102")); List<Employee> empList2 = List.of(new Employee("Charlie", "103"), new Employee("David", "101")); trips.add(new Trip(new Date(), new Date(), empList1)); trips.add(new Trip(new Date(), new Date(), empList2)); // 使用 Stream API 進行轉換 Map<String, List<Trip>> empMap = 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()) )); // 打印結果 System.out.println(empMap); } }
代碼解釋:
- trips.stream(): 將 List
轉換為 Stream 。 - flatMap(trip -> trip.getEmpList().stream().map(emp -> new TripEmployee(emp.getEmpId(), trip))):
- trip.getEmpList().stream(): 獲取每個 Trip 對象的 Employee 列表，并將其轉換為 Stream
。 - map(emp -> new TripEmployee(emp.getEmpId(), trip)): 將每個 Employee 對象轉換為 TripEmployee 對象，其中包含 empId 和 Trip 對象的引用。
- flatMap(): 將所有 Stream
對象合并為一個 Stream 。
- trip.getEmpList().stream(): 獲取每個 Trip 對象的 Employee 列表，并將其轉換為 Stream
- collect(Collectors.groupingBy(TripEmployee::empId, Collectors.mapping(TripEmployee::trip, Collectors.toList()))):
- groupingBy(TripEmployee::empId): 根據 TripEmployee 對象的 empId 進行分組。
- mapping(TripEmployee::trip, Collectors.toList()): 將每個分組中的 TripEmployee 對象轉換為 Trip 對象，并將所有 Trip 對象收集到一個 List
中。
注意事項
- 如果 Employee 列表為空，則不會生成任何 TripEmployee 對象。
- 如果多個 Employee 對象具有相同的 empId，則它們對應的 Trip 對象將出現在同一個 List
中。 - 確保正確導入所需的類和包，例如 java.util.stream.Collectors。
總結
本文介紹了如何使用Java 8 Stream API將一個包含Trip對象的List轉換為Map
