JDK 8提供了Lambda表达式进行流式编程,还提供了对List进行去重的简便写法。
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 5, 5, 5, 6, 7);
List<Integer> distinctNumbers = numbers.stream()
.distinct()
.collect(Collectors.toList());
System.out.println(distinctNumbers);//1, 2, 3, 4, 5, 6, 7
可见Lambda表达式确实做到了去重。但是如果是针对某一个对象的属性进行去重呢?
例如:
我们有一个List<User>
类似这样的对象, 要对 User 的 name 进行条件去重怎么办?
List<User> distinctUsers = users.stream()
.distinct(User::getName)
.collect(Collectors.toList());
但是很遗憾的是,distinct并没有提供构造参数进行条件传参,所以可以有灵台的方法,就是要先定义一个过滤器:
private static <T> Predicate<T> distinctByKey(Function<? super T, ?> keyExtractor) {
Set<Object> seen = ConcurrentHashMap.newKeySet();
return t -> seen.add(keyExtractor.apply(t));
}
然后就可以再distinct参数中传递这个过滤器
List<User> users = new LinkedList<>();
users.add(new User("Jim"));
users.add(new User("Jim"));
users.add(new User("Tom"));
users.add(new User("Leo"));
List<User> distinctUsers = users.stream()
.filter(distinctByKey(User::getName))
.collect(Collectors.toList());
System.out.println(distinctUsers);//[Jim, Tom, Leo]