【JDK源碼分析系列】ConcurrentHashMap 與 HashMap 擴容知識點總結

【JDK源碼分析系列】ConcurrentHashMap 與 HashMap 擴容知識點總結

【1】HashMap 擴容機制

下圖爲 16 擴充爲 32 的 resize 示意圖

HashMap 源碼

//初始化或者雙倍擴容,如果是空的,按照初始容量進行初始化
//擴容是雙倍擴容,要麼還在原來索引位置,要麼在擴容的空間上
final Node<K,V>[] resize() {
    // 把之前的數組變成 oldTab
    Node<K,V>[] oldTab = table;
    //old 的長度
    int oldCap = (oldTab == null) ? 0 : oldTab.length;
    //old 的臨界值
    int oldThr = threshold;
    //初始化new的長度和臨界值
    int newCap, newThr = 0;
    //oldCap > 0也就是說不是首次初始化,因爲hashMap用的是懶加載
    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)
            // 臨界值擴容爲 old 的臨界值2倍
            newThr = oldThr << 1; // double threshold
    }
       /**
        * 如果oldCap<0,但是已經初始化了,像把元素刪除完之後的情況,那麼它的臨界值肯定還存在,
        * 如果是首次初始化,它的臨界值則爲0
        */        
    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);
    }
    // 初始化時容量小於默認值16,此時newThr沒有賦值,計算新的resize上限
    if (newThr == 0) {
        // new的臨界值
        float ft = (float)newCap * loadFactor;
        // 判斷是否new容量是否大於最大值,臨界值是否大於最大值
        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其實是個空值,get有可能是空的
    // 賦予當前的table
    table = newTab;
    // 此處是把old中的元素,遍歷到new中
    if (oldTab != null) {
        for (int j = 0; j < oldCap; ++j) {
            // 臨時變量
            Node<K,V> e;
            // 當前哈希桶的位置值不爲null,也就是數組下標處有值,因爲有值表示可能會發生衝突
            if ((e = oldTab[j]) != null) {
                // 把已經賦值之後的變量置位null,爲了好回收,釋放內存
                oldTab[j] = null;
                //節點只有一個值,直接計算索引位置賦值
                // 如果下標處的節點沒有下一個元素
                if (e.next == null)
                    // 把該變量的值存入newTab中,e.hash & (newCap - 1)並不等於j
                    newTab[e.hash & (newCap - 1)] = e;
                //紅黑樹結構的 rehash 處理
                //該節點爲紅黑樹結構,也就是存在哈希衝突,該哈希桶中有多個元素
                else if (e instanceof TreeNode)
                    //把此樹進行轉移到newTab中
                    ((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
                //鏈表結構的 rehash 處理
                //規避了8版本以下的成環問題
                else { // preserve order
                    // loHead 表示老值,老值的意思是擴容後,該鏈表中計算出索引位置不變的元素
                    // hiHead 表示新值,新值的意思是擴容後,計算出索引位置發生變化的元素
                    // 舉個例子,數組大小是 8 ,在數組索引位置是 1 的地方掛着兩個值,
                    // 兩個值的 hashcode 是9和33。
                    // 當數組發生擴容時,新數組的大小是 16,
                    // 此時 hashcode 是 33 的值計算出來的數組索引位置仍然是 1,我們稱爲老值
                    // hashcode 是 9 的值計算出來的數組索引位置是 9,就發生了變化,我們稱爲新值

                       /**
                        * 此處表示爲鏈表結構,同樣把鏈表轉移到newTab中,
                        * 就是把鏈表遍歷後,把值轉過去,在置位null
                        */
                    Node<K,V> loHead = null, loTail = null;
                    Node<K,V> hiHead = null, hiTail = null;
                    Node<K,V> next;
                    // java 7 是在 while 循環裏面,單個計算好數組索引位置後,
                    // 單個的插入數組中,在多線程情況下,會有成環問題
                    // java 8 是等鏈表整個 while 循環結束後,纔給數組賦值,
                    // 所以多線程情況下,也不會成環
                    do {
                        next = e.next;
                        // 原索引
                        // (e.hash & oldCap) == 0 表示老值鏈表
                        if ((e.hash & oldCap) == 0) {
                            if (loTail == null)
                                loHead = e;
                            else
                                loTail.next = e;
                            loTail = e;
                        }
                        // 原索引+oldCap
                        // (e.hash & oldCap) != 0 表示新值鏈表
                        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;
}
//功能 : 對紅黑樹進行 rehash 操作
final void split(HashMap<K,V> map, Node<K,V>[] tab, int index, int bit) {
    TreeNode<K,V> b = this;
    // Relink into lo and hi lists, preserving order
    //老值紅黑樹頭、尾節點
    TreeNode<K,V> loHead = null, loTail = null;
    //新值紅黑樹頭、尾節點
    TreeNode<K,V> hiHead = null, hiTail = null;
    int lc = 0, hc = 0;
    //由於 TreeNode 節點之間存在雙端鏈表的關係,可以利用鏈表關係進行 rehash
    for (TreeNode<K,V> e = b, next; e != null; e = next) {
        //獲取當前節點的下一個節點
        next = (TreeNode<K,V>)e.next;
        e.next = null;
        //對比當前節點的 hash 值
        //(e.hash & bit) == 0 : 表示 hash 值爲老值
        if ((e.hash & bit) == 0) {
            //(e.prev = loTail) == null : 當前節點是老值紅黑樹的頭節點
            if ((e.prev = loTail) == null)
                loHead = e;
            else
                loTail.next = e;
            loTail = e;
            ++lc;
        }
        else {
            //(e.prev = hiTail) == null : 當前節點是新值紅黑樹的頭節點
            if ((e.prev = hiTail) == null)
                hiHead = e;
            else
                hiTail.next = e;
            hiTail = e;
            ++hc;
        }
    }
    //老值紅黑樹非空
    if (loHead != null) {
        if (lc <= UNTREEIFY_THRESHOLD)
            //紅黑樹轉鏈表
            tab[index] = loHead.untreeify(map);
        else {
            tab[index] = loHead;
            //新值存在則有必要重新構成紅黑樹
            if (hiHead != null) // (else is already treeified)
                loHead.treeify(tab);
        }
    }
    //新值紅黑樹非空
    if (hiHead != null) {
        if (hc <= UNTREEIFY_THRESHOLD)
            //紅黑樹轉鏈表
            tab[index + bit] = hiHead.untreeify(map);
        else {
            tab[index + bit] = hiHead;
            //老值存在有必要重新構成紅黑樹
            if (loHead != null)
                hiHead.treeify(tab);
        }
    }
}

【2】ConcurrentHashMap 擴容機制

【2.1】ConcurrentHashMap 擴容圖解

觸發擴容的時機

觸發擴容的時機 :
(1) 元素個數達到擴容閾值
(2) 調用 putAll 方法,但目前容量不足以存放所有元素時
(3) 某條鏈表長度達到8,但數組長度卻小於64時

CPU 核數與遷移任務hash桶數量分配的關係

 單線程下線程的任務分配與遷移操作

 多線程下線程的任務分配與遷移操作

普通鏈表的遷移處理

lastRun 節點概念

紅黑樹遷移處理

hash 桶遷移中以及遷移後處理存取請求

多線程遷移任務完成後的操作

【2.2】ConcurrentHashMap 擴容源碼

//如果table太小,開始擴容
//如果已經在擴容但還沒結束,幫忙轉移並且check
private final void addCount(long x, int check) {
    CounterCell[] as; long b, s;
    if ((as = counterCells) != null ||
        !U.compareAndSwapLong(this, BASECOUNT, b = baseCount, s = b + x)) {
        CounterCell a; long v; int m;
        boolean uncontended = true;
        if (as == null || (m = as.length - 1) < 0 ||
            (a = as[ThreadLocalRandom.getProbe() & m]) == null ||
            !(uncontended =
                U.compareAndSwapLong(a, CELLVALUE, v = a.value, v + x))) {
            fullAddCount(x, uncontended);
            return;
        }
        if (check <= 1)
            return;
        s = sumCount();
    }
    if (check >= 0) {
        Node<K,V>[] tab, nt; int n, sc;
        //檢查當前集合元素個數 s 是否達到擴容閾值 sizeCtl ,擴容時 sizeCtl 爲負數,依舊成立,
        //同時還得滿足數組非空且數組長度不能大於允許的數組最大長度這兩個條件才能繼續
        //這個 while 循環除了判斷是否達到閾值從而進行擴容操作之外
        //還有一個作用就是當一條線程完成自己的遷移任務後,如果集合還在擴容,則會繼續循環,
        //繼續加入擴容大軍,申請後面的遷移任務
        while (s >= (long)(sc = sizeCtl) && (tab = table) != null &&
                (n = tab.length) < MAXIMUM_CAPACITY) {
            int rs = resizeStamp(n);
            // sc < 0 說明集合正在擴容當中
            if (sc < 0) {
                //判斷擴容是否結束或者併發擴容線程數是否已達最大值,如果是的話直接結束while循環
                if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||
                    sc == rs + MAX_RESIZERS || (nt = nextTable) == null ||
                    transferIndex <= 0)
                    break;
                //後續線程加入擴容大軍時每次加 1
                //擴容還未結束,並且允許擴容線程加入,此時加入擴容大軍中
                if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1))
                    transfer(tab, nt);
            }
            //第一條擴容線程設置的某個特定基數
            //如果集合還未處於擴容狀態中,則進入擴容方法,並首先初始化 nextTab 數組即新數組
            //(rs << RESIZE_STAMP_SHIFT) + 2 爲首個擴容線程所設置的特定值,
            //後面擴容時會根據線程是否爲這個值來確定是否爲最後一個線程
            else if (U.compareAndSwapInt(this, SIZECTL, sc, (rs << RESIZE_STAMP_SHIFT) + 2))
                transfer(tab, null);
            s = sumCount();
        }
    }
}
//擴容狀態下其他線程對集合進行插入、修改、刪除、合併、compute等操作時遇到
//ForwardingNode 節點會調用該幫助擴容方法
final Node<K,V>[] helpTransfer(Node<K,V>[] tab, Node<K,V> f) {
    Node<K,V>[] nextTab; int sc;
    if (tab != null && (f instanceof ForwardingNode) &&
        (nextTab = ((ForwardingNode<K,V>)f).nextTable) != null) {
        int rs = resizeStamp(tab.length);
        //此處的 while 循環是上面 addCount 方法的簡版
        //nextTable 擴容後的數組
        while (nextTab == nextTable && table == tab &&
                (sc = sizeCtl) < 0) {
            //判斷擴容是否結束或者併發擴容線程數是否已達最大值
            //如果是的話直接結束while循環
            if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||
                sc == rs + MAX_RESIZERS || transferIndex <= 0)
                break;
            //後續線程加入擴容大軍時每次加 1
            //擴容還未結束,並且允許擴容線程加入,此時加入擴容大軍中
            if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1)) {
                transfer(tab, nextTab);
                break;
            }
        }
        return nextTab;
    }
    return table;
}
//putAll 批量插入或者插入節點後發現鏈表長度達到 8 個或以上,
//但數組長度爲64以下時觸發的擴容會調用到這個方法
private final void tryPresize(int size) {
    //>>>表示無符號右移,也叫邏輯右移
    //即若該數爲正,則高位補0,而若該數爲負數,則右移後高位同樣補0
    int c = (size >= (MAXIMUM_CAPACITY >>> 1)) ? MAXIMUM_CAPACITY :
        //size + (size >>> 1) + 1 : 爲了確保 tableSizeFor 獲取大於 size 的最小 2^n
        tableSizeFor(size + (size >>> 1) + 1);
    int sc;
    //如果不滿足條件,也就是 sizeCtl < 0
    //說明有其他線程正在擴容當中,這裏也就不需要自己去擴容了,結束該方法
    while ((sc = sizeCtl) >= 0) {
        Node<K,V>[] tab = table; int n;
        //如果數組未初始化則進行初始化
        if (tab == null || (n = tab.length) == 0) {
            n = (sc > c) ? sc : c;
            //初始化時將 sizeCtl 設置爲 -1,保證單線程初始化
            if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
                try {
                    if (table == tab) {
                        @SuppressWarnings("unchecked")
                        Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                        table = nt;
                        sc = n - (n >>> 2);
                    }
                } finally {
                    //初始化完成後 sizeCtl 用於記錄當前集合的負載容量值即觸發集合擴容的閾值
                    sizeCtl = sc;
                }
            }
        }
        //若需要的容量 c 小於 sc(當前集合的負載容量觸發的擴容閾值)
        //或 n(數組的長度) 大於數組的最大可支持容量
        //則無法擴容直接退出
        else if (c <= sc || n >= MAXIMUM_CAPACITY)
            break;
        //插入節點後發現鏈表長度達到8個或以上但數組長度爲64以下時
        //觸發的擴容會進入到下面這個 else if 分支
        else if (tab == table) {
            //計算針對給定的 n 的生成戳的大小
            int rs = resizeStamp(n);
            if (sc < 0) {
                //已經正在進行擴容
                Node<K,V>[] nt;
                //判斷擴容是否結束或者併發擴容線程數是否已達最大值
                //如果是的話直接結束while循環
                if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||
                    sc == rs + MAX_RESIZERS || (nt = nextTable) == null ||
                    transferIndex <= 0)
                    break;
                //後續線程加入擴容大軍時每次加 1
                //擴容還未結束,並且允許擴容線程加入,此時加入擴容大軍中
                if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1))
                    transfer(tab, nt);
            }
            //第一次進行擴容
            else if (U.compareAndSwapInt(this, SIZECTL, sc,
                                            (rs << RESIZE_STAMP_SHIFT) + 2))
                transfer(tab, null);
        }
    }
}
//調用該擴容方法的地方有 :
//java.util.concurrent.ConcurrentHashMap#addCount : 
//向集合中插入新數據後更新容量計數時發現到達擴容閾值而觸發的擴容
//java.util.concurrent.ConcurrentHashMap#helpTransfer : 
//擴容狀態下其他線程對集合進行插入、修改、刪除、合併、compute等操作時遇到 ForwardingNode 節點時觸發的擴容
//java.util.concurrent.ConcurrentHashMap#tryPresize : 
//putAll批量插入或者插入後發現鏈表長度達到8個或以上,但數組長度爲64以下時觸發的擴容
//
// 移動拷貝每個元素到新數組中去
// tab:原數組,nextTab:新數組
private final void transfer(Node<K,V>[] tab, Node<K,V>[] nextTab) {
    // 老數組的長度
    // 下面的代碼用於計算每個transfer中要遷移多少個hash桶,一個transfer任務完成後,可以再次申請
    int n = tab.length, stride;
    //計算每條線程處理的桶個數,每條線程處理的桶數量一樣,如果CPU爲單核,則使用一條線程處理所有桶
    //每條線程至少處理16個桶,如果計算出來的結果少於16,則一條線程處理16個桶
    //
    //NCPU : CPU 的核心數
    //MIN_TRANSFER_STRIDE : 16
    if ((stride = (NCPU > 1) ? (n >>> 3) / NCPU : n) < MIN_TRANSFER_STRIDE)
        stride = MIN_TRANSFER_STRIDE; // subdivide range
    // 如果新數組爲空,初始化,大小爲原數組的兩倍
    if (nextTab == null) {            // initiating
        try {
            @SuppressWarnings("unchecked")
            Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n << 1];
            nextTab = nt;
        } catch (Throwable ex) {      // try to cope with OOME
            sizeCtl = Integer.MAX_VALUE;
            return;
        }
        nextTable = nextTab;
        //將 transferIndex 指向最右邊的桶,也就是數組索引下標最大的位置
        transferIndex = n;
    }
    // 新數組的長度
    int nextn = nextTab.length;
    // 代表轉移節點,如果原數組上是轉移節點,說明該節點剛被擴容
    //新建一個佔位對象,該佔位對象的 hash 值爲 -1 該佔位對象存在時表示集合正在擴容狀態,
    //key、value、next 屬性均爲 null,nextTable 屬性指向擴容後的數組
    //該佔位對象主要有兩個用途 :
    //1、佔位作用,用於標識數組該位置的桶已經遷移完畢,處於擴容中的狀態
    //2、作爲一個轉發的作用,擴容期間如果遇到查詢操作,遇到轉發節點,
    //會把該查詢操作轉發到新的數組上去,不會阻塞查詢操作
    ForwardingNode<K,V> fwd = new ForwardingNode<K,V>(nextTab);
    //該標識用於控制是否繼續處理下一個桶,爲 true 則表示已經處理完當前桶,
    //可以繼續遷移下一個桶的數據
    boolean advance = true;
    //該標識用於控制擴容何時結束,
    //該標識還有一個用途是最後一個擴容線程會負責重新檢查一遍數組查看是否有遺漏的桶
    boolean finishing = false; // to ensure sweep before committing nextTab
    // 無限自旋,i 的值會從原數組的最大值開始,慢慢遞減到 0
    //
    //這個循環用於處理一個 stride 長度的任務,i 後面會被賦值爲該 stride 內最大的下標,
    //而 bound 後面會被賦值爲該 stride 內最小的下標
    //通過循環不斷減小 i 的值,從右往左依次遷移桶上面的數據,
    //直到 i 小於 bound 時結束該次長度爲 stride 的遷移任務
    //結束這次的任務後會通過外層 addCount、helpTransfer、tryPresize 方法的 
    //while 循環達到繼續領取其他任務的效果
    for (int i = 0, bound = 0;;) {
        Node<K,V> f; int fh;
        while (advance) {
            int nextIndex, nextBound;
            // 結束循環的標誌
            if (--i >= bound || finishing)
                advance = false;
            // 已經拷貝完成
            else if ((nextIndex = transferIndex) <= 0) {
                //transferIndex <= 0 說明數組的hash桶已被線程分配完畢,
                //沒有了待分配的hash桶,將 i 設置爲 -1,
                //後面的代碼根據這個數值退出當前線程的擴容操作
                i = -1;
                advance = false;
            }
            //只有首次進入for循環纔會進入這個判斷裏面去,
            //設置 bound 和 i 的值即領取到的遷移任務的數組區間
            else if (U.compareAndSwapInt
                        (this, TRANSFERINDEX, nextIndex,
                        nextBound = (nextIndex > stride ?
                                    nextIndex - stride : 0))) {
                bound = nextBound;
                i = nextIndex - 1;
                advance = false;
            }
        }
        // if 任意條件滿足說明拷貝結束了
        if (i < 0 || i >= n || i + n >= nextn) {
            int sc;
            // 拷貝結束,直接賦值,因爲每次拷貝一個數字,都在原數組上放到轉移節點
            // 原數組發現是轉移節點,是不會操作的,會一直等待轉移節點消失。
            // 也就是說數組節點一旦被標記爲轉移節點,是不會再發生任何變動的,
            //所以不會有任何線程安全的問題
            //
            //擴容結束後做後續工作,將 nextTable 設置爲 null,表示擴容已結束,
            //將 table 指向新數組,sizeCtl 設置爲擴容閾值
            if (finishing) {
                nextTable = null;
                table = nextTab;
                sizeCtl = (n << 1) - (n >>> 1);
                return;
            }
            //線程擴容完畢退出擴容操作時每次減 1
            //每當一條線程擴容結束就會更新一次 sizeCtl 的值,進行減 1 操作
            if (U.compareAndSwapInt(this, SIZECTL, sc = sizeCtl, sc - 1)) {
                //(sc - 2) != resizeStamp(n) << RESIZE_STAMP_SHIFT 成立,
                //說明該線程不是擴容大軍裏面的最後一條線程,直接return回到上層while循環
                if ((sc - 2) != resizeStamp(n) << RESIZE_STAMP_SHIFT)
                    return;
                //(sc - 2) == resizeStamp(n) << RESIZE_STAMP_SHIFT : 
                //說明這條線程是最後一條擴容線程
                //之所以能用這個來判斷是否是最後一條線程,因爲第一條擴容線程進行了如下操作 :
                //U.compareAndSwapInt(this, SIZECTL, sc, (rs << RESIZE_STAMP_SHIFT) + 2)
                //除了修改結束標識之外,還得設置 i = n; 
                //以便重新檢查一遍數組,防止有遺漏未成功遷移的桶
                finishing = advance = true;
                i = n; // recheck before commit
            }
        }
        else if ((f = tabAt(tab, i)) == null)
            //遇到數組上空的位置直接放置一個佔位對象,
            //以便查詢操作的轉發和標識當前處於擴容狀態
            advance = casTabAt(tab, i, null, fwd);
        else if ((fh = f.hash) == MOVED)
            //數組上遇到hash值爲MOVED,也就是 -1 的位置,說明該位置已經被其他線程遷移過了,
            //將 advance 設置爲 true ,以便繼續往下一個桶檢查並進行遷移操作
            advance = true; // already processed
        else {
            synchronized (f) {
                // 進行節點的拷貝
                if (tabAt(tab, i) == f) {
                    Node<K,V> ln, hn;
                    //該節點爲鏈表結構
                    if (fh >= 0) {
                        int runBit = fh & n;
                        Node<K,V> lastRun = f;
                        //遍歷整條鏈表,找出 lastRun 節點
                        for (Node<K,V> p = f.next; p != null; p = p.next) {
                            int b = p.hash & n;
                            if (b != runBit) {
                                runBit = b;
                                lastRun = p;
                            }
                        }
                        //根據 lastRun 節點的高位標識(0 或 1),
                        //首先將 lastRun設置爲 ln 或者 hn 鏈的末尾部分節點,
                        //後續的節點使用頭插法拼接
                        if (runBit == 0) {
                            ln = lastRun;
                            hn = null;
                        }
                        else {
                            hn = lastRun;
                            ln = null;
                        }
                        // 如果節點只有單個數據,直接拷貝,如果是鏈表,循環多次組成鏈表拷貝
                        //使用高位和低位兩條鏈表進行遷移,使用頭插法拼接鏈表
                        for (Node<K,V> p = f; p != lastRun; p = p.next) {
                            int ph = p.hash; K pk = p.key; V pv = p.val;
                            if ((ph & n) == 0)
                                ln = new Node<K,V>(ph, pk, pv, ln);
                            else
                                hn = new Node<K,V>(ph, pk, pv, hn);
                        }
                        // 在新數組位置上放置拷貝的值
                        //setTabAt方法調用的是 Unsafe 類的 putObjectVolatile 方法
                        //使用 volatile 方式的 putObjectVolatile 方法,能夠將數據直接更新回主內存,
                        //並使得其他線程工作內存的對應變量失效,達到各線程數據及時同步的效果
                        //使用 volatile 的方式將 ln 鏈設置到新數組下標爲 i 的位置上
                        setTabAt(nextTab, i, ln);
                        //使用 volatile 的方式將 hn 鏈設置到新數組下標爲 i + n(n爲原數組長度) 的位置上
                        setTabAt(nextTab, i + n, hn);
                        //在老數組位置上放上 ForwardingNode 節點
                        //
                        //遷移完成後使用 volatile 的方式將佔位對象設置到該 hash 桶上,
                        //該佔位對象的用途是標識該hash桶已被處理過,以及查詢請求的轉發作用
                        setTabAt(tab, i, fwd);
                        //advance 設置爲 true 表示當前 hash 桶已處理完,可以繼續處理下一個 hash 桶
                        advance = true;
                    }
                    // 紅黑樹的拷貝
                    //該節點爲紅黑樹結構
                    else if (f instanceof TreeBin) {
                        TreeBin<K,V> t = (TreeBin<K,V>)f;
                        //lo 爲低位鏈表頭結點,loTail 爲低位鏈表尾結點,
                        //hi 和 hiTail 爲高位鏈表頭尾結點
                        TreeNode<K,V> lo = null, loTail = null;
                        TreeNode<K,V> hi = null, hiTail = null;
                        int lc = 0, hc = 0;
                        //同樣也是使用高位和低位兩條鏈表進行遷移
                        //使用for循環以鏈表方式遍歷整棵紅黑樹,使用尾插法拼接 ln 和 hn 鏈表
                        for (Node<K,V> e = t.first; e != null; e = e.next) {
                            int h = e.hash;
                            //這裏面形成的是以 TreeNode 爲節點的鏈表
                            TreeNode<K,V> p = new TreeNode<K,V>
                                (h, e.key, e.val, null, null);
                            //老值紅黑樹
                            if ((h & n) == 0) {
                                if ((p.prev = loTail) == null)
                                    lo = p;
                                else
                                    loTail.next = p;
                                loTail = p;
                                ++lc;
                            }
                            //新值紅黑樹
                            else {
                                if ((p.prev = hiTail) == null)
                                    hi = p;
                                else
                                    hiTail.next = p;
                                hiTail = p;
                                ++hc;
                            }
                        }
                        //形成中間鏈表後會先判斷是否需要轉換爲紅黑樹 :
                        //1、如果符合條件則直接將 TreeNode 鏈表轉爲紅黑樹,再設置到新數組中去
                        //2、如果不符合條件則將 TreeNode 轉換爲普通的 Node 節點,
                        //再將該普通鏈表設置到新數組中去
                        //
                        //(hc != 0) ? new TreeBin<K,V>(lo) : t : 
                        //這行代碼的用意在於,如果原來的紅黑樹沒有被拆分成兩份,
                        //那麼遷移後它依舊是紅黑樹,可以直接使用原來的 TreeBin 對象
                        ln = (lc <= UNTREEIFY_THRESHOLD) ? untreeify(lo) :
                            (hc != 0) ? new TreeBin<K,V>(lo) : t;
                        hn = (hc <= UNTREEIFY_THRESHOLD) ? untreeify(hi) :
                            (lc != 0) ? new TreeBin<K,V>(hi) : t;
                        //setTabAt方法調用的是 Unsafe 類的 putObjectVolatile 方法
                        //使用 volatile 方式的 putObjectVolatile 方法,能夠將數據直接更新回主內存,
                        //並使得其他線程工作內存的對應變量失效,達到各線程數據及時同步的效果
                        //使用 volatile 的方式將 ln 鏈設置到新數組下標爲 i 的位置上
                        setTabAt(nextTab, i, ln);
                        //使用 volatile 的方式將 hn 鏈設置到新數組下標爲 i + n(n爲原數組長度) 的位置上
                        setTabAt(nextTab, i + n, hn);
                        // 在老數組位置上放上 ForwardingNode 節點
                        //遷移完成後使用 volatile 的方式將佔位對象設置到該 hash 桶上,
                        //該佔位對象的用途是標識該hash桶已被處理過,以及查詢請求的轉發作用
                        setTabAt(tab, i, fwd);
                        advance = true;
                    }
                }
            }
        }
    }
}

致謝

本博客爲博主的學習實踐總結,並參考了衆多博主的博文,在此表示感謝,博主若有不足之處,請批評指正。

【1】HashMap JDK1.8實現原理

【2】ConcurrentHashMap1.8 - 擴容詳解

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