Java如何使用 Lambda 表达式实现超强的排序功能
这篇文章主要介绍Java如何使用 Lambda 表达式实现超强的排序功能,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完!
首先,我们定义一个基础类,后面我们将根据这个基础类演示如何在内存中排序。
@Data@NoArgsConstructor@AllArgsConstructorpublic class Student { private String name; private int age; @Override public boolean equals(Object o) { if (this == o) { return true; } if (o == null || getClass() != o.getClass()) { return false; } Student student = (Student) o; return age == student.age && Objects.equals(name, student.name); } @Override public int hashCode() { return Objects.hash(name, age); }}
基于Comparator排序
在 Java8 之前,我们都是通过实现Comparator
接口完成排序,比如:
new Comparator() { @Override public int compare(Student h2, Student h3) { return h2.getName().compareTo(h3.getName()); }};
这里展示的是匿名内部类的定义,如果是通用的对比逻辑,可以直接定义一个实现类。使用起来也比较简单,如下就是应用:
@Testvoid baseSortedOrigin() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); Collections.sort(students, new Comparator () { @Override public int compare(Student h2, Student h3) { return h2.getName().compareTo(h3.getName()); } }); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
这里使用了 Junit5 实现单元测试,用来验证逻辑非常适合。
因为定义的Comparator
是使用name
字段排序,在 Java 中,String
类型的排序是通过单字符的 ASCII 码顺序判断的,J
排在T
的前面,所以Jerry
排在第一个。
使用 Lambda 表达式替换Comparator匿名内部类
使用过 Java8 的 Lamdba 的应该知道,匿名内部类可以简化为 Lambda 表达式为:
Collections.sort(students, (Student h2, Student h3) -> h2.getName().compareTo(h3.getName()));
在 Java8 中,List
类中增加了sort
方法,所以Collections.sort
可以直接替换为:
students.sort((Student h2, Student h3) -> h2.getName().compareTo(h3.getName()));
根据 Java8 中 Lambda 的类型推断,我们可以将指定的Student
类型简写:
students.sort((h2, h3) -> h2.getName().compareTo(h3.getName()));
至此,我们整段排序逻辑可以简化为:
@Testvoid baseSortedLambdaWithInferring() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); students.sort((h2, h3) -> h2.getName().compareTo(h3.getName())); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
通过静态方法抽取公共的 Lambda 表达式
我们可以在Student
中定义一个静态方法:
public static int compareByNameThenAge(Student s1, Student s2) { if (s1.name.equals(s2.name)) { return Integer.compare(s1.age, s2.age); } else { return s1.name.compareTo(s2.name); }}
这个方法需要返回一个int
类型参数,在 Java8 中,我们可以在 Lambda 中使用该方法:
@Testvoid sortedUsingStaticMethod() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); students.sort(Student::compareByNameThenAge); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
借助Comparator的comparing方法
在 Java8 中,Comparator
类新增了comparing
方法,可以将传递的Function
参数作为比较元素,比如:
@Testvoid sortedUsingComparator() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); students.sort(Comparator.comparing(Student::getName)); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
多条件排序
我们在静态方法一节中展示了多条件排序,还可以在Comparator
匿名内部类中实现多条件逻辑:
@Testvoid sortedMultiCondition() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12), new Student("Jerry", 13) ); students.sort((s1, s2) -> { if (s1.getName().equals(s2.getName())) { return Integer.compare(s1.getAge(), s2.getAge()); } else { return s1.getName().compareTo(s2.getName()); } }); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
从逻辑来看,多条件排序就是先判断第一级条件,如果相等,再判断第二级条件,依次类推。在 Java8 中可以使用comparing
和一系列thenComparing
表示多级条件判断,上面的逻辑可以简化为:
@Testvoid sortedMultiConditionUsingComparator() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12), new Student("Jerry", 13) ); students.sort(Comparator.comparing(Student::getName).thenComparing(Student::getAge)); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
这里的thenComparing
方法是可以有多个的,用于表示多级条件判断,这也是函数式编程的方便之处。
在Stream中进行排序
Java8 中,不但引入了 Lambda 表达式,还引入了一个全新的流式 API:Stream API,其中也有sorted
方法用于流式计算时排序元素,可以传入Comparator
实现排序逻辑:
@Testvoid streamSorted() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final Comparator comparator = (h2, h3) -> h2.getName().compareTo(h3.getName()); final List sortedStudents = students.stream() .sorted(comparator) .collect(Collectors.toList()); Assertions.assertEquals(sortedStudents.get(0), new Student("Jerry", 12));}
同样的,我们可以通过 Lambda 简化书写:
@Testvoid streamSortedUsingComparator() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final Comparator comparator = Comparator.comparing(Student::getName); final List sortedStudents = students.stream() .sorted(comparator) .collect(Collectors.toList()); Assertions.assertEquals(sortedStudents.get(0), new Student("Jerry", 12));}
倒序排列
调转排序判断
排序就是根据compareTo
方法返回的值判断顺序,如果想要倒序排列,只要将返回值取返即可:
@Testvoid sortedReverseUsingComparator2() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final Comparator comparator = (h2, h3) -> h3.getName().compareTo(h2.getName()); students.sort(comparator); Assertions.assertEquals(students.get(0), new Student("Tom", 10));}
可以看到,正序排列的时候,我们是h2.getName().compareTo(h3.getName())
,这里我们直接倒转过来,使用的是h3.getName().compareTo(h2.getName())
,也就达到了取反的效果。在 Java 的Collections
中定义了一个java.util.Collections.ReverseComparator
内部私有类,就是通过这种方式实现元素反转。
借助Comparator
的reversed
方法倒序
在 Java8 中新增了reversed
方法实现倒序排列,用起来也是很简单:
@Testvoid sortedReverseUsingComparator() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final Comparator comparator = (h2, h3) -> h2.getName().compareTo(h3.getName()); students.sort(comparator.reversed()); Assertions.assertEquals(students.get(0), new Student("Tom", 10));}
在Comparator.comparing中定义排序反转
comparing
方法还有一个重载方法,java.util.Comparator#comparing(java.util.function.Function super T,? extends U>, java.util.Comparator super U>)
,第二个参数就可以传入Comparator.reverseOrder()
,可以实现倒序:
@Testvoid sortedUsingComparatorReverse() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); students.sort(Comparator.comparing(Student::getName, Comparator.reverseOrder())); Assertions.assertEquals(students.get(0), new Student("Jerry", 12));}
在Stream中定义排序反转
在Stream
中的操作与直接列表排序类似,可以反转Comparator
定义,也可以使用Comparator.reverseOrder()
反转。实现如下:
@Testvoid streamReverseSorted() { final Liststudents = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final Comparator comparator = (h2, h3) -> h3.getName().compareTo(h2.getName()); final List sortedStudents = students.stream() .sorted(comparator) .collect(Collectors.toList()); Assertions.assertEquals(sortedStudents.get(0), new Student("Tom", 10));}@Testvoid streamReverseSortedUsingComparator() { final List students = Lists.newArrayList( new Student("Tom", 10), new Student("Jerry", 12) ); final List sortedStudents = students.stream() .sorted(Comparator.comparing(Student::getName, Comparator.reverseOrder())) .collect(Collectors.toList()); Assertions.assertEquals(sortedStudents.get(0), new Student("Tom", 10));}
null 值的判断
前面的例子中都是有值元素排序,能够覆盖大部分场景,但有时候我们还是会碰到元素中存在null
的情况:
列表中的元素是 null
列表中的元素参与排序条件的字段是 null
如果还是使用前面的那些实现,我们会碰到NullPointException
异常,即 NPE,简单演示一下:
@Testvoid sortedNullGotNPE() { final Liststudents = Lists.newArrayList( null, new Student("Snoopy", 12), null ); Assertions.assertThrows(NullPointerException.class, () -> students.sort(Comparator.comparing(Student::getName)));}
所以,我们需要考虑这些场景。
元素是 null 的笨拙实现
最先想到的就是判空:
@Testvoid sortedNullNoNPE() { final Liststudents = Lists.newArrayList( null, new Student("Snoopy", 12), null ); students.sort((s1, s2) -> { if (s1 == null) { return s2 == null ? 0 : 1; } else if (s2 == null) { return -1; } return s1.getName().compareTo(s2.getName()); }); Assertions.assertNotNull(students.get(0)); Assertions.assertNull(students.get(1)); Assertions.assertNull(students.get(2));}
我们可以将判空的逻辑抽取出一个Comparator
,通过组合方式实现:
class NullComparatorimplements Comparator { private final Comparator real; NullComparator(Comparator super T> real) { this.real = (Comparator ) real; } @Override public int compare(T a, T b) { if (a == null) { return (b == null) ? 0 : 1; } else if (b == null) { return -1; } else { return (real == null) ? 0 : real.compare(a, b); } }}
在 Java8 中已经为我们准备了这个实现。
使用Comparator.nullsLast
和Comparator.nullsFirst
使用Comparator.nullsLast
实现null
在结尾:
@Testvoid sortedNullLast() { final Liststudents = Lists.newArrayList( null, new Student("Snoopy", 12), null ); students.sort(Comparator.nullsLast(Comparator.comparing(Student::getName))); Assertions.assertNotNull(students.get(0)); Assertions.assertNull(students.get(1)); Assertions.assertNull(students.get(2));}
使用Comparator.nullsFirst
实现null
在开头:
@Testvoid sortedNullFirst() { final Liststudents = Lists.newArrayList( null, new Student("Snoopy", 12), null ); students.sort(Comparator.nullsFirst(Comparator.comparing(Student::getName))); Assertions.assertNull(students.get(0)); Assertions.assertNull(students.get(1)); Assertions.assertNotNull(students.get(2));}
是不是很简单,接下来我们看下如何实现排序条件的字段是 null 的逻辑。
排序条件的字段是 null
这个就是借助Comparator
的组合了,就像是套娃实现了,需要使用两次Comparator.nullsLast
,这里列出实现:
@Testvoid sortedNullFieldLast() { final Liststudents = Lists.newArrayList( new Student(null, 10), new Student("Snoopy", 12), null ); final Comparator nullsLast = Comparator.nullsLast( Comparator.nullsLast( // 1 Comparator.comparing( Student::getName, Comparator.nullsLast( // 2 Comparator.naturalOrder() // 3 ) ) ) ); students.sort(nullsLast); Assertions.assertEquals(students.get(0), new Student("Snoopy", 12)); Assertions.assertEquals(students.get(1), new Student(null, 10)); Assertions.assertNull(students.get(2));}
代码逻辑如下:
代码 1 是第一层 null-safe 逻辑,用于判断元素是否为 null;
代码 2 是第二层 null-safe 逻辑,用于判断元素的条件字段是否为 null;
代码 3 是条件
Comparator
,这里使用了Comparator.naturalOrder()
,是因为使用了String
排序,也可以写为String::compareTo
。如果是复杂判断,可以定义一个更加复杂的Comparator
,组合模式就是这么好用,一层不够再套一层。
以上是"Java如何使用 Lambda 表达式实现超强的排序功能"这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注行业资讯频道!