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

网友投稿 786 2023-04-22

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

还在自己写迭代器进行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);}

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

但是当我们把代码改成删除元素 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=财务部)]

是不是已经做到了呢?

你学会了么?

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:让领导懂数据库是数据库国产化替代成功的一半
下一篇:搞懂MySQL聚集索引与非聚集索
相关文章