HashMap源码分析

Java容器

HashMap

摘要

HashMap是Java中使用频率最高的用于映射(键值对)处理的数据类型,在JDK1.8之后对HashMap做了底层优化,例如引入了红黑树的数据结构以及对数据扩容做了优化。

简介

Java为数据结构的映射定义了Java.util.Map接口,此接口主要由四种常用的实现类,包括HashMap,HashTable,LinkedHashMap和TreeMap。它们之间的关系如下图所示:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-wErcbcuL-1580547639681)(en-resource://database/4099:1)]

下面对各个实现类做具体说明:

  1. HashMap:它根据键的hashCode()存储数据,大多数情况下可以直接定位到具体的值,因而具有很快的访问速度,但是遍历顺序确实不确定的。HashMap最多只允许一条记录的键为null,允许多条记录的值为null。HashMap是线程不安全的,如果任意一个时刻多个线程同时写HashMap,可能会导致数据不一致的问题。如果需要满足线程安全,可以用 Collections的synchronizedMap方法使HashMap具有线程安全的能力,或者使用ConcurrentHashMap。

  2. HashTable:HashTable继承自Dictionary抽象类,与HashMap功能类似,不同的是HashTable是线程安全的。HashTable不建议在新代码中使用,当不需要线程安全时,我们可以使用HashMap,当需要线程安全时,我们使用ConcurrentHashMap。

  3. LinkedHashMap:LinkedHashMap继承自HashMap,它保存了记录的插入顺序,在用Iterator遍历LinkedHashMap时,先得到的记录肯定是先插入的,也可以在构造时带参数,按照访问次序排序。

  4. TreeMap:TreeMap实现SortedMap接口,能够把它保存的记录根据键排序,默认是按键值的升序排序,也可以指定排序的比较器,当用Iterator遍历TreeMap时,得到的记录是排过序的。如果使用排序的映射,建议使用TreeMap。在使用TreeMap时,key必须实现Comparable接口或者在构造TreeMap传入自定义的Comparator,否则会在运行时抛出java.lang.ClassCastException类型的异常。

存储结构——字段

HashMap从结构上来看,它的实现主要是数组+链表的形式,在JDK1.8之后加入了红黑树数据结构。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xd4RjnH5-1580547639685)(en-resource://database/4101:1)]
这里需要讲明白两个问题:数据底层具体存储的是什么?这样的存储方式有什么优点呢?

  • 数据底层具体存储:从源码来看,HashMap最重要的一个字段是Node<K,V>[] table,即哈希桶数组。我们从源码来分析这个数据结构。
static class Node<K, V> implements Map.Entry<K,V> {
    final int hash;                  // 用来定位数组索引位置
    final K key;
    V value;
    Node<K,V> next;           // 链表的下一个Node
    
    Node(int hash, K key, V value, Node<K,V> next) {    
        this.hash = hash;    
        this.key = key;    
        this.value = value;    
        this.next = next;
    }
   
    
    Node(int hash, K key, V value, Node<K,V> next) { ... } 
    public final K getKey(){ ... } public final V getValue() { ... }    
    public final String toString() { ... } 
    public final int hashCode() { ... } 
    public final V setValue(V newValue) { ... } 
    public final boolean equals(Object o) { ... }
    
}

Node是HashMap的一个内部类,实现了Map.Entry接口,本质是就是一个映射(键值对)。上图中的每个黑色圆点就是一个Node对象。

  • HashMap就是使用哈希表来进行存储的。哈希表为解决地址冲突,主要采用开放定址法和链地址法,Java中的HashMap主要采用链地址法(数组+链表)。在每个数组元素上都存储着一个链表,当数据被Hash后,得到数组下标,然后把数据放在相应数组下表所对应的链表上。例如程序:
    map.put("tom", "jerry");

系统会首先调用key(“tom”)的hashCode()方法计算得到hashCode值,然后利用Hash算法的后两部(高位运算和取模运算)来定位存储的位置,又是计算的两个Hash值会相同,即发生了碰撞,当然Hash算法计算结果越分散均匀,Hash碰撞的概率就越小,map的存取效率就会越高。

如果哈希桶数组很大,即使较差的Hash算法也会比较分散,如果哈希桶数组数组很小,即使好的Hash算法也会出现较多碰撞,所以就需要在空间成本和时间成本之间权衡,其实就是在根据实际情况确定哈希桶数组的大小,并在此基础上设计好的hash算法减少Hash碰撞。那么通过什么方式来控制map使得Hash碰撞的概率又小,哈希桶数组(Node[] table)占用空间又少呢?答案就是好的Hash算法和扩容机制。

我们首先来理解HashMap的几个基本字段。

    int threshold;                  // 所能容纳的key-value对极限 
    final float loadFactor;    // 负载因子
    int modCount;  
    int size;

Node<K,V>[] table的初始化长度length(默认为16),loadFactor为负载因子(默认是0.75),threshold是HashMap中所能容纳的key-value对的极限个数。threshold=length * loadFactor。当超过这个threshold后就会进行扩容(resize),扩容后的HashMap容量是之前容量的两倍。

在HashMap中,哈希桶数组table的length必须为2的n次方。

不论负载因子和Hash算法设计的多么合理,总会存在HashMap中链表过长造成影响HashMap性能,在JDK1.8之后引入了红黑树,当链表长度大于8时,链表会转化为红黑树,从而提高HashMap的性能。

功能方法——实现

HashMap的内部功能实现很多,本文主要从根据key获取哈希桶数组索引位置、put方法的详细执行、扩容过程三个具有代表性的点深入展开讲解。

1. 确定哈希桶数组索引位置

HashMap定位数组索引位置,直接决定了hash方法的离散性能。先看看源码的实现(方法一+方法二):


方法一: 
static final int hash(Object key) { //jdk1.8 & jdk1.7 
    int h; 
    // h = key.hashCode()  为第一步 取hashCode值 
    // h ^ (h >>> 16)       为第二步 高位参与运算 
    return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);    
 } 

