条件去重
我们知道, Java8 lambda自带的去重为 distinct 方法, 但是只能过滤整体对象, 不能实现对象里的某个值进行判定去重, 比如:
Java
List numbers = Arrays.asList(1, 2, 3, 4, 5, 5, 5, 5, 6, 7);
List distinctNumbers = numbers.stream()
.distinct()
.collect(Collectors.toList());
System.out.println(distinctNumbers);//1, 2, 3, 4, 5, 6, 7
1
2
3
4
5
6
Listnumbers=Arrays.asList(1,2,3,4,5,5,5,5,6,7);
ListdistinctNumbers=numbers.stream()
.distinct()
.collect(Collectors.toList());
System.out.println(distinctNumbers);//1, 2, 3, 4, 5, 6, 7
但是, 如果我们有一个 List 类似这样的对象, 要对 User 的 name 进行条件去重怎么办?
我们想要的效果是这样的:
Java
List distinctUsers = users.stream()
.distinct(User::getName)
.collect(Collectors.toList());
1
2
3
ListdistinctUsers=users.stream()
.distinct(User::getName)
.collect(Collectors.toList());
但是很遗憾, distinct()方法并不能设置条件. 解决方案如下:
首先定义一个过滤器:
Java
public static Predicate distinctByKey(Function super T, Object> keyExtractor) {
Map seen = new ConcurrentHashMap<>();
return object -> seen.putIfAbsent(keyExtractor.apply(object), Boolean.TRUE) == null;
}
1
2
3
4
publicstaticPredicatedistinctByKey(Function<?superT ,Object>keyExtractor){
Mapseen=newConcurrentHashMap<>();
returnobject->seen.putIfAbsent(keyExtractor.apply(object),Boolean.TRUE)==null;
}
然后就可以进行条件去重啦:
Java
List users = new LinkedList<>();
users.add(new User("Jim"));
users.add(new User("Jim"));
users.add(new User("Tom"));
users.add(new User("Leo"));
List distinctUsers = users.stream()
.filter(distinctByKey(User::getName))
.collect(Collectors.toList());
System.out.println(distinctUsers);//[Jim, Tom, Leo]
1
2
3
4
5
6
7
8
9
10
11
Listusers=newLinkedList<>();
users.add(newUser("Jim"));
users.add(newUser("Jim"));
users.add(newUser("Tom"));
users.add(newUser("Leo"));
ListdistinctUsers=users.stream()
.filter(distinctByKey(User::getName))
.collect(Collectors.toList());
System.out.println(distinctUsers);//[Jim, Tom, Leo]
条件分组
还是一样的例子, 我们有一个 List, User 有 name 和 age. 现在我们想把这个 List 按年龄分成三组:0
直接上代码:
Java
List users = new LinkedList<>();
users.add(new User("Jim", 12));
users.add(new User("John", 18));
users.add(new User("Tom", 21));
users.add(new User("Leo", 30));
users.add(new User("Kate", 44));
users.add(new User("Lio", 50));
Map> tripleUsers = users.stream()
.collect(Collectors.groupingBy((Function) user -> {
String key;
if (user.getAge() <= 20) {
key = "less20";
} else if (user.getAge() <= 40) {
key = "less40";
} else {
key = "more40";
}
return key;
}, Collectors.toList()));
System.out.println(tripleUsers);
//{more40=[Kate, Lio], less40=[Tom, Leo], less20=[Jim, John]}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
Listusers=newLinkedList<>();
users.add(newUser("Jim",12));
users.add(newUser("John",18));
users.add(newUser("Tom",21));
users.add(newUser("Leo",30));
users.add(newUser("Kate",44));
users.add(newUser("Lio",50));
Map>tripleUsers=users.stream()
.collect(Collectors.groupingBy((Function)user->{
Stringkey;
if(user.getAge()<=20){
key="less20";
}elseif(user.getAge()<=40){
key="less40";
}else{
key="more40";
}
returnkey;
},Collectors.toList()));
System.out.println(tripleUsers);
//{more40=[Kate, Lio], less40=[Tom, Leo], less20=[Jim, John]}