利用Java Lambda表達式實現Java流式處理的動態分組
在Java流式處理中，Collectors.groupingBy 方法提供了強大的分組功能。但如果分組依據需要動態變化，例如有時按性別分組，有時按年齡分組，該如何實現呢？本文將演示如何使用Lambda表達式實現這種動態分組。
直接在 Collectors.groupingBy 中傳入變量是不可行的，因為它需要一個確定的方法引用。解決方法是利用Java的泛型和函數式接口 function。Function 接口接受一個輸入類型并返回一個輸出類型，這正好滿足動態傳入方法引用的需求。
我們可以創建一個新方法，該方法接收一個學生列表和一個 Function 對象作為參數。這個 Function 對象代表一個函數，它接受一個學生對象作為輸入，并返回用于分組的鍵值。
立即學習“Java免費學習筆記（深入）”；
以下是具體的實現代碼：
import java.util.Arrays; import java.util.Collection; import java.util.List; import java.util.Map; import java.util.function.Function; import java.util.stream.Collectors; class Student { String name; String sex; int age; public Student(String name, String sex, int age) { this.name = name; this.sex = sex; this.age = age; } public String getSex() { return sex; } public Integer getAge() { return age; } @Override public String toString() { return name + "(" + sex + "," + age + ")"; } } public class DynamicGrouping { public static <T, K> List<T> groupBy(List<T> list, Function<T, K> classifier) { Map<K, List<T>> grouped = list.stream().collect(Collectors.groupingBy(classifier)); return grouped.values().stream().flatMap(Collection::stream).collect(Collectors.toList()); } public static void main(String[] args) { List<Student> students = Arrays.asList( new Student("Alice", "female", 20), new Student("Bob", "male", 22), new Student("Charlie", "male", 23), new Student("Alice", "female", 21), new Student("Bob", "male", 22) ); List<Student> groupedBySex = groupBy(students, Student::getSex); List<Student> groupedByAge = groupBy(students, Student::getAge); System.out.println("Grouped by Sex: " + groupedBySex); System.out.println("Grouped by Age: " + groupedByAge); } }
在這個例子中，groupBy 方法接收一個列表和一個 Function 對象作為參數。Collectors.groupingBy 使用該 Function 對象進行分組，最后返回一個包含所有學生的新列表。在 main 方法中，我們可以分別傳入 Student::getSex 和 Student::getAge 來實現按性別和按年齡分組。 Function
通過這種方式，我們巧妙地實現了根據不同條件動態傳入方法引用進行分組的功能。
? 版權聲明
文章版權歸作者所有，未經允許請勿轉載。
THE END
