Java有線程安全的set嗎？

來源：blog.csdn.net/li_canhui/article/details/91393247

在多線程環(huán)境下，要使用線程安全的集合，比如，ConcurrentHashMap是線程安全的HashMap，CopyOnWriteArrayList是線程安全的ArrayList。那么HashSet對應(yīng)的線程安全集合，是什么呢？java有沒有提供默認(rèn)實(shí)現(xiàn)呢？

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

經(jīng)過多次執(zhí)行，結(jié)果都是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數(shù)據(jù)結(jié)構(gòu)來實(shí)現(xiàn)。在add元素時，采用的是可重入鎖來實(shí)現(xiàn)線程安全。

推薦閱讀
“拍一拍” 能撤回了 ！?。?/a>
5款Chrome插件，第1款絕對良心！
為開發(fā)色情游戲，這家公司赴日尋找AV女優(yōu)拍攝，期望暴力賺錢結(jié)果...
拼多多終于釀成慘劇
華為阿里下班時間曝光：所有的光鮮，都有加班的味道

關(guān)于程序員大白

程序員大白是一群哈工大，東北大學(xué)，西湖大學(xué)和上海交通大學(xué)的碩士博士運(yùn)營維護(hù)的號，大家樂于分享高質(zhì)量文章，喜歡總結(jié)知識，歡迎關(guān)注[程序員大白]，大家一起學(xué)習(xí)進(jìn)步！