循环增删 ArrayList ,小心有坑
编程过程中常常需要使用到集合,比如:ArrayList,当我们在 for 循环增删的时候,一不小心就会踩坑。
如下代码
List<String> arrayList1 = new ArrayList<String>();
arrayList1.add("1");
arrayList1.add("2");
for (String s : arrayList1) {
if("1".equals(s)){
arrayList1.remove(s);
}
}
List<String> arrayList2 = new ArrayList<String>();
arrayList2.add("2");
arrayList2.add("1");
for (String s : arrayList2) {
if("1".equals(s)){
arrayList2.remove(s);
}
}
但从代码逻辑上来看,似乎并没有什么问题
但是,当程序运行时,结果如下:
arrayList1的remove方法成功执行, arrayList2的remove方法运行抛出ConcurrentModificationException异常。
我们查看源代码来分析异常原因,因为foreach的本质就是使用迭代器Iterator,所有的Collecation集合类都会实现Iterable接口。
找到ArrayList类的iterator()方法
public Iterator<E> iterator() {
return new Itr();
}
迭代器的本质是先调用hasNext()方法判断存不存在下一元素,然后再使用next()方法取下一元素
public boolean hasNext() {
return cursor != size;
}
@SuppressWarnings("unchecked")
public E next() {
checkForComodification();
int i = cursor;
if (i >= size)
throw new NoSuchElementException();
Object\[\] elementData = ArrayList.this.elementData;
if (i >= elementData.length)
throw new ConcurrentModificationException();
cursor = i + 1;
return (E) elementData\[lastRet = i\];
}
上面arraylist1为什么能remove成功呢?其实它只循环了一次,所以成功了。
因为它在remove元素1之后,它的size-1变成1,然后Itr内部的cursor变量由0变成1,此时1=1,循环结束,所以成功了。
arraylist2为什么remove失败呢?因为它在循环第二次的时候,也remove成功了,但是第三次判断next的时候cursor的值为2导致不等于现在的size 1,所以执行了next方法,最重要的来了,之前remove的操作导致ArrayList的modCount值加1,然后Itr类中的expectedModCount保持不变,所以会抛出异常。
final void checkForComodification() {
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
}
同理可得,由于add操作也会导致modCount自增,所以不允许在foreach中删除, 增加,修改ArrayList中的元素。
对此,推荐大家使用迭代器Iterator删除元素。
Iterator<String> ite = arrayList2.iterator();
while(ite.hasNext()) {
if("1".equals(ite.next())) {
ite.remove();
}
}
如果存在并发操作,还需要对Iterator进行加锁操作。
来源:toutiao.com/i6754322606561690116/