总阅读量:次

⭐⭐⭐ Spring Boot 项目实战 ⭐⭐⭐ Spring Cloud 项目实战
《Dubbo 实现原理与源码解析 —— 精品合集》 《Netty 实现原理与源码解析 —— 精品合集》
《Spring 实现原理与源码解析 —— 精品合集》 《MyBatis 实现原理与源码解析 —— 精品合集》
《Spring MVC 实现原理与源码解析 —— 精品合集》 《数据库实体设计合集》
《Spring Boot 实现原理与源码解析 —— 精品合集》 《Java 面试题 + Java 学习指南》

摘要: 原创出处 blog.csdn.net/li_canhui/article/details/91393247 「bugcoder321」欢迎转载，保留摘要，谢谢！

🙂🙂🙂关注**微信公众号：【芋道源码】**有福利：

  1. RocketMQ / MyCAT / Sharding-JDBC 所有源码分析文章列表
  2. RocketMQ / MyCAT / Sharding-JDBC 中文注释源码 GitHub 地址
  3. 您对于源码的疑问每条留言都将得到认真回复。甚至不知道如何读源码也可以请教噢。
  4. 新的源码解析文章实时收到通知。每周更新一篇左右。
  5. 认真的源码交流微信群。

在多线程环境下，要使用线程安全的集合，比如，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元素时，采用的是可重入锁来实现线程安全。

文章目录