HashMap源碼解析(JDK 1.8)

寫在前面

本人是java初學者,僅以此篇文章作爲自己在HashMap學習過程中的一點私人見解。如有不正確或不合適之處,歡迎大佬在下方評論進行指正。

一、HashMap主要參數及其含義

這部分內容網上博客有很多,但在此處還是列舉一下:

    /**
        HashMap的初始容量,這邊我們看到默認是16,可根據自身需求進行指定,如果是一個合適的值可避 
     免擴容操作帶來的性能損耗
       **/
    static final int DEFAULT_INITIAL_CAPACITY = 1 << 4; // aka 16

    /**
     * HashMap的最大容量 我們看到相當於Integer整形的最大值
     */
    static final int MAXIMUM_CAPACITY = 1 << 30;

    /**
     * 加載因子,我個人更喜歡稱之爲負載因子 Hash桶內元素個數達到容量*負載因子時進行擴容操作
     */
    static final float DEFAULT_LOAD_FACTOR = 0.75f;

    /**
     *HashMap的樹化閾值,jdk1.8後引入紅黑樹來優化之前由於鏈表過長帶來性能損耗的問題
     */
    static final int TREEIFY_THRESHOLD = 8;

    /**
     * HashMap的解樹化閾值,jdk1.8通過鏈表高低位巧妙的解決1.7死鎖問題,同時如果長度小於此值則進 
     行解樹化
     */
    static final int UNTREEIFY_THRESHOLD = 6;

    /**
     * HashMap的最小樹化容量  其實並不一定是鏈表長度達到8進行擴容,容量如果沒有64 會優先進行擴容
        這一點下面會說明
     */
    static final int MIN_TREEIFY_CAPACITY = 64;

二、put方法

我們可以看到,put方法其實調用了putVal方法

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

    

來看一下putVal方法,這裏面比較關鍵,我儘可能一行一行解釋。首先,上面的註解寫明瞭方法的參數,第一個很好理解,對key做hash之後的hash值,第二個和第三個參數分別是key和value,第四個參數如果爲true,則不覆蓋已有的key。這邊類似於redis的setNx命令,如果你用過redis,那麼肯定很好理解;最後一個參數我們這邊不做講解,是提供給子類LinkedHashMap使用的。

