利用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
