Java stream sorted使用 Comparator 进行多字段排序的方法

时间:2023-03-23 楼兰胡杨人气:0

摘要：介绍使用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(类::属性二);

Reference

https://blog.csdn.net/weixin_48212105/article/details/126175816

加载全部内容