點擊上方「Java基基」，選擇「設為星標」
做積極的人，而不是積極廢人！
每天 14:00 更新文章，每天掉億點點頭髮...
來源：blog.csdn.net/li_canhui/
article/details/91393247
在多線程環境下，要使用線程安全的集合，比如，ConcurrentHashMap是線程安全的HashMap，CopyOnWriteArrayList是線程安全的ArrayList。
那麼HashSet對應的線程安全集合，是什麼呢？java有沒有提供默認實現呢？
在java的concurrent包中，我找到了CopyOnWriteArraySet，那麼它是線程安全的嗎？下面是測試代碼。
public static void main(String[] args) {
Set<String> set = new CopyOnWriteArraySet<>();
ExecutorService service = Executors.newFixedThreadPool(12);
int times = 10000;
AtomicInteger flag = new AtomicInteger(0);
for(int i = 0; i < times; i ++){
service.execute(()->{
set.add("a" + flag.getAndAdd(1));
});
}
service.shutdown();
try {
service.awaitTermination(Long.MAX_VALUE, TimeUnit.DAYS);
}catch (Exception e){
e.printStackTrace();
}
System.out.println(set.size());
}
經過多次執行，結果都是10000。可以說明，CopyOnWriteArraySet是線程安全的Set。
那麼CopyOnWriteArraySet是如何保證寫入時的線程安全呢？以下是CopyOnWriteArraySet的add源碼。
public boolean add(E e) {
return al.addIfAbsent(e);
}
public boolean addIfAbsent(E e) {
Object[] snapshot = getArray();
return indexOf(e, snapshot, 0, snapshot.length) >= 0 ? false :
addIfAbsent(e, snapshot);
}
private static int indexOf(Object o, Object[] elements,
int index, int fence) {
if (o == null) {
for (int i = index; i < fence; i++)
if (elements[i] == null)
return i;
} else {
for (int i = index; i < fence; i++)
if (o.equals(elements[i]))
return i;
}
return -1;
}
private boolean addIfAbsent(E e, Object[] snapshot) {
final ReentrantLock lock = this.lock;
lock.lock();
try {
Object[] current = getArray();
int len = current.length;
if (snapshot != current) {
// Optimize for lost race to another addXXX operation
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();
}
}
從源碼可以看出，CopyOnWriteArraySet底層採用了CopyOnWriteArrayList數據結構來實現。在add元素時，採用的是可重入鎖來實現線程安全。
歡迎加入我的知識星球，一起探討架構，交流源碼。加入方式，長按下方二維碼噢：
已在知識星球更新源碼解析如下：
最近更新《芋道 SpringBoot 2.X 入門》系列，已經 101 餘篇，覆蓋了 MyBatis、Redis、MongoDB、ES、分庫分表、讀寫分離、SpringMVC、Webflux、權限、WebSocket、Dubbo、RabbitMQ、RocketMQ、Kafka、性能測試等等內容。
提供近 3W 行代碼的 SpringBoot 示例，以及超 6W 行代碼的電商微服務項目。
獲取方式：點「在看」，關注公眾號並回復 666 領取，更多內容陸續奉上。
文章有幫助的話，在看，轉發吧。
謝謝支持喲 (*^__^*）