HashSet源码解析

HashSet

HashMap实现原理

HashSet 是一个不允许存储重复元素的集合，它的实现比较简单，底层实际上是一个HashMap.使用HashMap的键存储使用者想要存入HashSet中的值,这样就可以保证set不会存储重复的元素了.所以,只要理解了 HashMap，HashSet 就毫无难度了。

成员变量

首先通过源代码了解下 HashSet 的成员变量:

    private transient HashMap<E,Object> map;//使用HashMap存储数据,transient关键字保证其不参与自动序列化
    private static final Object PRESENT = new Object();

主要就上面两个变量:

• map ：用于存放数据的HashMap。
• PRESENT ：在HashSet中，元素都存到内部HashMap键值对的Key中，而Value都有一个统一的值,这个值就是PRESENT

构造函数

    public HashSet() {
        map = new HashMap<>();
    }
    
    public HashSet(int initialCapacity, float loadFactor) {
        map = new HashMap<>(initialCapacity, loadFactor);
    }    
	 public HashSet(Collection<? extends E> c) {
        map = new HashMap<>(Math.max((int) (c.size()/.75f) + 1, 16));
        addAll(c);
    }
	public HashSet(int initialCapacity) {
        map = new HashMap<>(initialCapacity);
    }

构造函数很简单，利用了 HashMap 初始化了 map ,可以指定内部HashMap的初始容量和加载因子,也可以直接传递一个集合对象过来,将集合对象中所有的数据添加到内部HashMap中.

方法

内部的方法有以下几个:

我们选择其中的一部分进行源码分析.

add

    public boolean add(E e) {
        return map.put(e, PRESENT)==null;
    }

比较关键的就是这个 add() 方法。
可以看出它是将存放的对象当做了 HashMap 的键，value 都是相同的 PRESENT 。由于 HashMap 的 key 是不能重复的，所以每当有重复的值写入到 HashSet 时，value 会被覆盖(实质上值都是一致的)，但 key 不会受到影响，这样就保证了 HashSet 中只能存放不重复的元素。

当我们向HashSet对象中放入同个元素时,第一个元素放入成功,add方法返回true,而第二个元素放入失败,add方法会返回false,因为在进行第二次add操作时,map.put(e, PRESENT)方法会返回老值(oldValue),这个值为PRESENT,也就不等于null了.试验代码如下:

        Set<String> set = new HashSet<>();
        System.out.println(set.add("chen"));
        System.out.println(set.add("chen"));

remove和clear

   public boolean remove(Object o) {
        return map.remove(o)==PRESENT;
    }
   public void clear() {
        map.clear();
    }
	

remove和clear方法实际上也是调用了HashMap的对应方法.

writeObject和readObject

这两个方法就是当HashSet进行序列化时调用的方法,重写这两个方法的原因就是HashSet内部的HashMap存在着空位,如果使用默认的序列化方法会导致序列化失败.

总结

HashSet 的原理比较简单，几乎全部借助于 HashMap 来实现的。所以 HashMap 会出现的线程不安全问题 HashSet 依然不能避免。在多线程下使用HashSet可能会使其内部的HashMap对象中的链表出现环状引用的情况,当查找该链表所在位置的某个不存在的键值对时,会导致CPU占用达到百分之百的情况.
顺带提一下,HashMap中可放的数据是要注意以下几点的:

• HashMap 的 key 和 value 都可以是 null

• Map 的 key 和 value 都 不允许 是 基本数据类型

• HashMap 的 key 可以是 任意对象，但如果 对象的 hashCode 改变了，那么将找不到原来该 key 所对应的 value(此处对HashSet无影响)