java集合对象根据多个字段去重

￡神魔★判官ぃ 2021-10-03 03:08 805阅读 0赞

大家很关心的结果来了，之前我们要去重对象中的重复的数据 可能需要两个for循环，从今天开始就不再需要了。  
昨天看了一篇文章说TreeSet能去重多个，我知道能去重一次，后来测试研究如下：

// 这是查看的文章
    List<User> list = new ArrayList<>();
    list.add(new User("李白","123",1));
    list.add(new User("李白1","1234",1));
    list.add(new User("李白1","1234",1));
    list.add(new User("李白","123",1));
    for(User u : list){
        System.out.println("去重前的数据" + u);
    }
    Set<User> set = new TreeSet<>(new Comparator<User>() {
        @Override
        public int compare(User user0, User t1) {
            int count = 1;
            if(StringUtils.equals(user0.getUsername(), t1.getUsername()) &&
                    StringUtils.equals(user0.getPassword(),t1.getPassword())){
                count = 0;
            }
            return count;
        }
    });
    set.addAll(list);
    System.out.println("________________");
    for(User u : set){
        System.out.println("去重后的数据" + u);
    }
    
    打印结果如下：
    去重前的数据User{username='李白', password='123', age=1}
    去重前的数据User{username='李白1', password='1234', age=1}
    去重前的数据User{username='李白1', password='1234', age=1}
    去重前的数据User{username='李白', password='123', age=1}
    ________________
    去重后的数据User{username='李白', password='123', age=1}
    去重后的数据User{username='李白1', password='1234', age=1}

我发现他这方法有点问题，如果我把集合中的值改变,如下：

List<User> list = new ArrayList<>();
    list.add(new User("李白","123",1));
    list.add(new User("李白1","1234",1));
    list.add(new User("李白1","12345",1));
    list.add(new User("李白","123",1));
    for(User u : list){
        System.out.println("去重前的数据" + u);
    }
    Set<User> set = new TreeSet<>(new Comparator<User>() {
        @Override
        public int compare(User user0, User t1) {
            int count = 1;
            if(StringUtils.equals(user0.getUsername(), t1.getUsername()) &&
                    StringUtils.equals(user0.getPassword(),t1.getPassword())){
                count = 0;
            }
            return count;
        }
    });
    set.addAll(list);
    System.out.println("________________");
    for(User u : set){
        System.out.println("去重后的数据" + u);
    }

到此我只改变了list中第三条数据，接下来看一下打印结果:

去重前的数据User{username='李白', password='123', age=1}
    去重前的数据User{username='李白1', password='1234', age=1}
    去重前的数据User{username='李白1', password='12345', age=1}
    去重前的数据User{username='李白', password='123', age=1}
    ________________
    去重后的数据User{username='李白', password='123', age=1}
    去重后的数据User{username='李白1', password='1234', age=1}
    去重后的数据User{username='李白1', password='12345', age=1}
    去重后的数据User{username='李白', password='123', age=1}

这样就不满足我们的需求了

#### 重要方法如下 ####

我把它稍微修改了一下

List<User> list = new ArrayList<>();
    list.add(new User("李白","123",1));
    list.add(new User("李白1","1234",1));
    list.add(new User("李白1","1235",1));
    list.add(new User("李白","123",1));
    for(User u : list){
        System.out.println("去重前的数据" + u);
    }
    
    Set<User> set = new TreeSet<>(new Comparator<User>() {
        @Override
        public int compare(User user0, User t1) {
            int count = 1;//=0表示去重
    		count = user0.getUsername().compareTo(t1.getUsername()) +
    				user0.getPassword().compareTo(t1.getPassword());
            return count;
        }
    });
    set.addAll(list);
    System.out.println("________________");
    for(User u : set){
        System.out.println("去重后的数据" + u);
    }

显示结果如下

去重前的数据User{username='李白', password='123', age=1}
    去重前的数据User{username='李白1', password='1234', age=1}
    去重前的数据User{username='李白1', password='1235', age=1}
    去重前的数据User{username='李白', password='123', age=1}
    ________________
    去重后的数据User{username='李白', password='123', age=1}
    去重后的数据User{username='李白1', password='1234', age=1}
    去重后的数据User{username='李白1', password='1235', age=1}