写代码时如果不想通过SQL查询做分组统计,通过Stream流也是可以做到的,现用简单demo记录一下。
先创建一个学生类
import lombok.AllArgsConstructor;
import lombok.Data;
@Data
@AllArgsConstructor
public class Student {
private Integer id;
private String name;
private String sex;
private Integer age;
private Integer classNumber;
}
然后造几个数据,放到一个List里面
import java.util.ArrayList;
import java.util.List;
public class Test {
public static void main(String[] args) {
List<Student> studentList = new ArrayList<Student>();
studentList.add(new Student(1, "张三", "男", 20, 2));
studentList.add(new Student(2, "李四", "男", 21, 1));
studentList.add(new Student(3, "王五", "女", 19, 1));
studentList.add(new Student(4, "赵六", "男", 19, 1));
studentList.add(new Student(5, "王大锤", "男", 20, 2));
}
1.如果需要对学生集合按班级分组,则可以这样写
Map<Integer, List<Student>> studentGroup = studentList.stream().collect(Collectors.groupingBy(Student::getClassNumber));
for(Map.Entry<Integer, List<Student>> entry : studentGroup.entrySet()){
System.out.println("key :"+entry.getKey()+", value :"+entry.getValue());
}
运行结果:
key :1, value :[Student(id=2, name=李四, sex=男, age=21, classNumber=1), Student(id=3, name=王五, sex=女, age=19, classNumber=1), Student(id=4, name=赵六, sex=男, age=19, classNumber=1)]
key :2, value :[Student(id=1, name=张三, sex=男, age=20, classNumber=2), Student(id=5, name=王大锤, sex=男, age=20, classNumber=2)]
2.如果是对学生按班级分组,并统计每个班的人数
Map<Integer, Long> map = studentList.stream().collect(Collectors.groupingBy(Student::getClassNumber, Collectors.counting()));
for(Map.Entry<Integer, Long> entry : map.entrySet()){
System.out.println("key :"+entry.getKey()+", value :"+entry.getValue());
}
运行结果
key :1, value :3
key :2, value :2