项目场景:
使用List.removeAll()时失效
问题描述
接到一个IR:返回不在数据库里的数据
实体类示例:
public class User {
private int id;
private String name;
public User (int id, String name) {
this.id = id;
this.name = name;
}
}
代码示例:
public static void main(String[] args) {
List<User> allUsers = new ArrayList<>();
List<User> lessUsers = new ArrayList<>();
for (int i = 0; i < 10; i++) {
User user = new User(i, "name is " + i);
allUsers.add(user);
}
for (int i = 0; i < 5; i++) {
User user = new User(i, "name is " + i);
lessUsers.add(user);
}
System.out.println("allUsers.size()------before-------------->" + allUsers.size());
System.out.println("remove result : " + allUsers.removeAll(lessUsers));
System.out.println("allUsers.size()-------after-------------->" + allUsers.size());
}
控制台输出:
allUsers.size()------before-------------->10
remove result : false
allUsers.size()-------after-------------->10
可以发现,此次List.removeAll失效
原因分析:
一切BUG都来自于没看源码,所以我们一起去源码中获取答案吧!
看源码可看出,最后是调用到了equals()方法:
public boolean equals(Object obj) {
return (this == obj);
}
可以看出默认实现是比较对象在内存的地址。
提问:为什么List<String>.removeAll()不会出现删除失效问题呢?
答:因为String类重写了equals()方法,它比较的是字符串是否相同,不是比较的地址!
解决方案:
既然知道是equals方法导致删除数据失效,那么该怎么解决呢?
既然String类重写equals()方法就可以了,那么我们是不是也可以重写equals()来解决这个问题呢?
答案是可以!
public class User {
Integer id;
String name;
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
User user = (User) o;
if (!id.equals(user.id)) {
return false;
}
if (!name.equals(user.name)) {
return false;
}
return true;
}
@Override
public int hashCode() {
int result = id;
result = 31 * result + name.hashCode();
return result;
}
}
再次执行:
allUsers.size()------before-------------->10
remove result : true
allUsers.size()-------after-------------->5
可以发现,可以正常删除数据了!!!
PS:最后一点很重要!覆盖 equals 时,一定要同时覆盖 hashCode
这样做的目的是保证每一个 equals()返回 true 的两个对像,要有两个相同的 hashCode 。