利用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 List groupBy(List list, Function classifier) {
Map> grouped = list.stream().collect(Collectors.groupingBy(classifier));
return grouped.values().stream().flatMap(Collection::stream).collect(Collectors.toList());
}
public static void main(String[] args) {
List 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 groupedBySex = groupBy(students, Student::getSex);
List 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 的泛型参数确保了方法的灵活性和可重用性。
通过这种方式，我们巧妙地实现了根据不同条件动态传入方法引用进行分组的功能。
