【JAVA】jdk8 Stream 排序精通

news2024/11/26 5:32:18

背景

jdk8的stream流能方便的排序，但是每次都要查资料，非常不方便，不确定，所以这次直接弄懂，不再迷茫。

转载请注明来源，创作不易，请多多支持。

基础排序

stream流大家应该都比较熟悉了，毕竟jdk8出来多久了，言简意赅的讲解下

stream 流提供的排序的方法其实就两个：
一个是无参的 Stream<T> sorted();
一个是有参的 Stream<T> sorted(Comparator<? super T> comparator);

基础写法-无参

无参的排序很简单，按照默认排序，测试代码如下

    @Data
    @Builder
    public static class User {
        private String username;
        private Integer age;
    }

    @Test
    public void StreamSort() {
        List<String> list1 = Lists.newArrayList("1", "2", "01", "02", "10", "11", "a");
        List<Integer> list2 = Lists.newArrayList(1, 2, 0, 3, 7);
        List<Boolean> list3 = Lists.newArrayList(true, false, true, false, false);
        List<User> list4 = Lists.newArrayList(User.builder().age(1).build()
                , User.builder().age(2).build()
                , User.builder().age(3).build());
        list1.stream().sorted().forEach(
                p -> System.out.print(p + " ")
        );
    }

结论：其实只要知道基础类型默认是怎么排序的就好了，对象类型本质还是指定到基础类型去，记住一个词 点名(从小到大)

1、字符串排序结果是 01 02 1 10 11 2 a,字符串比较很简单，就是字符依次比较，从小到大，第一位相同比第二位，依次排序。

2、数值类型排序结果 0 1 2 3 7,数值类型的都是一样的规律，从小到大和报数一样

3、布尔类型排序结果 false false false true true,可以类比 0是false 1是true 约定俗成 0为false 1为true, 依然是上升趋势

基础写法-有参

Comparator 是一个专门排序用的接口,只有一个核心方法 int compare(T o1, T o2);

严格的写法

一个匿名内部类实现排序的逻辑

        list2.stream().sorted(
                new Comparator<Integer>() {
                    @Override
                    public int compare(Integer p1, Integer p2) {
                        return p1 - p2;
                    }
                }
        ).forEach(p -> System.out.print(p + " "));

jdk8可以用lambda表达式简写，非数值类型的需要自定义规则

                (p1, p2) -> {
                    return p1 - p2;
                }
                
或者 进一步简写    (p1, p2) ->  p1 - p2

常用便捷写法

大部分其实都是数值比较,除了非数值类型的需要自定义规则，

比如要实现默认的字符串的比较，可以自定义以下逻辑

        list1.stream().sorted((p1, p2) -> {
                    char[] chars1 = p1.toCharArray();
                    char[] chars2 = p2.toCharArray();
                    //取短数组长度
                    for (int i = 0; i < Math.min(chars1.length, chars2.length); i++) {
                        if (chars1[i] != chars2[i]) {
                            return chars1[i] - chars2[i];
                        }
                    }
                    return 0;
                }
        ).forEach(p -> System.out.print(p + " "));

但是对于完全的数值比较，其实就按照默认的比较即可，给Stream 排序的值即可,特别是对象，只需要指定字段即可

        list4.stream().sorted(Comparator.comparingInt(User::getAge)).forEach(p -> System.out.print(p + " "));

支持的数值方法有 comparingInt 、comparingLong、comparingDouble

基础写法-倒序

默认的排序是点名,从小到大，而有时候我们需要从大到小

除了自定义的排序方法，还有两种快捷的方式

1、在已完成的Comparator对象上再调用 reversed()方法，依然返回Comparator对象，不过是翻转后的，从小到大就变成了从大到小

2、对于对象，可以直接使用Comparator.comparing(User::getAge,Comparator.naturalOrder()), 这样的快捷构造来指定排序。

第一个就是排序的字段，第二个表示自然的排序naturalOrder(从小到大) 还是倒序reverseOrder (从大到小)

因为对象比基本类型占用空间大，如果排完序之后再翻转，而同时数组长度比较大，那么就可能有性能问题，所以可以直接指定排序的方向，避免不必要的浪费

这一点可以追溯一下，这里传入的第二个参数，其实就是排序方法的实例
在这里插入图片描述

在这里插入图片描述
NaturalOrderComparator 比较是 c1.compareTo(c2);
而 ReverseComparator 比较是 c2.compareTo(c1);

那么就比较明显了，他不是最终翻转而是比较过程中就已经取反了

因此对象排序，优先使用这样的方式去指定排序，特别是倒序的时候

高级排序

部分高级排序用法，程序设计的非常完善，有效好用

空值处理

Stream流处理不会处理空值，也就是Null，如果原始Stream流里面有空值，那么就直接会报空指针异常

如果可能有空值，一般需要过滤掉空值之后再排序，filter 返回布尔值，只保留返回true的数据

        list4.stream().filter(p -> p.getAge() != null)
                .sorted(Comparator.comparing(User::getAge, Comparator.reverseOrder()))
                .forEach(p -> System.out.print(p + " "));

如果空值还需要处理(比如追加列表最后)，一般用单独的数组存放

        List<User> emptyList = list4.stream().filter(p -> p.getAge() == null).collect(Collectors.toList());
        List<User> resList = list4.stream().filter(p -> p.getAge() != null)
                .sorted(Comparator.comparing(User::getAge, Comparator.reverseOrder()))
                .collect(Collectors.toList());
        resList.addAll(emptyList);

多字段排序

可能单个字段排序不够，需要两次排序，甚至多次排序

其实也已经提供了现成的方法,thenComparing,同样返回的还是Comparator对象，那么可以继续追加

比如我的原始数据是

        List<User> list4 = Lists.newArrayList(User.builder().age(1).score(2).build()
                , User.builder().age(2).score(2).build()
                , User.builder().age(2).score(1).build()
                , User.builder().age(2).score(8).build()
                , User.builder().age(3).score(8).username("qiushi").build()
                , User.builder().age(3).score(5).build());

而排序方法是

.sorted(Comparator.comparing(User::getAge, Comparator.naturalOrder()).thenComparing(User::getScore, Comparator.naturalOrder()))

那么最终结果就是先 age 顺序，score顺序

User(username=null, age=1, score=2)
User(username=null, age=2, score=1)
User(username=null, age=2, score=2)
User(username=null, age=2, score=8)
User(username=null, age=3, score=5)
User(username=qiushi, age=3, score=8)

所以你明白了么？先写到这里，后续再来补充，欢迎讨论指正，biu~