CopyOnWriteArrayList源码解读——CopyOnWrite思想的利与弊

最新推荐文章于 2024-11-16 17:49:09 发布

2401_85958899

最新推荐文章于 2024-11-16 17:49:09 发布

阅读量860

点赞数 21

文章标签： java 算法 python

本文链接：https://blog.csdn.net/2401_85958899/article/details/141373512

版权

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)

// numMoved=0 即是在数组的最后加入element

newElements = Arrays.copyOf(elements, len + 1);

else {

newElements = new Object[len + 1];

// elements 从0开始拷贝元素至 newElements（从0开始），length=index

System.arraycopy(elements, 0, newElements, 0, index);

// elements 从index开始拷贝元素至 newElements（从index+1开始），length=numMoved

System.arraycopy(elements, index, newElements, index + 1,

numMoved);

}

// newElements的index位置留给新元素element

newElements[index] = element;

// newElements 赋值给旧数组array

setArray(newElements);

} finally {

lock.unlock();

}

3、set(int index, E element)替换指定位置元素

找到旧元素指定位置的元素与新元素比较，不相等则复制替换。
新旧元素相等，则不替换，但是也要进行setArray赋值操作，为的是确保volatile写语意。

public E set(int index, E element) {

final ReentrantLock lock = this.lock;

lock.lock();

try {

Object[] elements = getArray();

E oldValue = get(elements, index);

// oldValue和element不相等，进行替换

if (oldValue != element) {

int len = elements.length;

Object[] newElements = Arrays.copyOf(elements, len);

newElements[index] = element;

setArray(newElements);

} else {

// Not quite a no-op; ensures volatile write semantics

// 二者相等，不是完全没有操作;确保volatile写语义，volatile write happen before read

setArray(elements);

}

return oldValue;

} finally {

lock.unlock();

}

4、addIfAbsent(E e)添加元素时判断元素是否存在

addIfAbsent(E e)操作是两个操作，先判断数组中是否有新元素，有则返回false不添加，无则继续添加逻辑。添加逻辑是有锁线程安全的，而判断元素是否存在是不安全的，如何保证这两个操作加起来线程安全呢？答案是double check！

在单例模式-懒汉式中double check思想非常典型，先查该单例是否存在，存在直接返回，不存在，加锁，二次判断单例是否存在，存在则返回，不存在则新建赋值。

addIfAbsent(E e)运用了同样的思想：

首先获取原数组快照，判断新增元素是否存在，存在则返回false。（first check）
新增元素不存在，则继续添加逻辑addIfAbsent(e, snapshot)。
addIfAbsent(e, snapshot)方法是加锁的，获取锁后首先判断当前数组和快照是否相等，相等则说明数组没有改动，可以直接进行新增元素的操作。如果有修改，则需要判断当前数组中是否有新增元素，有则返回false，无则新增。（second check）

public boolean addIfAbsent(E e) {

// 获取数组快照

Object[] snapshot = getArray();

return indexOf(e, snapshot, 0, snapshot.length) >= 0 ? false :

addIfAbsent(e, snapshot);

}

private boolean addIfAbsent(E e, Object[] snapshot) {

final ReentrantLock lock = this.lock;

lock.lock();

try {

Object[] current = getArray();

int len = current.length;

// 快照与当前数组比较，double-check，

// 这里1.7和1.8不同，1.8对1.6进行了优化

// 看源码时需要对比不同版本的差异

if (snapshot != current) {

// 数组可能已经被修改，

// Optimize for lost race to another addXXX operation

// remove common=len

// add common=snapshot.length 先检查current中前snapshot.length个，然后检查新增的

int common = Math.min(snapshot.length, len);

for (int i = 0; i < common; i++)

if (current[i] != snapshot[i] && eq(e, current[i]))

return false;

if (indexOf(e, current, common, len) >= 0)

return false;

}

Object[] newElements = Arrays.copyOf(current, len + 1);

newElements[len] = e;

setArray(newElements);

return true;

} finally {

lock.unlock();

}

注意：java1.8之前的addIfAbsent(E e)和java8之后的代码有些许不同，以下摘自java7源码，java7中addIfAbsent(E e)并没有使用快照和double check的思想，而是直接将两个操作加锁，虽然保证了线程安全，但是因为方法一上来就加锁，性能比较。

所以java8中对其进行了优化，加了double check思想，第一次判断元素要是在数组中就不进行添加操作，也就不会加锁；而第二次判断，是先判断快照地址和当前数组地址，地址判断当然比遍历数组性能要高了。

public boolean addIfAbsent(E e) {

final java.util.concurrent.locks.ReentrantLock lock = this.lock;

lock.lock();

try {

// Copy while checking if already present.

// This wins in the most common case where it is not present

Object[] elements = getArray();

int len = elements.length;

Object[] newElements = new Object[len + 1];

for (int i = 0; i < len; ++i) {

if (eq(e, elements[i]))

return false; // exit, throwing away copy

else

newElements[i] = elements[i];

}

newElements[len] = e;

setArray(newElements);

return true;

} finally {

lock.unlock();

}

5、remove(int index)删除指定位置元素

删除中间位置的元素，需要将index后面的元素前移填补空缺，同样使用分两半复制的方式达到删除元素并前移的效果。

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)

// numMoved == 0 即是删除数组的最后一个元素，则不需要移动其他元素。

setArray(Arrays.copyOf(elements, len - 1));

else {

// 新数组 length= len-1

Object[] newElements = new Object[len - 1];

System.arraycopy(elements, 0, newElements, 0, index);

// 丢掉旧数组index位置的元素，达到删除的效果

System.arraycopy(elements, index + 1, newElements, index,

numMoved);

setArray(newElements);

}

return oldValue;

} finally {

lock.unlock();

}

6、remove(Object o)通过元素值删除元素

通过元素值删除元素，需要先查询数组中是否有该元素，无则不做操作，有则继续走删除逻辑，同样使用了double check的思想。

public boolean remove(Object o) {

// 获取旧数组快照

Object[] snapshot = getArray();

// 判断需要删除的元素是否在数组中，不在则直接返回false，是则继续删除操作

int index = indexOf(o, snapshot, 0, snapshot.length);

return (index < 0) ? false : remove(o, snapshot, 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;

// double check

// 快照与当前数组进行比较，不相等说明数组已经被其他线程修改

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;

}

// 上面找了一遍没有找到

// index >= len 说明上面查找的是当前整个数组，需要删除的元素已经被修改

if (index >= len)

return false;

// 当前数组变长了，current index位置的元素依然是需要删除的元素，停止判断

if (current[index] == o)

break findIndex;

// 走到这了数组变长了，index位置的元素也已经被删除，但是不代表其他线程新增的元素没有需要删除的元素，继续判断

index = indexOf(o, current, index, len);

if (index < 0)

return false;

}

// 新数组长度减一

Object[] newElements = new Object[len - 1];

// 从0开始复制旧数组index个元素到新数组的0至index-1位置

System.arraycopy(current, 0, newElements, 0, index);

// 跳过index，从index+1复制旧数组剩下的元素到新数组的index至最后的位置。

System.arraycopy(current, index + 1,

newElements, index,

len - index - 1);

// 新数组赋值给旧数组

setArray(newElements);

return true;

} finally {

lock.unlock();

}

7、COWIterator迭代器没有`fail-fast`检查

例如ArrayList在迭代器遍历想删除元素，只能使用迭代器的删除方法，而使用外部ArrayList删除方法会抛ConcurrentModificationException异常，为什么呢？

因为ArrayList成员变量维护了一个modCount，每次修改操作都会modCount++，而迭代器中也维护了一个expectedModCount，新建迭代器时modCount赋值给expectedModCount，在迭代器遍历时会检查modCount != expectedModCount，不相等则抛出ConcurrentModificationException，在迭代器遍历中调用迭代器外部的修改方法只会更新modCount，不会更新迭代器的expectedModCount，而迭代器内部提供的修改方法既更新了modCount，同时也将最新的modCount值赋值给expectedModCount。