一、需求描述
在实际业务开发过程中,有时会遇到一个集合包含很多重复数据,需要根据指定的属性对集合进行筛选。我们可以使用Java 8 中的流式操作根据类的某个特定属性对list集合进行去重。
二、代码描述
以下面的Person类为例子:
import lombok.Data;
@Data
public class Person {
private String id;
private String sex;
private String age;
private String name;
}
筛选测试代码如下:
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
public class PersonTest {
public static void main(String args[]){
List<Person> list = new ArrayList<>();
Person p1 = new Person("1","小明1","1","1");
Person p2 = new Person("2","小明2","1","1");
Person p3 = new Person("3","小明3","1","1");
Person p4 = new Person("3","小明4","1","1");
list.add(p1);
list.add(p2);
list.add(p3);
list.add(p4);
System.out.println("未筛选前======");
for(Person p: list){
System.out.println(p.getId()+":"+p.getName());
}
List<Person> distinctList = list.stream()
.collect(Collectors.collectingAndThen(
Collectors.toCollection(() -> new ArrayList<>(list.size())),
l -> l.stream().collect(
Collectors.toMap(Person::getId, vo -> vo, (a, b) -> a))
.values().stream()
.collect(Collectors.toList())
));
System.out.println("筛选后======");
for(Person p: distinctList){
System.out.println(p.getId()+":"+p.getName());
}
}
}
三、代码解析
下面是关键代码的解析和原理说明:
list.stream()
将原始的List转换为一个流(Stream)对象。collectingAndThen()
方法接收两个参数:第一个参数是一个Collector,用于收集流中的元素;第二个参数是一个Function,用于对收集的结果进行后续的处理。Collectors.toCollection(() -> new ArrayList<>(list.size()))
创建一个新的ArrayList,用于收集流中的元素。l.stream().collect(Collectors.toMap(Person::getId, vo -> vo, (a, b) -> a))
将流中的元素收集到一个Map中。这里使用Person
对象的id
作为键,对象本身作为值。如果出现重复的键,则选择保留第一个出现的对象。values().stream().collect(Collectors.toList())
将Map中的值转换为一个List,这样就得到了根据id去重后的List。- 最后,将去重后的List作为结果返回。
总结起来,这段代码的原理是通过流的操作将原始的List转换为一个Map,根据指定的键进行去重,然后将去重后的值转换为一个新的List返回。