HashSet源码解析

本文基于JDK1.8对HaseSet源码分析

1.类结构

HashSet类层级关系结构图

可以看到HashSet 实现了Set接口，那为什么叫HashSet？因为HashSet内部采用HashMap存储数据。

HashSet类的成员变量：

    //使用HashMap存储数据，HashSet存储的数据实际为HashMap的key
    private transient HashMap<E,Object> map;

    //HashMap value的占位值
    private static final Object PRESENT = new Object();

由于HashMap的key允许null值，key不允许重复，所以HashSet可以存储null值，但是元素不允许重复，和HashMap一样，也线程不安全。

2.方法解析

2.1 构造函数

public HashSet() {
        //无参构造方法，初始化HashMap
        map = new HashMap<>();
    }

public HashSet(Collection<? extends E> c) {
        //初始化map，计算map容量，公式为 c.size/0.75+1,如果值小于16，则取值16
        map = new HashMap<>(Math.max((int) (c.size()/.75f) + 1, 16));
        //将集合中所有元素添加进去
        addAll(c);
    }

public HashSet(int initialCapacity, float loadFactor) {
        //手动指定容量和负载因子
        map = new HashMap<>(initialCapacity, loadFactor);
    }

 public HashSet(int initialCapacity) {
        //手动指定容量
        map = new HashMap<>(initialCapacity);
    }

可以看到创建HashSet时，本质就是初始化HashMap。

2.2 add(E e)

add(E e) 添加指定元素

 public boolean add(E e) {
        //往map里添加元素，如果key已经存在则返回false，否则返回true
        return map.put(e, PRESENT)==null;
    }

2.3 remove(Object o)

remove(Object o) 删除指定元素

 public boolean remove(Object o) {
        //本质是通过key删除map中的元素，如果key存在，则返回true，否则返回false
        return map.remove(o)==PRESENT;
    }

2.4 contains(Object o) 

contains(Object o) 判断是否包含指定元素

public boolean contains(Object o) {
        //判断map中是否包含指定的key
        return map.containsKey(o);
    }

2.5 size()

size() 获取元素个数

public int size() {
        //获取map的元素个数
        return map.size();
    }

2.6 isEmpty()

isEmpty() 判断集合是否为空

public boolean isEmpty() {
        //判断map是否为空
        return map.isEmpty();
    }

2.7 iterator()

iterator() 获取迭代器

public Iterator<E> iterator() {
        //获取map key的迭代器
        return map.keySet().iterator();
    }

2.8 clear()

clear() 清空集合

public void clear() {
        //本质是清空map
        map.clear();
    }