transient Node<K,V>[] table;
// 表示若是某條鏈表的節點數量大於等於這個值的時候,則將其轉化爲樹形結構。 static final int TREEIFY_THRESHOLD = 8;
// 若是樹的節點小於等於閾值的時候就開始轉換成鏈表 static final int UNTREEIFY_THRESHOLD = 6;
// 因爲有這個限制,會使第一個值在知足這個條件時纔會生效,具體看後面解釋 static final int MIN_TREEIFY_CAPACITY = 64;
static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> { TreeNode<K,V> parent; // red-black tree links . 能夠理解爲紅黑樹 TreeNode<K,V> left; // 左節點 TreeNode<K,V> right; // 右節點 TreeNode<K,V> prev; // needed to unlink next upon deletion boolean red; // 區分是否爲紅節點 TreeNode(int hash, K key, V val, Node<K,V> next) { super(hash, key, val, next); } ... }
public V put(K key, V value) { // 內部作事情的仍是putVal函數 return putVal(hash(key), key, value, false, true); }
static final int hash(Object key) { int h; return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16); }
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爲空,則經過擴容來建立,後面在看擴容函數 if ((tab = table) == null || (n = tab.length) == 0) n = (tab = resize()).length; // 根據key的hash值 與 數組長度進行取模來獲得數組索引 if ((p = tab[i = (n - 1) & hash]) == null) // 空鏈表,建立節點 tab[i] = newNode(hash, key, value, null); else { Node<K,V> e; K k; // 不爲空,則判斷是否與當前節點同樣,同樣就進行覆蓋 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); // 檢測是否該從鏈表變成樹(注意:這裏是先插入節點,沒有增長binCount,因此判斷條件是大於等於閾值-1) if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st // 知足則樹形化 treeifyBin(tab, hash); break; } if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) break; p = e; } } // 存在相同的key,則替換value if (e != null) { // existing mapping for key V oldValue = e.value; if (!onlyIfAbsent || oldValue == null) e.value = value; // 注意這裏,這裏是供子類LinkedHashMap實現 afterNodeAccess(e); return oldValue; } } ++modCount; // 注意細節:先加入節點,再加長度與閾值進行判斷,是否須要擴容。 if (++size > threshold) resize(); // 注意這裏,這裏是供子類LinkedHashMap實現 afterNodeInsertion(evict); return null; }
final Node<K,V>[] resize() { // copy 數組、容量、閾值 Node<K,V>[] oldTab = table; int oldCap = (oldTab == null) ? 0 : oldTab.length; int oldThr = threshold; int newCap, newThr = 0; // 判斷舊容量是否大於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 // 若是舊容量小於等於0 and 舊閾值大於0, 則將舊閾值賦給新容量 newCap = oldThr; else { // zero initial threshold signifies using defaults // 不然都使用默認值 newCap = DEFAULT_INITIAL_CAPACITY; newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY); } if (newThr == 0) { // 若是新的閾值是 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; // 遍歷每一個位置,將元素賦值給e if ((e = oldTab[j]) != null) { // 置空原來元素,方便GC回收 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 // 高效之處 // 利用哈希值的高低位去區分存儲位置,若是高位是0,則存儲在原來的位置;若是是1則存儲在原來位置+oldCap。 // 低位鏈表的頭結點、尾節點 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; } // 將高位鏈表存放在 原索引+oldCap if (hiTail != null) { hiTail.next = null; newTab[j + oldCap] = hiHead; } } } } } return newTab; }
final void treeifyBin(Node<K,V>[] tab, int hash) { int n, index; Node<K,V> e; // 會判斷數組長度是否大於最小樹化容量,若是不大於先進行擴容減小衝突。 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); } }