Java并发集合类之CopyOnWriteArraySet源码分析

简介

      CopyOnWriteArraySet底层是用CopyOnWriteArrayList存储元素的，并不是使用Map来存储元素的，但是我们知道CopyOnWriteArrayList底层其实是一个数组，它是允许元素重复的，那么用它来实现CopyOnWriteArraySet是怎么保证元素不重复呢？

继承结构

 

源码分析

Set类的源码比较少，其中的方法大多数是调用CopyOnWriteArrayList中的方法实现，这里直接贴源码一行一行分析。

public class CopyOnWriteArraySet<E> extends AbstractSet<E>
        implements java.io.Serializable {
    private static final long serialVersionUID = 5457747651344034263L;
    //使用CopyOnWriteArrayList存储元素
    private final CopyOnWriteArrayList<E> al;
    //构造方法  创建一个空的CopyOnWriteArrayList
    public CopyOnWriteArraySet() {
        al = new CopyOnWriteArrayList<E>();
    }
    
    //将集合c中的元素初始化到CopyOnWriteArraySet中
    public CopyOnWriteArraySet(Collection<? extends E> c) {
        //如果c是CopyOnWriteArraySet类型，说明里面没有重复元素，可以直接调用CopyOnWriteArrayList的构造方法初始化al
        if (c.getClass() == CopyOnWriteArraySet.class) {
            @SuppressWarnings("unchecked")          CopyOnWriteArraySet<E> cc =
                (CopyOnWriteArraySet<E>)c;
            //直接调用CopyOnWriteArrayList的构造方法
            al = new CopyOnWriteArrayList<E>(cc.al);
        }
        else {
            //如果不是CopyOnWriteArraySet类型，说明可能有重复元素
            //调用CopyOnWriteArrayList的addAllAbsent()方法初始化
            //它会把重复元素排除掉
            al = new CopyOnWriteArrayList<E>();
            al.addAllAbsent(c);
        }
    }

    //返回set的长度
    public int size() {
        return al.size();
    }

    //判断set是否为空  调用的方法基本都是CopyOnWriteArrayList中的
    public boolean isEmpty() {
        return al.isEmpty();
    }

    //set中是否包含元素o
    public boolean contains(Object o) {
        return al.contains(o);
    }

    //转换为Object数组
    public Object[] toArray() {
        return al.toArray();
    }
    //转换为某一类型的数组
    public <T> T[] toArray(T[] a) {
        return al.toArray(a);
    }

    //清除set中所有元素
    public void clear() {
        al.clear();
    }

    //删除某个元素
    public boolean remove(Object o) {
        return al.remove(o);
    }

    //添加某个元素
    //它是调用CopyOnWriteArrayList的addIfAbsent()方法
    //它会检测元素不存在时添加
    public boolean add(E e) {
        return al.addIfAbsent(e);
    }

    //是否包含c中所有元素
    public boolean containsAll(Collection<?> c) {
        return al.containsAll(c);
    }

    //并集
    public boolean addAll(Collection<? extends E> c) {
        return al.addAllAbsent(c) > 0;
    }
    
    //单方向差集
    public boolean removeAll(Collection<?> c) {
        return al.removeAll(c);
    }

    //交集
    public boolean retainAll(Collection<?> c) {
        return al.retainAll(c);
    }

    //迭代器
    public Iterator<E> iterator() {
        return al.iterator();
    }

    //equals()方法
    public boolean equals(Object o) {
        //如果两者是同一个对象，返回true
        if (o == this)
            return true;
        //如果o不是Set对象  返回false
        if (!(o instanceof Set))
            return false;
        Set<?> set = (Set<?>)(o);
        Iterator<?> it = set.iterator();
        
        //集合元素的快照
        Object[] elements = al.getArray();
        int len = elements.length;
        // Mark matched elements to avoid re-checking
        boolean[] matched = new boolean[len];
        int k = 0;
        outer: while (it.hasNext()) {
            if (++k > len)
                return false;
            //取值
            Object x = it.next();
            //遍历检查元素是否在集合中
            for (int i = 0; i < len; ++i) {
                if (!matched[i] && eq(x, elements[i])) {
                    matched[i] = true;
                    continue outer;
                }
            }
            //不在 返回false
            return false;
        }
        return k == len;
    }

    public boolean removeIf(Predicate<? super E> filter) {
        return al.removeIf(filter);
    }

    public void forEach(Consumer<? super E> action) {
        al.forEach(action);
    }

    public Spliterator<E> spliterator() {
        return Spliterators.spliterator
            (al.getArray(), Spliterator.IMMUTABLE | Spliterator.DISTINCT);
    }

    //比较两个元素是否相等
    private static boolean eq(Object o1, Object o2) {
        return (o1 == null) ? o2 == null : o1.equals(o2);
    }
}

总结

1. CopyOnWriteArraySet底层是用CopyOnWriteArrayList实现的
2. CopyOnWriteArraySet底层方法大多数是调用CopyOnWriteArrayList中的方法实现
3. CopyOnWriteArraySet是并发安全的，而且实现了读写分离
4. CopyOnWriteArraySet通过调用CopyOnWriteArrayList的addIfAbsent（）方法来保证元素不重复