Java8：Lambda表达式增强版Comparator和排序

1、概述
在这篇教程里，我们将要去了解下**JDK 8 **中的Lambda表达式——特别是怎样使用它来编写Comparator和对集合（Collection）进行排序。
首先，让我们先定义一个简单的实体类：

 1 public class Human {
 2     
 3     private String name;
 4     private int age;
 5 
 6     public Human() {      
 7         super();
 8     }
 9   
10     public Human(String name, int age) {
11         super();
12 
13         this.name = name;
14         this.age = age;
15     }
16 
17    // standard getters and setters
18 
19 }

2、不使用Lambda表达式的基本排序
在Java 8之前，对集合进行排序要为Comparator创建一个匿名内部类用来排序：

1 new  Comparator<Human>() {
2     @Override
3     public int compare(Human h1, Human h2) {
4         return h1.getName().compareTo(h2.getName());
5     }
6 }

简单地用它来对Human实体列表进行排序：

 1 @Test
 2 public void givenPreLambda_whenSortingEntitiesByName_thenCorrectlySorted() {
 3     List<Human> humans = Lists.newArrayList(
 4         new Human("Sarah",10),
 5         new Human("Jack",12)
 6     );
 7 
 8     Collections.sort(humans,new Comparator<Human>() {
 9         @Override
10         public int compare(Human h1, Human h2) {
11             return h1.getName().compareTo(h2.getName());
12         }
13     });
14 
15     Assert.assertThat(humans.get(0), equalTo(new Human("Jack",12)));
16 }

3、使用Lambda表达式的基本排序
根据Lambda表达式的介绍，我们现在可以不使用匿名内部类，只使用简单实用的语义就可以得到相同的结果。

1 (Human h1, Human h2) -> h1.getName().compareTo(h2.getName());

类似地，我们现在可以像之前那样来测试它的行为：

 1 @Test
 2 public void whenSortingEntitiesByName_thenCorrectlySorted() {
 3     List<Human> humans = Lists.newArrayList(
 4         new Human("Sarah",10),
 5         new Human("Jack",12));
 6 
 7     humans.sort((Human h1, Human h2) -> h1.getName().compareTo(h2.getName()));
 8 
 9     Assert.assertThat(humans.get(0), equalTo(new Human("Jack",12)));
10 
11 }

注意：我们同样使用新的sort API，这个API在Java 8里被添加到java.util.List ——而不是旧的Collections.sort API。

4、没有类型定义（ Type Definitions）的基本排序
我们通过不指定类型定义来进一步简化表达式 ——编译器自己可以进行类型判断：

1 (h1, h2) -> h1.getName().compareTo(h2.getName())

测试仍然很相似：

 1 @Test
 2 public void givenLambdaShortForm_whenSortingEntitiesByName_thenCorrectlySorted() {
 3     List<Human> humans = Lists.newArrayList(
 4         new Human("Sarah",10),
 5         new Human("Jack",12)
 6     );
 7     
 8     humans.sort((h1, h2) -> h1.getName().compareTo(h2.getName()));
 9 
10     Assert.assertThat(humans.get(0), equalTo(new Human("Jack",12)));
11 }

5、使用静态方法的引用来排序
下面我们将要使用带有静态方法引用的Lambda表达式去进行排序。
首先，我们要定义compareByNameThenAge方法 ——这个方法拥有与Comparator<Human>对象里的compareTo方法完全相同的签名：

1 public static int compareByNameThenAge(Human lhs, Human rhs) {
2     if(lhs.name.equals(rhs.name)) {    
3         return lhs.age - rhs.age;
4     } else {
5         return lhs.name.compareTo(rhs.name);
6     }
7 }

现在，我们要使用这个引用去调用humans.sort方法：

1 humans.sort(Human::compareByNameThenAge);

最终结果是一个使用静态方法作为Comparator的有效的排序集合：

 1 @Test
 2 public void
 3 givenMethodDefinition_whenSortingEntitiesByNameThenAge_thenCorrectlySorted() {
 4     List<Human> humans = Lists.newArrayList(
 5         new Human("Sarah",10),
 6         new Human("Jack",12));
 7 
 8     humans.sort(Human::compareByNameThenAge);
 9 
10     Assert.assertThat(humans.get(0), equalTo(new Human("Jack",12)));
11 }
6、提取Comparator进行排序
我们可以通过使用实例方法的引用和Comparator.comparing方法来避免定义比较逻辑——它会提取和创建一个基于那个函数的Comparable。
我们准备使用getName() getter方法去建造Lambda表达式并通过name对列表进行排序：
 1 Test
 2 public void givenInstanceMethod_whenSortingEntitiesByNameThenAge_thenCorrectlySorted() {    
 3     List<Human> humans = Lists.newArrayList(
 4         new Human("Sarah",10),
 5         new Human("Jack",12));
 6 
 7     Collections.sort(humans, Comparator.comparing(Human::getName));
 8 
 9     Assert.assertThat(humans.get(0), equalTo(new Human("Jack",12)));
10 }

7、反转排序
JDK 8同样提供了一个有用的方法用来反转Comparator（reverse Comparator）——我们可以快速地利用它来反转我们的排序：

 1 @Test
 2 public void
 3 whenSortingEntitiesByNameReversed_thenCorrectlySorted() {
 4 
 5     List<Human> humans = Lists.newArrayList(
 6         new Human("Sarah",10),
 7         new Human("Jack",12));
 8 
 9     Comparator<Human> comparator = (h1, h2) -> h1.getName().compareTo(h2.getName());
10 
11     humans.sort(comparator.reversed());
12     
13     Assert.assertThat(humans.get(0), equalTo(new Human("Sarah",10)));
14 
15 }

8、多条件排序
比较操作的Lambda表达式不一定都是这么简单的——我们同样可以编写更复杂的表达式，比如先根据name后根据age来对实体进行排序：

 1 @Test
 2 
 3 public void whenSortingEntitiesByNameThenAge_thenCorrectlySorted() {
 4 
 5     List<Human> humans = Lists.newArrayList(
 6         new Human("Sarah",12),
 7         new Human("Sarah",10),
 8         new Human("Zack",12));
 9 
10     humans.sort((lhs, rhs) -> {
11         if(lhs.getName().equals(rhs.getName())) {    
12             return lhs.getAge() - rhs.getAge();
13         } else {
14             return lhs.getName().compareTo(rhs.getName());
15         }
16     });
17 
18     Assert.assertThat(humans.get(0), equalTo(new Human("Sarah",10)));
19 
20 }

9、多条件组合排序
同样的比较逻辑——先根据name进行排序其次是age，同样可以通过Comparator新的组合支持来实现。
从JDK 8开始，我们现在可以把多个Comparator链在一起（chain together）去建造更复杂的比较逻辑：

 1 @Test
 2 public void givenComposition_whenSortingEntitiesByNameThenAge_thenCorrectlySorted() {
 3     List<Human> humans = Lists.newArrayList(  
 4         new Human("Sarah",12),
 5         new Human("Sarah",10),
 6         new Human("Zack",12));
 7 
 8       humans.sort(Comparator.comparing(Human::getName)
 9                                               .thenComparing(Human::getAge));
10 
11       Assert.assertThat(humans.get(0), equalTo(new Human("Sarah",10)));
12 }

参考博客：https://www.jianshu.com/p/25cc423377c5

posted @ 2018-12-03 16:19  llawliet0001  阅读(315)  评论(0)  编辑  收藏  举报