还在自己写迭代器进行remove?快来看看新方法

我们都知道 List 中是不允许在循环的过程中去进行移除元素的,为什么呢?一般的新人可能会遇到这个问题,比如说会从 List 的遍历的过程中去进行 remove 数据,但是干过几年的开发的有经验的工作人员,是肯定不会这么干的,很简单,会报错。

List 进行 remove

我们可以来看一段代码:

public static void main(String[] args) {
List list= new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");

for (String s: list) {
if (s.equals("1")) {
list.remove(s);
}
}

System.out.println(list);
}

上面这段代码,一般都是初入开发行业的小伙伴可能会这么写,但是当你去运行的时候,就会发现会报一个错误。

Exception in thread "main" java.util.ConcurrentModificationException
at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:909)
at java.util.ArrayList$Itr.next(ArrayList.java:859)
at com.example.fastdfs.Test.main(Test.java:22)

但是当我们把代码改成删除元素 2 的时候,发现又成功了!!!

是真的,成功了,我们看代码和运行结果:

public static void main(String[] args) {
List list= new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");

for (String s: list) {
if (s.equals("2")) {
list.remove(s);
}
}

System.out.println(list);
}

运行结果如下:

为什么会出现这种情况,我删除第一个元素不行,删除第二个元素好用,删除第三个元素又不行了,到底是什么原因导致的呢?

这时候我们就得去看看他的源码编译出来是什么样子的。源码如下:

public static void main(String[] args) {
List list = new ArrayList();
list.add("1");
list.add("2");
list.add("3");
Iterator var2 = list.iterator();

while(var2.hasNext()) {
String s = (String)var2.next();
if (s.equals("3")) {
list.remove(s);
}
}

System.out.println(list);
}

也就是说,foreach 的循环内部,就是采用的iteratior形式,使用的核心方法是hasnext()和next()。

既然都使用迭代器了,为啥还是不行呢?我们来看看迭代器的源码,然后分析一下为啥不行.

其实我们可以从报错都能看出点端倪,报错信息是ArrayList.java:909 

checkForComodification() {
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
}

源代码在执行 remove 方法的时候后,也是调用的 list 当中的remove 方法,源代码中,就是这段:

public boolean remove(Object o) {
if (o == null) {
for (int index = 0; index < size; index++)
if (elementData[index] == null) {
fastRemove(index);
return true;
}
} else {
for (int index = 0; index < size; index++)
if (o.equals(elementData[index])) {
fastRemove(index);
return true;
}
}
return false;
}

中间调用的 fastRemove 方法中,中间就看到了:

    private void fastRemove(int index) {
modCount++;
int numMoved = size - index - 1;
if (numMoved > 0)
System.arraycopy(elementData, index+1, elementData, index,
numMoved);
elementData[--size] = null; // clear to let GC do its work
}

这里我们的modCount++了

而当我们再一次循环的时候,调用的是list内部类itr的next方法,

在我们调用的list的remove的时候,modCount++了,而我们的expectedModCount是等于最开始modCount值.

这时候二者的值不相等的时候,就出现异常了。

归根结底,虽然这个地方使用的是迭代器的遍历,但是remove 的方法可不是迭代器的方法呀。

那么我们使用迭代器遍历然后移除是什么样子的呢?

    public static void main(String[] args) {
List list= new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");

Iterator iterator = list.iterator();
while (iterator.hasNext()){
String next = iterator.next();
if ("3".equals(next)){
iterator.remove();
}
}

System.out.println(list);
}

这么写是不是有点多,那么应该如何快速的写完这段代码呢?

其实一行代码就能很快解决这个事情,我们先来看代码怎么写的:

list.removeIf(vo-> "3".equals(vo));

也不用管返回值了,只要是能满足这个条件的 ,那么就会从集合中给移除掉。

话不多说,看结果:

这么一看,是不是发现非常简单方便,而且还快捷,而且如果要是对于代码量来说的话,那肯定是非常的少的,但凡满足条件的,肯定可以。

removeIf 的进阶玩法

阿粉为什么称之为进阶玩法,实际上也并不是完整的进阶玩法,比如说如果我们有一个功能是这样的,要求做一个导入的功能,然后导入的数据只有一个车牌号是唯一值,之前导入的数据,不做处理,新增的文件中,可能会包含所有的数据,要求数据库中已经存在的数据,不处理,然后导入数据库中不存在的数据。

如果字段少的话,那么实现思路可能会有几种。

第一种:

mybatis 的 SelectKey 标签,判断是否存在,如果存在就不进行新增。

第二种:

导入之前,查询数据库数据,比对数据,然后直接进行remove,最后不存在的数据导入

这两种方法实际上都能实现,但是他们的适用情况就不太一样了,如果字段非常多呢?

自己写sql 的话,那么代价实在是有点大,如果你们使用的还是 Mybatis-plus 的话,那么肯定第一种方式好像就没办法使用了,只能使用第二种了。

那么我们的 removeIf 应该怎么来写呢?

  //创建第一个UserList
List userList = new ArrayList<>();
User user = new User();
user.setId(UUID.randomUUID().toString());
user.setName("张三");
user.setAge(20);
user.setDept("开发部");
userList.add(user);

User user1 = new User();
user1.setId(UUID.randomUUID().toString());
user1.setName("李四");
user1.setAge(22);
user1.setDept("测试部");
userList.add(user1);

User user2 = new User();
user2.setId(UUID.randomUUID().toString());
user2.setName("王五");
user2.setAge(27);
user2.setDept("财务部");
userList.add(user2);

//创建第二个UserList

List userEnd= new ArrayList<>();
User user3= new User();
user3.setId(UUID.randomUUID().toString());
user3.setName("张三");
user3.setAge(20);
user3.setDept("开发部");
userEnd.add(user3);

User user4= new User();
user4.setId(UUID.randomUUID().toString());
user4.setName("李四");
user4.setAge(22);
user4.setDept("测试部");
userEnd.add(user4);

如果我们这时候要把第一个userList 中的数据导入数据库,而 userEnd 则是数据库中的数据,这时候,我们要根据姓名来区分的话,是不是应该之导入王五才对,这时候我们得筛选出王五的数据来,然后做导入,这个时候 removeIf 就派上用场了。

        userList.removeIf(us1-> userEnd.stream().anyMatch(u ->us1.getName().equals(u.getName())));

System.out.println(Arrays.toString(userList.toArray()));

我们最后来看看结果:

[User(id=029b0b0f-ad42-4c15-8341-a3bb401be6d6, name=王五, age=27, dept=财务部)]

是不是已经做到了呢?

你学会了么?

本文名称:还在自己写迭代器进行remove?快来看看新方法
转载注明:http://www.shufengxianlan.com/qtweb/news24/434174.html

网站建设、网络推广公司-创新互联,是专注品牌与效果的网站制作,网络营销seo公司;服务项目有等

广告

声明:本网站发布的内容(图片、视频和文字)以用户投稿、用户转载内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-86922220;邮箱:631063699@qq.com。内容未经允许不得转载,或转载时需注明来源: 创新互联