主要介绍使用Java Stream流排序器Comparator对List集合进行多字段排序的方法,包括复杂实体对象多字段升降序排序方法。
工作中,一般使用SQL中的order by进行排序,但有时候在Java代码中进行排序,例如合并多个list对象的数据后,以年龄降序排列,这显然是无法通过SQL语句搞定的,而一般的冒泡排序、希尔排序等需要手写实现,容易出错,而且代码量大,测试工作量自然不容小觑。这时,就需要搬出Stream sort方法进行排序,重写其中的Comparator。
重写Comparable接口
重写compareTo方法实现排序,即流中泛型元素需实现Comparable接口
import lombok.*;
@Data
public class Student implements Comparable<Student> {
private int id;
private String name;
private int age;
@Override
public int compareTo(Student ob) {
return name.compareTo(ob.getName());
}
@Override
public boolean equals(final Object obj) {
if (obj == null) {
return false;
}
final Student std = (Student) obj;
if (this == std) {
return true;
} else {
return (this.name.equals(std.name) && (this.age == std.age));
}
}
@Override
public int hashCode() {
int hashno = 7;
hashno = 13 * hashno + (name == null ? 0 : name.hashCode());
return hashno;
}
}
缺点是所有类都会使用这个排序规则,不适用于排序规则灵活多变的复杂业务场景。
使用Comparator排序
使用stream的sorted(Comparator com)基于自定义规则排序,这需要自定义Comparator排序器。
sorted排序结果默认升序排序:
list = list.stream().sorted().collect(Collectors.toList());
下面是根据年龄升序排序的示例:
list = list.stream().sorted(Comparator.comparing(Student::getAge))
.collect(Collectors.toList());
如果想实现降序排列,可以使用Comparator 提供的reverseOrder() 方法
list = list.stream().sorted(Comparator.reverseOrder()).collect(Collectors.toList());
下面是根据年龄降序排列的示例:
list = list.stream().sorted(Comparator.comparing(Student::getAge).reversed())
.collect(Collectors.toList());
像Integer、Long等这些基本类型的包装类已经实现了Comparable接口,在使用sorted的时候,可以使用comparingInt、thenComparingInt、thenComparingLong等。
多字段排序
对象集合以类属性一升序、属性二升序排序:
Comparator<类> comparator = Comparator.comparing(类::属性一).thenComparing(类::属性二);
list=list.stream().sorted(comparator).collect(Collectors.toList());
例如,先按学生姓名升序,姓名相同时则按年龄升序。
List<Student> sortedList=list.sorted(Comparator.comparing(Student::getName).thenComparing(Student::getAge))
.collect(Collectors.toList());
sortedList.stream().forEach(System.out::println);
升序结果以属性一降序,属性二升序排列:
Comparator<类> comparator = Comparator.comparing(类::属性一,Comparator.reverseOrder()).thenComparing(类::属性二);
list=list.stream().sorted(comparator).collect(Collectors.toList());
这里自定义了一个比较器对象,修改对象排序规则即可。如果某个属性需要降序,则在comparing中声明Comparator.reverseOrder(),例如:
Comparator<Student> comparator = Comparator.comparing(Student::getName, Comparator.reverseOrder()).thenComparing(Student::getAge)
list=list.sorted(comparator).collect(Collectors.toList());
当然了,也可以把Comparator.reverseOrder()放到属性二的位置,此时表示以属性一升序、属性二降序排列:
list=list.stream().sorted(Comparator.comparing(类::属性一).thenComparing(类::属性二,Comparator.reverseOrder()))
.collect(Collectors.toList());
注意事项
sorted()方法返回的结果集是一个新的对象,和被排序对象的引用不一样。
1、降序排列时,只需要在 comparator 末尾写一个 reversed(),不需要每个比较属性都写
Comparator<类> comparator1 =
Comparator.comparing(类::属性一).thenComparing(类::属性二).reversed();
但是,不建议这样写,推荐如下语义更清晰的语法糖:
Comparator<类> comparator1 = Comparator.comparing(类::属性一, Comparator.reverseOrder()).thenComparing(类::属性二, Comparator.reverseOrder())
2、构建比较器时如果分两行,不能写成下列形式,否则会排序不正确
Comparator<类> comparator2 = Comparator.comparing(类::属性一);
comparator2.thenComparing(类::属性二);
可以写成
Comparator<类> comparator2 = Comparator.comparing(类::属性一);
comparator2 = comparator2.thenComparing(类::属性二);
评论区