” />
本文將指導你如何使用 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
