Java stream流实现List＜对象＞通过对象中的多个字段去重

UnIQUE Eason2023-12-18 9:09

遇到同时通过多个字段对list进行去重需求，记录一下

先创建一个实体类

java 复制代码

@Data
public class User {

    private Long id;

    private String name;

    private String code;
    
    private String phone;
    
    private Integer age;
    
 	public User(Long id, String name, String code, String phone, Integer age) {  
        this.id = id;  
        this.name = name;  
        this.code = code;  
        this.phone = phone;  
        this.age = age;  
    }
    
    public User() {}

}

创建一个List对象，并添加一些重复元素

java 复制代码

List<User> userList = new ArrayList<>();
userList.add(new User(1L, "张三", "1001", "99999999", 21));
userList.add(new User(1L, "张三", "1001", "99999999", 21)); // 重复
userList.add(new User(3L, "张三", "1001", "99999999", 22));
userList.add(new User(4L, "张三", "1001", "99999999", 21));
userList.add(new User(5L, "张四", "1002", "88888888", 21));

普通去重写法

java 复制代码

List<User> list = userList.stream()  
    .distinct() // 使用默认的equals方法进行去重  
    .collect(Collectors.toList());

输出：

java 复制代码

[User(id=1, name=张三, code=1001, phone=99999999, age=21),
 User(id=3, name=张三, code=1001, phone=99999999, age=22),
 User(id=4, name=张三, code=1001, phone=99999999, age=21),
 User(id=5, name=张四, code=1002, phone=88888888, age=21)]

比如现在要通过name、code、phone去重

java 复制代码

List<User> list = userList.stream()
	.collect(Collectors.collectingAndThen(Collectors.toCollection(() ->
                new TreeSet<>
                (Comparator.comparing(User::getName)
                .thenComparing(User::getCode)
                .thenComparing(User::getPhone))),
                 ArrayList::new));

以上是使用Collectors.collectingAndThen收集器将流中的元素收集到一个TreeSet中。这个TreeSet会根据User对象的name、code和phone字段进行排序和去重。最后，再使用ArrayList::new将排序后的TreeSet中的元素收集到一个新的ArrayList中。

输出结果：

java 复制代码

[User(id=1, name=张三, code=1001, phone=99999999, age=21), 
User(id=5, name=张四, code=1002, phone=88888888, age=21)]

可以看到结果都去重成功了