java遍历list时删除元素_JAVA List遍历的四种写法以及遍历时删除元素测试

import org.junit.Test;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.concurrent.CopyOnWriteArrayList;

/**

* 遍历List的集中写法，

* 以及List遍历时删除元素的测试

*/

public class ListForTests {

static List list = new ArrayList<>();

static{

list.add("China");

list.add("America");

list.add("Japan");

list.add("Korean");

list.add("England");

list.add("Russia");

}

@Test

public void testLoopList(){

System.out.println("--- fori 遍历 ---");

for (int i = 0; i < list.size(); i++)

System.out.println(list.get(i));

System.out.println("--- foreach 遍历 ---");

for (String s : list) System.out.println(s);

System.out.println("--- lamda 遍历 ---");

list.forEach(s-> System.out.println(s));

System.out.println("--- Iterator 遍历 ---");

Iterator iter = list.iterator();

while(iter.hasNext()) System.out.println(iter.next());

}

// 测试当遍历时删除元素

@Test

public void testWhenLoopDelete(){

// 使用这种写法删除是可以的，每次循环都去判断size()值

for (int i = 0; i < list.size(); i++) {

if(list.get(i).equals("Japan"))

list.remove(i);

System.out.println(list.get(i));

}

}

/**

* 这个从测试结果可以看到，抛出了ConcurrentModificationException异常

* 从下面的再次轮循发现，成功删除了元素，但是遍历自身报错了

* foreach写法使用的是Iterator，所以Iterator遍历也会报同样的错误

*/

@Test

public void testWhenLoopDelete_Iterator(){

try{

for (String s : list) {

if(s.equals("America"))

list.remove(s);

System.out.println(s);

}

}catch (Exception ex){ ex.printStackTrace();}

System.out.println(" ------------- ");

for (String s : list) System.out.println(s);

System.out.println(" ------------- ");

System.out.println(" 测试Iterator写法 ");

System.out.println(" ------------- ");

try{

Iterator itr = list.iterator();

while (itr.hasNext()){

String ele = itr.next();

System.out.println(ele);

if(ele.equals("Korean")) list.remove(ele);

}

}catch (Exception ex){ex.printStackTrace();}

System.out.println(" ------------- ");

for (String s : list) System.out.println(s);

}

// Lamda的写法也报错了。List会在遍历时检查有没有修改元素个数

@Test

public void testWhenLoopDelete_Lamda(){

list.forEach(str -> {

if(str.equals("England")){

list.remove(str);

}

System.out.println(str);

});

}

/**

* 这种写法也是可以的，

*/

@Test

public void testWhenLoopDelete_CopyOnWriteArrayList(){

CopyOnWriteArrayList cwList = new CopyOnWriteArrayList<>(list);

for (String s : cwList) {

if(s.equals("China")) cwList.remove(s);

System.out.println(s);

}

}

/**

* 从Iterator的源码中可以看到，

* 是因为判断了modCount != expectedModCount所以才抛出异常的。

* 这两个值是怎么来的？

* expectedModCount是调用iterator()时复制的modCount值

* modCount值会随着修改次数不断自增，所以遍历时修改元素个数肯定会modCount != expectedModCount

*

* 有一种情况遍历时修改不会异常，删除倒数第二个元素。

* 因为删除倒数第二个元素，当最后一次调用hasNext()时，会提前结束，漏掉最后一次循环,导致不调用next()方法。

* 所以就不会抛出异常

*

* 下面示例中'England'为倒数第二个元素，此方法执行不会抛出异常

*/

@Test

public void testWhenLoopDelete_IteratorV2() {

try{

Iterator itr = list.iterator();

while (itr.hasNext()){

String ele = itr.next();

System.out.println(ele);

if(ele.equals("England")) {

list.remove(ele);

}

}

}catch (Exception ex){ex.printStackTrace();}

System.out.println(" ------------- ");

for (String s : list) System.out.println(s);

}

}

/**

* 衍生问题：

* 为什么CopyOnWriteArrayList遍历时删除元素不会报错？

* ArrayList不让在循环时删除元素的本质原因时什么？

*/