方法二: 
static int indexFor(int h, int length) { //jdk1.7源码,jdk1.8没有这个方法,但实现原理一样
    return h & (length-1); //第三步 取模运算 
}

Hash算法的本质就只有三步:1.调用hashcode()方法得到hashcode值;2. 高位参与运算;3. 取模运算。

对于任意给定对象,只要他的hashcode值相同,那么通过方法一得到的值就会相同,为了使能够在Hash表中分布均匀,我们需要对得到的哈希值进行取模运算,一般的取模运算是通过哈希值对哈希表长度取余,但是由于这样效率不高,我们采用方法二来进行取余运算,效果是一样的,但是效率比较高。

这个方法非常巧妙,它通过h & (table.length -1)来得到该对象的保存位,而HashMap底层数组的长度总是2的n次方,这是HashMap在速度上的优化。当length总是2的n次方时,h& (length-1)运算等价于对length取模,也就是h%length,但是&比%具有更高的效率。

在JDK1.8的实现中,优化了高位运算的算法,通过hashCode()的高16位异或低16位实现的:(h = k.hashCode()) ^ (h >>> 16),主要是从速度、功效、质量来考虑的,这么做可以在数组table的length比较小的时候,也能保证考虑到高低Bit都参与到Hash的计算中,同时不会有太大的开销。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UarZHlVQ-1580547639687)(en-resource://database/4107:1)]

2. 分析HashMap的put方法
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-4T56UTsy-1580547639687)(en-resource://database/4108:1)]

JDK1.8HashMap的put方法源码如下:

public V put(K key, V value) {    
    return putVal(hash(key), key, value, false, true);
}

final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
        Node<K,V>[] tab; Node<K,V> p; int n, i;
        // 如果当前哈希表为空,则进行扩容
        if ((tab = table) == null || (n = tab.length) == 0)
            n = (tab = resize()).length;
        // 计算index,并对null值做处理
        if ((p = tab[i = (n - 1) & hash]) == null)
            tab[i] = newNode(hash, key, value, null);
        else {
            Node<K,V> e; K k;
            // 如果key值存在,则直接进行覆盖
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p;
            // 如果为红黑树,则把当前节点加入到红黑树中
            else if (p instanceof TreeNode)
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            // 添加到链表中去
            else {
                for (int binCount = 0; ; ++binCount) {
                    if ((e = p.next) == null) {
                        p.next = newNode(hash, key, value, null);
                        // 链表长度大于8时,转为红黑树
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            treeifyBin(tab, hash);
                        break;
                    }
                    // key值存在直接进行覆盖
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    p = e;
                }
            }
            if (e != null) { // existing mapping for key
                V oldValue = e.value;
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
        ++modCount;
        // 超过最大容量 就扩容
        if (++size > threshold)
            resize();
        afterNodeInsertion(evict);
        return null;
    }

3. 扩容机制
扩容(resize)就是重新计算容量,向HashMap对象里不停的添加元素,而HashMap对象内部的数组无法装载更多的元素时,对象就需要扩大数组的长度,以便能装入更多的元素。

首先来分析一下resize()的源码:

void resize(int newCapacity) {   //传入新的容量
	Entry[] oldTable = table;    //引用扩容前的Entry数组
   	int oldCapacity = oldTable.length;         
    if (oldCapacity == MAXIMUM_CAPACITY) {  //扩容前的数组大小如果已经达到最大(2^30)了
        threshold = Integer.MAX_VALUE; //修改阈值为int的最大值(2^31-1),这样以后就不会扩容了
        return;
    }
   
    Entry[] newTable = new Entry[newCapacity];  //初始化一个新的Entry数组
    transfer(newTable);                         //!!将数据转移到新的Entry数组里
    table = newTable;                           //HashMap的table属性引用新的Entry数组
    threshold = (int)(newCapacity * loadFactor);//修改阈值
}

以上就是引用一个容量更大的数组来替换之前容量较小的数组,transfer()方法将原有Entry数组的元素拷贝到新的Entry数组里。

void transfer(Entry[] newTable) {
    Entry[] src = table;                   //src引用了旧的Entry数组
    int newCapacity = newTable.length;
    for (int j = 0; j < src.length; j++) { //遍历旧的Entry数组
        Entry<K,V> e = src[j];             //取得旧Entry数组的每个元素
        if (e != null) {
            src[j] = null;//释放旧Entry数组的对象引用(for循环后,旧的Entry数组不再引用任何对象)
            do {
                Entry<K,V> next = e.next;
                int i = indexFor(e.hash, newCapacity); //!!重新计算每个元素在数组中的位置
                e.next = newTable[i]; //标记[1]
                newTable[i] = e;      //将元素放在数组上
                e = next;             //访问下一个Entry链上的元素
            } while (e != null);
        }
    }
}

newTable[i]的引用赋给了e.next,也就是使用了单链表的头插入方式,同一位置上新元素总会被放在链表的头部位置;这样先放在一个索引上的元素终会被放到Entry链的尾部(如果发生了hash冲突的话),这一点和Jdk1.8有区别,下文详解。在旧数组中同一条Entry链上的元素,通过重新计算索引位置后,有可能被放到了新数组的不同位置上。

下面我们讲解下JDK1.8做了哪些优化。经过观测可以发现,我们使用的是2次幂的扩展(指长度扩为原来2倍),所以,元素的位置要么是在原位置,要么是在原位置再移动2次幂的位置。看下图可以明白这句话的意思,n为table的长度,图(a)表示扩容前的key1和key2两种key确定索引位置的示例,图(b)表示扩容后key1和key2两种key确定索引位置的示例,其中hash1是key1对应的哈希与高位运算结果。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-seBfFRl8-1580547639688)(en-resource://database/4112:1)]

元素在重新计算hash之后,因为n变为2倍,那么n-1的mask范围在高位多1bit(红色),因此新的index就会发生这样的变化:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-PYZxfAei-1580547639689)(en-resource://database/4114:1)]

因此,我们在扩充HashMap的时候,不需要像JDK1.7的实现那样重新计算hash,只需要看看原来的hash值新增的那个bit是1还是0就好了,是0的话索引没变,是1的话索引变成“原索引+oldCap”,可以看看下图为16扩充为32的resize示意图:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-855ToRYp-1580547639690)(en-resource://database/4116:1)]

这个设计确实非常的巧妙,既省去了重新计算hash值的时间,而且同时,由于新增的1bit是0还是1可以认为是随机的,因此resize的过程,均匀的把之前的冲突的节点分散到新的bucket了。这一块就是JDK1.8新增的优化点。有一点注意区别,JDK1.7中rehash的时候,旧链表迁移新链表的时候,如果在新表的数组索引位置相同,则链表元素会倒置,但是从上图可以看出,JDK1.8不会倒置。有兴趣的同学可以研究下JDK1.8的resize源码,写的很赞,如下:

final Node<K,V>[] resize() {
        Node<K,V>[] oldTab = table;
        int oldCap = (oldTab == null) ? 0 : oldTab.length;
        int oldThr = threshold;
        int newCap, newThr = 0;
        if (oldCap > 0) {
            // 超过最大值就不再扩充了,就只好随你碰撞去吧
            if (oldCap >= MAXIMUM_CAPACITY) {
                threshold = Integer.MAX_VALUE;
                return oldTab;
            }
            // 没超过最大值,就扩充为原来的2倍
            else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY &&
                     oldCap >= DEFAULT_INITIAL_CAPACITY)
                newThr = oldThr << 1; // double threshold
        }
        else if (oldThr > 0) // initial capacity was placed in threshold
            newCap = oldThr;
        else {               // zero initial threshold signifies using defaults
            newCap = DEFAULT_INITIAL_CAPACITY;
            newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);
        }
        if (newThr == 0) {
            float ft = (float)newCap * loadFactor;
            newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ?
                      (int)ft : Integer.MAX_VALUE);
        }
        threshold = newThr;
        @SuppressWarnings({"rawtypes","unchecked"})
        Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];
        table = newTab;
        if (oldTab != null) {
            for (int j = 0; j < oldCap; ++j) {
                Node<K,V> e;
                if ((e = oldTab[j]) != null) {
                    oldTab[j] = null;
                    if (e.next == null)
                        newTab[e.hash & (newCap - 1)] = e;
                    else if (e instanceof TreeNode)
                        ((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
                    else { // preserve order
                        Node<K,V> loHead = null, loTail = null;
                        Node<K,V> hiHead = null, hiTail = null;
                        Node<K,V> next;
                        do {
                            next = e.next;
                            if ((e.hash & oldCap) == 0) {
                                if (loTail == null)
                                    loHead = e;
                                else
                                    loTail.next = e;
                                loTail = e;
                            }
                            else {
                                if (hiTail == null)
                                    hiHead = e;
                                else
                                    hiTail.next = e;
                                hiTail = e;
                            }
                        } while ((e = next) != null);
                        if (loTail != null) {
                            loTail.next = null;
                            newTab[j] = loHead;
                        }
                        if (hiTail != null) {
                            hiTail.next = null;
                            newTab[j + oldCap] = hiHead;
                        }
                    }
                }
            }
        }
        return newTab;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值