博文首发地址:https://blog.virde.xyz

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<String> 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<String> 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<String> 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<String> 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<String> 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不让在循环时删除元素的本质原因时什么?
     */
posted @ 2020-06-12 17:40  黑风风  阅读(414)  评论(1)    收藏  举报

博文首发地址:https://blog.virde.xyz