Java集合框架之三:HashMap源码解析
版权声明:本文为博主原创文章,转载请注明出处,欢迎交流学习!
HashMap在我们的工作中应用的非常广泛,在工作面试中也经常会被问到,对于这样一个重要的集合模型我们有必要弄清楚它的使用方法和它底层的实现原理。HashMap是通过key-value键值对的方式来存储数据的,通过put、get方法实现键值对的快速存取,这是HashMap最基本的用法。HashMap底层是通过数组和链表相结合的混合结构来存放数据的。我们通过分析底层源码来详细了解一下HashMap的实现原理。
1、HashMap的初始化
在HashMap实例化时我们要了解两个概念:初始容量和加载因子。HashMap是基于哈希表的Map接口实现,初始容量是哈希表在创建时的容量。加载因子是哈希表在其容量自动增加之前可以达到多满的一种尺度。当哈希表中的条目数超过了加载因子与当前容量的乘积时,则要对该哈希表进行rehash操作(即重建内部数据结构),从而哈希表将具有大约两倍于当前容量的新的容量。
以上是Java API中HashMap的构造方法,其源码如下:
1 static final int DEFAULT_INITIAL_CAPACITY = 16;//默认初始容量16 2 static final int MAXIMUM_CAPACITY = 1 << 30;//定义最大容量 3 static final float DEFAULT_LOAD_FACTOR = 0.75f;//默认负载因子0.75 4 transient Entry[] table; 5 int threshold; //临界值,值为容量与加载因子的乘积 6 final float loadFactor; //加载因子 7 8 public HashMap() { 9 this.loadFactor = DEFAULT_LOAD_FACTOR; 10 threshold = (int)(DEFAULT_INITIAL_CAPACITY * DEFAULT_LOAD_FACTOR); 11 table = new Entry[DEFAULT_INITIAL_CAPACITY]; 12 init(); 13 } 14 15 void init() { 16 }
以上构造方法定义了一个空的HashMap,其默认初始容量为16,默认初始加载因子为0.75,同时声明了一个Entry类型的数组,数组初始长度为16。那么这里出现的Entry对象是如何定义的呢?看一下它的实现代码:
1 static class Entry<K,V> implements Map.Entry<K,V> { 2 final K key; 3 V value; 4 Entry<K,V> next; //指向下一个Entry节点 5 final int hash; //哈希值 6 7 /** 8 * Creates new entry. 9 */ 10 Entry(int h, K k, V v, Entry<K,V> n) { 11 value = v; 12 next = n; 13 key = k; 14 hash = h; 15 } 16 17 public final K getKey() { 18 return key; 19 } 20 21 public final V getValue() { 22 return value; 23 } 24 25 public final V setValue(V newValue) { 26 V oldValue = value; 27 value = newValue; 28 return oldValue; 29 } 30 //重写equals方法,判断两个Entry是否相等,如果两个Entry对象的key和value相等,则返回true,否则返回false 31 public final boolean equals(Object o) { 32 if (!(o instanceof Map.Entry)) 33 return false; 34 Map.Entry e = (Map.Entry)o; 35 Object k1 = getKey(); 36 Object k2 = e.getKey(); 37 if (k1 == k2 || (k1 != null && k1.equals(k2))) { 38 Object v1 = getValue(); 39 Object v2 = e.getValue(); 40 if (v1 == v2 || (v1 != null && v1.equals(v2))) 41 return true; 42 } 43 return false; 44 } 45 //重写hashCode方法,返回key的hashCode值与value的hashCode值异或运算所得的值 46 public final int hashCode() { 47 return (key==null ? 0 : key.hashCode()) ^ 48 (value==null ? 0 : value.hashCode()); 49 } 50 //重写toString方法,返回此Entry对象的“key=value”映射关系 51 public final String toString() { 52 return getKey() + "=" + getValue(); 53 } 54 55 /** 56 * This method is invoked whenever the value in an entry is 57 * overwritten by an invocation of put(k,v) for a key k that's already 58 * in the HashMap. 59 */ 60 void recordAccess(HashMap<K,V> m) { //当向HashMap中添加键值对时,会调用此方法,这里方法体为空,即不做处理 61 } 62 63 /** 64 * This method is invoked whenever the entry is 65 * removed from the table. 66 */ 67 void recordRemoval(HashMap<K,V> m) { //当向HashMap中删除键值对映射关系时,会调用此方法,这里方法体为空,即不做处理 68 } 69 }
Entry类是HashMap的内部类,其实现了Map.Entry接口。Entry类里定义了4个属性:Object类型的key、value(K、V类型可以看成Object类型),Entry类型的next属性(这个next其实就是一个指向下一个Entry对象的引用,形成了一个链表,通过此Entry对象的next属性可以找到其下一个Entry对象)和int型的hash值。HashMap底层维护的就是一个个Entry对象。在Entry类里还重写了equals方法,若两个Entry的key和value都相等,则返回true,否则返回false,同时还重写了hashCode方法。
2、HashMap的底层数据结构
前面提到过HashMap的底层是基于数组和链表来实现的,那么如何决定一个Entry对象是存放在数组中的哪个位置的呢?它是通过计算hash值来决定存储位置的,同时在查找元素的时候同样也是计算出一个值来找到对应的位置,因此它具有相当快的查询速度。HashMap是根据key的hashCode值来计算hash值的,相同的hashCode值计算出来的hash值也是相同的。当存储的对象达到了一定数量,就有可能出现不同对象的key的hashCode值是相同的,因此计算出来的hash值也相同,这样就出现了冲突。哈希冲突的解决方法有很多,比如再哈希法,这种方法是同时构造多个不同的哈希函数,当发生冲突时就换另外的函数重新计算hash值,直到不再产生冲突为止。HashMap是通过单链表来解决哈希冲突的,这种方法也被称为拉链法。如图所示:
在上图中,左边的部分是哈希表(也称为哈希数组),右边是一个单链表,单链表是用来解决哈希冲突的,数组里的每一个元素都是一个单链表的头节点,当不同的key计算出的数组中的存放位置相同时,就将此对象添加到单链表中。
3、数据存储
在HashMap中定义了put方法来向集合中添加数据,数据以键值对的形式存储,put方法的实现如下:
1 public V put(K key, V value) { 2 //如果存入HashMap的key为null,则将该键值对添加到table[0]中 3 if (key == null) 4 return putForNullKey(value); 5 //key不为null,调用hash方法计算key的hashCode值对应的hash值 6 int hash = hash(key.hashCode()); 7 //根据计算出的hash值,结合数组的长度计算出数组中的插入位置i 8 int i = indexFor(hash, table.length); 9 //遍历数组下标为i处的链表,如果链表上存在元素,其hash值与上述计算得到的hash值相等, 10 //并且其key值与新增的键值对的key值相等,那么就以新增键值对的value替换此元素的value值, 11 //并返回此元素原来的value 12 for (Entry<K,V> e = table[i]; e != null; e = e.next) { 13 Object k; 14 if (e.hash == hash && ((k = e.key) == key || key.equals(k))) { 15 V oldValue = e.value; 16 e.value = value; 17 e.recordAccess(this); 18 return oldValue; 19 } 20 } 21 22 modCount++; //操作次数加1 23 //如果链表上不存在满足条件的元素,则将键值对对应生成的Entry对象添加到table[i]处, 24 //并将下标为i处原先的Entry对象链接到新的Entry对象后面 25 addEntry(hash, key, value, i); 26 return null; 27 } 28 29 30 31 private V putForNullKey(V value) { 32 //遍历数组下标为0处的链表,如果链表中存在元素其key为null,则用value覆盖此元素原来的value 33 for (Entry<K,V> e = table[0]; e != null; e = e.next) { 34 if (e.key == null) { 35 V oldValue = e.value; 36 e.value = value; 37 e.recordAccess(this); 38 return oldValue; 39 } 40 } 41 modCount++; //操作数加1 42 //如果链表中不存在满足条件的元素,则将此键值对生成的Entry对象存放到table[0] 43 addEntry(0, null, value, 0);//key为null,计算出的hash值为0 44 return null; 45 } 46 47 48 //计算hash值 49 static int hash(int h) { 50 // This function ensures that hashCodes that differ only by 51 // constant multiples at each bit position have a bounded 52 // number of collisions (approximately 8 at default load factor). 53 h ^= (h >>> 20) ^ (h >>> 12); 54 return h ^ (h >>> 7) ^ (h >>> 4); 55 } 56 57 58 59 //根据hash值和数组长度,计算出在数组中的索引位置 60 static int indexFor(int h, int length) { 61 return h & (length-1); //计算出的值不会超出数组的长度 62 } 63 64 65 66 void addEntry(int hash, K key, V value, int bucketIndex) { 67 //获取table[bucketIndex]处的Entry对象 68 Entry<K,V> e = table[bucketIndex]; 69 //根据key-value生成新的Entry对象,并将新的Entry对象存入table[bucketIndex]处,将其next引用指向原来的对象 70 table[bucketIndex] = new Entry<K,V>(hash, key, value, e); 71 //如果数组容量大于或等于临界值,则进行扩容 72 if (size++ >= threshold) 73 resize(2 * table.length); //容量为原来的2倍 74 }
以上就是put方法的实现原理,我给出了详细的代码注释。上面已经讲到过HashMap底层的数据结构是由数组和单向链表构成的,当我们向HashMap中put一对key-value键值对时,首先判断key是否为null,如果为null,则遍历table[0]处的链表,若此链表上存在key为null的元素,则用value覆盖此元素的value值,如果不存在这样的元素,那么将此键值对生成的Entry对象存放到table[0]中;如果key不为null,首先根据key的hashCode值计算出hash值,根据hash值和数组长度计算出要存放到数组中的位置i,然后遍历table[i]处的链表,如果链表上存在元素其hash值与计算得到的hash值相等并且其key值与新增的key相等,那么就以新增的value覆盖此元素原来的value并返回原来的value值;如果链表上不存在满足上面条件的元素,则将key-value生成的Entry对象存放到table[i]处,并将其next指向此处原来的Entry对象。这样经过多次put操作,就构成了数组加链表的存储结构。
4、数据读取
HashMap的get方法可以根据key返回其对应的value,如果key为null,则返回null。
1 public V get(Object key) { 2 //如果key为null,则循环table[0]处的单链表 3 if (key == null) 4 return getForNullKey(); 5 //key不为null,根据key的hashCode计算出一个hash值 6 int hash = hash(key.hashCode()); 7 //根据hash值和数组长度计算出一个数组下标值,并且遍历此下标处的单链表 8 for (Entry<K,V> e = table[indexFor(hash, table.length)]; 9 e != null; 10 e = e.next) { 11 Object k; 12 //如果Entry对象的hash值跟上面计算得到的hash值相等,并且key也相等,那么就返回此Entry对象value 13 if (e.hash == hash && ((k = e.key) == key || key.equals(k))) 14 return e.value; 15 } 16 //如果单链表上不存在满足上述条件的Entry对象,则表明HashMap不包含该key的映射关系,返回null 17 return null; 18 } 19 20 21 22 private V getForNullKey() { 23 //获取table[0]处的Entry对象,并循环其链接的单链表,如果单链表上存在不为null的对象, 24 //并且其key为null,那么就返回此对象的value 25 for (Entry<K,V> e = table[0]; e != null; e = e.next) { 26 if (e.key == null) 27 return e.value; 28 } 29 //如果单链表上不存在满足条件的对象,则返回null 30 return null; 31 }
了解了put方法的原理,我们就不难理解get的实现原理了,与之类似也是要根据key的hashCode值来计算出一个hash值,然后根据hash值和数组长度计算出一个数组下标值,接着循环遍历此下标处的单链表,寻找满足条件的Entry对象并返回value,此value就是HashMap中该key所映射的value。注意分析当key为null时的情况:如果HashMap中有key为null的映射关系,那么就返回null映射的value,否则就表明HashMap中不存在key为null的映射关系,返回null。同理,当get方法返回的值为null时,并不一定表明该映射不包含该键的映射关系,也可能是该映射将该键显示的映射为null,即put(key, null)。可使用containKey方法来区分这两种情况。
5、移除映射关系
remove方法根据指定的key从HashMap映射中移除相应的映射关系(如果存在),此方法返回一个value。
1 public V remove(Object key) { 2 Entry<K,V> e = removeEntryForKey(key); 3 return (e == null ? null : e.value); 4 } 5 6 7 8 final Entry<K,V> removeEntryForKey(Object key) { 9 //根据key的hashCode计算hash值 10 int hash = (key == null) ? 0 : hash(key.hashCode()); 11 //根据hash值和数组长度计算数组下标值i 12 int i = indexFor(hash, table.length); 13 //获取下标为i处的数组元素 14 Entry<K,V> prev = table[i]; 15 Entry<K,V> e = prev; 16 //遍历数组下标为i处的单链表 17 while (e != null) { 18 Entry<K,V> next = e.next; 19 Object k; 20 //如果此单链表上存在Entry对象e,其hash值与计算出的hash值相等并且其key也跟传入的key"相等",则从单链表上移除e 21 if (e.hash == hash && 22 ((k = e.key) == key || (key != null && key.equals(k)))) { 23 modCount++; 24 size--; //map中的映射数减1 25 //判断满足条件的Entry对象是在数组下标i处还是在数组外面的单链表上 26 if (prev == e) 27 table[i] = next; 28 else 29 prev.next = next; 30 e.recordRemoval(this); 31 return e; 32 } 33 prev = e; 34 e = next; 35 } 36 37 return e; 38 }
从上面的源码可以看出,remove方法的原理是先找出满足条件的Entry对象,然后从单链表上删除该对象,并返回该对象中的value,本质上是对单链表的操作。
6、总结
从以上源码的分析中我们知道了HashMap底层维护的是数组加链表的混合结构,这是HashMap的核心,只要掌握了这一点我们就能很容易弄清楚HashMap中映射关系的各种操作原理,其本质是对数组和链表的操作。要注意的是HashMap不是线程安全的,我们可以使用Collections.synchoronizedMap方法来获得线程安全的HashMap。例如:
Map map = Collections.sychronizedMap(new HashMap());
以上是我个人对HashMap底层原理的一点理解,不妥的地方欢迎指正!