/**
     * Implements Map.put and related methods.
     *
     * @param hash hash for key
     * @param key the key
     * @param value the value to put
     * @param onlyIfAbsent if true, don't change existing value
     * @param evict if false, the table is in creation mode.
     * @return previous value, or null if none
     */ 
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
        Node<K,V>[] tab; Node<K,V> p; int n, i;
        //首先看一下table是否爲空,長度是否爲0,如果是的話,對hash表進行擴容操作
        if ((tab = table) == null || (n = tab.length) == 0)
            n = (tab = resize()).length;
        //這裏的n是擴容後的hash表長度,(n-1)& hash相當於對hash進行取餘操作,只不過位運算
        //速度上更快,這裏不得不佩服大師的智慧
        if ((p = tab[i = (n - 1) & hash]) == null)
            //如果數組該位置爲空的話,就新建一個node節點直接放在那個位置
            tab[i] = newNode(hash, key, value, null);
        else {
            //以下走的都是數組對應位置不爲空的邏輯
            Node<K,V> e; K k;
            //這裏的p是上面if獲取的鏈表對應位置的節點,如果和我們新加入的key相等的話,則將該節點
            //保存到 上面聲明的e節點
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p;
            //否則如果p節點是紅黑樹節點的話,就走對應樹節點插入邏輯
            else if (p instanceof TreeNode)
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            //這個else理解一下,既不是key相等,也不是樹節點,那是什麼呢?聰明的你應該能想到
            //該位置已有節點,key不相同,且可能爲長度大於1的鏈表
            else {
                //這邊是個死循環,其實目的是爲了獲得鏈表長度binCount
                for (int binCount = 0; ; ++binCount) {
                    //如果該位置下一個節點爲空,則走此判斷
                    if ((e = p.next) == null) {
                        //將當前插入的節點掛在後面
                        p.next = newNode(hash, key, value, null);
                        //如果binCount大於樹化的閾值,則走樹化邏輯
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            treeifyBin(tab, hash);
                        //跳出循環
                        break;
                    }
                    //如果便利鏈表到某個位置,該節點的key和我們插入的key相等,則跳出循環
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    //把e賦值給p方便便利鏈表,這邊很好理解
                    p = e;
                }
            }
            //這裏對應上面的判斷鏈表中已經找到相同key的情況,否則e==null,會插入新節點到鏈表尾
            //覆蓋該節點的value
            if (e != null) { // existing mapping for key
                V oldValue = e.value;
                //這裏就是上面提到的第三個參數,可以回去看看,當他爲否纔會覆蓋
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
        ++modCount;
        //完事後hash表的長度如果大於擴容閾值,則進行擴容
        if (++size > threshold)
            resize();
        afterNodeInsertion(evict);
        return null;
    }

接着來看一下hashmap是如何擴容的,上resize()方法:老規矩,看一下上面的註釋,我英文水平低下,理解不對請大佬在下面評論區指正:首先初始化或者給數組容量加倍,如果爲空,根據我們設定的初始容量值分配空間;下面半句是關於擴容後元素移動的,我翻譯不來,讀者自行翻譯

 /**
     * Initializes or doubles table size.  If null, allocates in
     * accord with initial capacity target held in field threshold.
     * Otherwise, because we are using power-of-two expansion, the
     * elements from each bin must either stay at same index, or move
     * with a power of two offset in the new table.
     *
     * @return the table
     */
    final Node<K,V>[] resize() {
        //保存原始hash表
        Node<K,V>[] oldTab = table;
        //保存原始表的原始容量,如果原始表爲空則爲0 否則是原始表的長度
        int oldCap = (oldTab == null) ? 0 : oldTab.length;
        //保存老擴容閾值
        int oldThr = threshold;
        int newCap, newThr = 0;
        //如果原始容量大於0,走這個邏輯
        if (oldCap > 0) {
            //如果原始容量大於2的30次方
            if (oldCap >= MAXIMUM_CAPACITY) {
                //把閾值設置爲整形最大值,意思是不再進行擴容
                threshold = Integer.MAX_VALUE;
                return oldTab;
            }
            //否則如果原始容量擴大兩倍後仍然小於2的30次方並且原始容量大於等於16,新容量等於老容            
             //量乘以2
            else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY &&
                     oldCap >= DEFAULT_INITIAL_CAPACITY)
                //新的閾值也變爲原來的兩倍,這個很好理解
                newThr = oldThr << 1; // double threshold
        }
        //如果老的擴容閾值閾值大於0,這個判斷一般是hashmap初始化的時候第一次擴容會走
        else if (oldThr > 0) // initial capacity was placed in threshold
            //把新的容量設置爲原始的閾值
            newCap = oldThr;
        //這裏看註釋很好理解,如果我們hashmap初始化把容量設置爲0走這個判斷
        else {
               // zero initial threshold signifies using defaults
            //把新的容量設置爲16
            newCap = DEFAULT_INITIAL_CAPACITY;
            //新的閾值設置爲16*0.75,等於多少自己算
            newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);
        }
        //猜一下這個判斷什麼時候會走,很明顯上面如果走else分支 newThr絕對不可能等於0,所以只可能
        //走上面兩個判斷分支,很顯然第一個if判斷分支內部兩個小分支第一個直接返回第二個給newThr賦            
        //了值,所以只能是走中間分支,也就是第一次初始化的時候表容量爲0,但是擴容閾值不是的情況
        if (newThr == 0) {
            //計算新的擴容閾值,如果新的容量小於2的30次方並且新的容量乘以負載因子之後小於2的30次        
            //方,則新的閾值等於新的容量乘以負載因子,否則設置爲整數的最大值
            float ft = (float)newCap * loadFactor;
            newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ?
                      (int)ft : Integer.MAX_VALUE);
        }
        //將當前擴容閾值設置爲新的閾值
        threshold = newThr;
        @SuppressWarnings({"rawtypes","unchecked"})
        //用新的表容量初始化一個新的table
        Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];
        //把當前的hash表設置爲新的table
        table = newTab;
        //下面是擴容之後的rehash操作
        if (oldTab != null) {
            for (int j = 0; j < oldCap; ++j) {
                Node<K,V> e;
                //遍歷數組節點,如果當前節點不等於null,則進行下面的遷移操作
                if ((e = oldTab[j]) != null) {
                    //先把節點保存到e,然後把數組該位置設置爲null
                    oldTab[j] = null;
                    //如果e的下一個節點等於null,也就是當前位置只有一個節點,鏈表長度爲1,
                    //那麼進行和新的表容量進行取餘運算後放到newTable的對應節點
                    if (e.next == null)
                        newTab[e.hash & (newCap - 1)] = e;
                    //如果此節點是樹節點,對它進行樹的拆分,我對紅黑樹瞭解不多,這裏不做過多介紹
                    else if (e instanceof TreeNode)
                        ((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
                    //這裏走的是節點是個長度大於1的鏈表,但是又沒到樹化閾值的情況
                    else { // preserve order
                        Node<K,V> loHead = null, loTail = null;
                        Node<K,V> hiHead = null, hiTail = null;
                        Node<K,V> next;
                        //循環
                        do {
                            //獲取e的下一個節點,保存到next
                            next = e.next;
                            //如果e的hash值與上oldCap等於0,這裏解釋一下,oldCap是一個2的
                            //冪次方,轉化成二進制只有1位等於1,其他位都是0,所以結果要麼爲
                            //0,要麼爲oldCap
                            if ((e.hash & oldCap) == 0) {
                                //等於0,代表低位,如果低位鏈表尾節點等於null,將e賦值給低位
                                //鏈表頭節點
                                if (loTail == null)
                                    loHead = e;
                                //否則將e賦值給尾節點的下一個節點,再將e節點設置爲尾節點,這                
                                //是一個標準的尾插法
                                else
                                    loTail.next = e;
                                loTail = e;
                            }
                            //如果結果不等於0
                            else {
                                //邏輯和上面一樣,只是放到高位鏈表
                                if (hiTail == null)
                                    hiHead = e;
                                else
                                    hiTail.next = e;
                                hiTail = e;
                            }
                        } while ((e = next) != null);
                        //如果低位尾節點不等於null,將其next設置爲null,並且將低位頭節點
                        //放置到newTable的j位置,也就是說位置是一樣的
                        if (loTail != null) {
                            loTail.next = null;
                            newTab[j] = loHead;
                        }
                        //如果是高位,則放置到j加上oldCap偏移量的位置,比如如果原來容量是16,        
                        //新容量是32,原來j=8,在第九個位置,現在放到8+16=24,的位置
                        if (hiTail != null) {
                            hiTail.next = null;
                            newTab[j + oldCap] = hiHead;
                        }
                    }
                }
            }
        }
        return newTab;
    }

關於擴容上面已經寫的較爲詳細了,讀者如果有不明白的地方歡迎評論或者私信我。

我接下來要講的是一個比較容易忽略的點,還記得上面分析putVal的樹化操作嗎?來分析一下代碼:注意上面的註釋,把所有鏈表節點都換成樹節點除非表太小,將使用擴容代替。所以網上很多博客都是裝逼的,根本沒提到這個點,所以當你你出去面試,面試官問:樹化的閾值是8,那鏈表長度到8一定會樹化嗎?你看別人的博客,一定會說:對啊,達到8就會樹化,我看過put的代碼。然而很遺憾,你要回去等消息了,因爲你根本沒點進來這個方法,關於樹化的條件,我在下面代碼中已經註釋,請自行閱讀。

 /**
     * Replaces all linked nodes in bin at index for given hash unless
     * table is too small, in which case resizes instead.
     */
    final void treeifyBin(Node<K,V>[] tab, int hash) {
        int n, index; Node<K,V> e;
        //首先第一件事,什麼也不幹,先判斷tab是否等於null以及tab的長度是否小於64,如果是的話
        //會擴容而不是樹化
        if (tab == null || (n = tab.length) < MIN_TREEIFY_CAPACITY)
            resize();
        else if ((e = tab[index = (n - 1) & hash]) != null) {
            TreeNode<K,V> hd = null, tl = null;
            do {
                TreeNode<K,V> p = replacementTreeNode(e, null);
                if (tl == null)
                    hd = p;
                else {
                    p.prev = tl;
                    tl.next = p;
                }
                tl = p;
            } while ((e = e.next) != null);
            if ((tab[index] = hd) != null)
                hd.treeify(tab);
        }
    }
/**
     * The smallest table capacity for which bins may be treeified.
     * (Otherwise the table is resized if too many nodes in a bin.)
     * Should be at least 4 * TREEIFY_THRESHOLD to avoid conflicts
     * between resizing and treeification thresholds.
     */
    static final int MIN_TREEIFY_CAPACITY = 64;

三、get方法

可以看到他真實調用的是getNode方法

 public V get(Object key) {
        Node<K,V> e;
        return (e = getNode(hash(key), key)) == null ? null : e.value;
    }

 關於第一個判斷,一個是 == ,一個是equals,我們都知道一個是比較地址,一個是內容。所以我們重寫equals的時候建議重寫hashcode,畢竟如果作爲key的話就會有點尷尬,不重寫顯然是獲取不到的,因爲他首先比較的就是hashcode。

final Node<K,V> getNode(int hash, Object key) {
        Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
        //如果hash表不等於null且長度大於0,且和key的hash取餘之後的節點位置不爲null,走裏面邏        
          //輯,否則直接返回null,說明map裏沒有
        if ((tab = table) != null && (n = tab.length) > 0 &&
            (first = tab[(n - 1) & hash]) != null) {
            //上面已經把hash取餘後的位置的節點賦值給first節點,首先檢查第一個節點的hash值是不是
            //等於傳進來的key的hash值並且,first節點的key等於傳進來的key或者傳進來的key不等於     
             //null並且內容相等,則直接返回第一個節點
            if (first.hash == hash && // always check first node
                ((k = first.key) == key || (key != null && key.equals(k))))
                return first;
            //走到這裏說明不是第一個節點,判斷他的下一個節點是否爲null
            if ((e = first.next) != null) {
             // 如果是樹化的節點,則走紅黑樹的遍歷邏輯   
                if (first instanceof TreeNode)
                    return ((TreeNode<K,V>)first).getTreeNode(hash, key);
                do {
                    //否則一直循環到鏈表尾,如果找到相同的key則返回該節點
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        return e;
                } while ((e = e.next) != null);
            }
        }
        //這裏有兩種情況,第一種是表爲空或者表長度爲0,直接返回null,另一種是鏈表遍歷完沒有找到
        //對應的key,則返回null
        return null;
    }

關於hashmap暫時先寫到這裏,後期完善他的其他部分。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章