《java.util.concurrent 包源码阅读》08 CopyOnWriteArrayList和CopyOnWriteArraySet

CopyOnWriteArrayList和CopyOnWriteArraySet从数据结构类型上来说是类似的，都是用数组实现的保存一组数据的数据结构，区别也简单就是List和set的区别。因此这里就先讨论CopyOnWriteArrayList，然后再说CopyOnWriteArraySet。

这里重点关注的是CopyOnWrite，从字面上很容易理解，每当写操作的时候复制存储数据的数组，把拷贝上修改完再覆盖原先的数组。那么这样的数据结构适用的情况必然是读操作占绝大多数（很少进行写操作），且数据量不大的场合（例如黑名单）。

因此再来看CopyOnWriteArrayList的实现就很好理解了，必然会有一个volatile的数组，一把锁用于写操作（读操作的时候是不要锁的）。

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

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

复制整个数组使用Arrays.copyOf，部分复制数组使用System.arraycopy。下面以add方法为例：

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();
}
}

可以看到如果在最后添加元素，则使用Arrays.copyOf新建并复制整个数组，如果在中间插入，则使用System.arraycopy分两段复制，然后插入元素。setArray方法的作用就是覆盖原来的数组，代码如下：

final void setArray(Object[] a) {
array = a;
}

下面来简单谈谈CopyOnWriteArraySet，其实CopyOnWriteArraySet就是一个CopyOnWriteArrayList的包装类：

private final CopyOnWriteArrayList<E> al;

区别在于因为Set不允许重复元素，因此CopyOnWriteArraySet的add方法调用的是CopyOnWriteArrayList的addIfAbsent方法

public boolean add(E e) {
return al.addIfAbsent(e);
}