java併發資料結構之CopyOnWriteArrayList

DaFanJoy發表於2022-12-05

CopyOnWriteArrayList是一個執行緒安全的List實現,其在對物件進行讀操作時,由於物件沒有發生改變,因此不需要加鎖,反之在物件進行增刪等修改操作時,它會先複製一個物件副本,然後對副本進行修改,最後將修改後的副本物件寫回,從而保證操作的執行緒安全,下面我們看一下具體的程式碼實現。

###建構函式

透過CopyOnWriteArrayList連結串列的構造,可以看出主要是依賴ReentrantLock與陣列實現執行緒安全的連結串列

/** The lock protecting all mutators */
    final transient ReentrantLock lock = new ReentrantLock();

    /** The array, accessed only via getArray/setArray. */
    private transient volatile Object[] array;

     /**
     * Creates an empty list.
     */
    public CopyOnWriteArrayList() {
        setArray(new Object[0]);
    }

寫操作

add實現

add是一個標準的使用ReentrantLock加鎖保證執行緒安全操作的實現

    /**
     * Appends the specified element to the end of this list.
     *
     * @param e element to be appended to this list
     * @return {@code true} (as specified by {@link Collection#add})
     */
    public boolean add(E e) {
        final ReentrantLock lock = this.lock;
        lock.lock();//加鎖
        try {
            Object[] elements = getArray();//獲取自身陣列物件
            int len = elements.length;
            Object[] newElements = Arrays.copyOf(elements, len + 1);//copy一個副本物件
            newElements[len] = e;//賦值
            setArray(newElements);//把物件寫回去
            return true;
        } finally {
            lock.unlock();//釋放鎖
        }
    }

    /**
     * Inserts the specified element at the specified position in this
     * list. Shifts the element currently at that position (if any) and
     * any subsequent elements to the right (adds one to their indices).
     *
     * @throws IndexOutOfBoundsException {@inheritDoc}
     */
    public void add(int index, E element) {
        final ReentrantLock lock = this.lock;
        lock.lock();
        try {
            Object[] elements = getArray();//獲取自身陣列物件
            int len = elements.length;
            if (index > len || index < 0)//判斷是否越界
                throw new IndexOutOfBoundsException("Index: "+index+
                                                    ", Size: "+len);
            Object[] newElements;
            int numMoved = len - index;//計算需要移動的陣列長度
            if (numMoved == 0)
                newElements = Arrays.copyOf(elements, len + 1);
            else {
                newElements = new Object[len + 1];
                System.arraycopy(elements, 0, newElements, 0, index);
                System.arraycopy(elements, index, newElements, index + 1,
                                 numMoved);
            }
            newElements[index] = element;//賦值
            setArray(newElements);//把物件寫回去
        } finally {
            lock.unlock();//釋放鎖
        }
    }

remove實現

在remove的實現中我們可以看到在實際執行操作之前,會對物件的執行緒安全進行再次檢查,另外在執行定位下標操作時基於原有下標進行分段定位的最佳化,一定機率上會降低迴圈複雜度

public E remove(int index) {
        final ReentrantLock lock = this.lock;
        lock.lock();//加鎖
        try {
            Object[] elements = getArray();//獲取自身陣列物件
            int len = elements.length;
            E oldValue = get(elements, index);//根據下標取值
            int numMoved = len - index - 1;//計算需要移動的陣列長度
            if (numMoved == 0)
                setArray(Arrays.copyOf(elements, len - 1));
            else {
                Object[] newElements = new Object[len - 1];//宣告一個新陣列
                System.arraycopy(elements, 0, newElements, 0, index);
                System.arraycopy(elements, index + 1, newElements, index,
                                 numMoved);
                setArray(newElements);
            }
            return oldValue;
        } finally {
            lock.unlock();
        }
    }

    public boolean remove(Object o) {
        Object[] snapshot = getArray();
        int index = indexOf(o, snapshot, 0, snapshot.length);//遍歷陣列定位元素下標
        return (index < 0) ? false : remove(o, snapshot, index);
    }

    /**
     * A version of remove(Object) using the strong hint that given
     * recent snapshot contains o at the given index.
     */
    private boolean remove(Object o, Object[] snapshot, int index) {
        final ReentrantLock lock = this.lock;
        lock.lock();//加鎖
        try {
            Object[] current = getArray();
            int len = current.length;
            //以下這段程式碼保證資料執行緒安全,再次對陣列是否發生改變進行判斷,如果發生改變進行分段輪詢,提高效率
            if (snapshot != current) findIndex: {//這裡判斷陣列是否已經被修改,如果有修改就重新定位下標
                int prefix = Math.min(index, len);//取最小值
                for (int i = 0; i < prefix; i++) {//提高效率先按最小迴圈次數遍歷
                    if (current[i] != snapshot[i] && eq(o, current[i])) {
                        index = i;
                        break findIndex;
                    }
                }
                if (index >= len)//下標超過當前陣列長度返回false
                    return false;
                if (current[index] == o)//下標未改變,直接返回
                    break findIndex;
                index = indexOf(o, current, index, len);//遍歷剩餘部分
                if (index < 0)
                    return false;
            }
            Object[] newElements = new Object[len - 1];//建立一個長度len - 1的陣列,執行復制操作
            System.arraycopy(current, 0, newElements, 0, index);
            System.arraycopy(current, index + 1,
                             newElements, index,
                             len - index - 1);
            setArray(newElements);//覆蓋原陣列
            return true;
        } finally {
            lock.unlock();
        }
    }

讀操作

讀操作非常簡單,無需加鎖

/**
     * {@inheritDoc}
     *
     * @throws IndexOutOfBoundsException {@inheritDoc}
     */
    public E get(int index) {
        return get(getArray(), index);
    }

    @SuppressWarnings("unchecked")
    private E get(Object[] a, int index) {
        return (E) a[index];
    }

      透過對原始碼的分析,可以看到CopyOnWriteArrayList只在需要保證執行緒安全的寫操作上加鎖,核心思想就是減少鎖競爭,從而提高併發時的讀取效能,適用於寫少讀多的應用場景。

      以上就是對CopyOnWriteArrayList內部核心原始碼的基本走讀與解析,其執行緒安全的實現模式很有代表意義,十分值得初學者參考與學習,希望對大家能有所幫助,其中如有不足與不正確的地方還望指正與海涵,十分感謝。

 

關注微信公眾號,檢視更多技術文章。

相關文章