Java ArrayList移除元素相关
@Test
public void testList() {
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
for (String s : list) {
if ("1".equals(s)) {
list.remove(s);
}
}
List<String> list1 = new ArrayList<>();
list1.add("1");
list1.add("2");
list1.add("3");
list1.add("4");
for (String s : list1) {
if ("1".equals(s)) {
list1.remove(s);
}
}
}
执行结果如下:
list的remove方法成功执行,
list1的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];
}
上面list为什么能remove成功呢?其实它只循环了一次,所以成功了。
因为它在remove元素1之后,它的size-1变成1,然后Itr内部的cursor变量由0变成1,此时1=1,循环结束,所以成功了。
list1为什么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 = list1.iterator();
while(ite.hasNext()) {
if("1".equals(ite.next())) {
ite.remove();
}
}
如果存在并发操作,还需要对Iterator进行加锁操作。