這是在ConcurrentHashMap里存儲數(shù)據(jù)的過程,
final V putVal(K key, V value, boolean onlyIfAbsent) {
// 不允許key和value為空
if (key == null || value == null) throw new NullPointerException();
// 計(jì)算Hash
int hash = spread(key.hashCode());
int binCount = 0;
// 進(jìn)入無限循環(huán),直到插入為止
for (Node<K,V>[] tab = table;;) {
Node<K,V> f; int n, i, fh;
if (tab == null || (n = tab.length) == 0)
// 初始化tab
tab = initTable();
else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
if (casTabAt(tab, i, null,
new Node<K,V>(hash, key, value, null)))
// 頭節(jié)點(diǎn)為空 直接插入 不用鎖
break; // no lock when adding to empty bin
}
else if ((fh = f.hash) == MOVED)
// 正在擴(kuò)容 協(xié)助操作
tab = helpTransfer(tab, f);
else {
// 鎖住頭節(jié)點(diǎn)
V oldVal = null;
synchronized (f) {
if (tabAt(tab, i) == f) {
if (fh >= 0) {
binCount = 1;
for (Node<K,V> e = f;; ++binCount) {
K ek;
if (e.hash == hash &&
((ek = e.key) == key ||
(ek != null && key.equals(ek)))) {
oldVal = e.val;
if (!onlyIfAbsent)
e.val = value;
break;
}
Node<K,V> pred = e;
if ((e = e.next) == null) {
pred.next = new Node<K,V>(hash, key,
value, null);
break;
}
}
}
else if (f instanceof TreeBin) {
Node<K,V> p;
binCount = 2;
if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,
value)) != null) {
oldVal = p.val;
if (!onlyIfAbsent)
p.val = value;
}
}
else if (f instanceof ReservationNode)
throw new IllegalStateException("Recursive update");
}
}
if (binCount != 0) {
if (binCount >= TREEIFY_THRESHOLD)
treeifyBin(tab, i);
if (oldVal != null)
return oldVal;
break;
}
}
}
addCount(1L, binCount);
return null;
}
PUT方法
再這之前,先簡單說一下PUT的具體操作:
①先傳入一個(gè)k和v的鍵值對,不可為空(HashMap是可以為空的),如果為空就直接報(bào)錯(cuò)。
②接著去判斷table是否為空,如果為空就進(jìn)入初始化階段。
③如果判斷數(shù)組中某個(gè)指定的桶是空的,那就直接把鍵值對插入到這個(gè)桶中作為頭節(jié)點(diǎn),而且這個(gè)操作不用加鎖。
④如果這個(gè)要插入的桶中的hash值為-1,也就是MOVED狀態(tài)(也就是這個(gè)節(jié)點(diǎn)是forwordingNode),那就是說明有線程正在進(jìn)行擴(kuò)容操作,那么當(dāng)前線程就進(jìn)入?yún)f(xié)助擴(kuò)容階段。
⑤需要把數(shù)據(jù)插入到鏈表或者樹中,如果這個(gè)節(jié)點(diǎn)是一個(gè)鏈表節(jié)點(diǎn),那么就遍歷這個(gè)鏈表,如果發(fā)現(xiàn)有相同的key值就更新value值,如果遍歷完了都沒有發(fā)現(xiàn)相同的key值,就需要在鏈表的尾部插入該數(shù)據(jù)。插入結(jié)束之后判斷該鏈表節(jié)點(diǎn)個(gè)數(shù)是否大于8,如果大于就需要把鏈表轉(zhuǎn)化為紅黑樹存儲。
⑥如果這個(gè)節(jié)點(diǎn)是一個(gè)紅黑樹節(jié)點(diǎn),那就需要按照樹的插入規(guī)則進(jìn)行插入。
⑦put結(jié)束之后,需要給map已存儲的數(shù)量+1,在addCount方法中判斷是否需要擴(kuò)容