HashMap put源码分析
HashMap 采用的是 数组 + 链表 + 红黑树的结构
public V put(K key, V value) {
// 根据key计算hash值
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)
// 没有数据,则创建一个新的 Node 节点,初始化其值
n = (tab = resize()).length;
// 判断 tab[i] 即数组当前索引是否有值
if ((p = tab[i = (n - 1) & hash]) == null)
// 若没有,则直接赋值
tab[i] = newNode(hash, key, value, null);
// 若有
else {
Node<K,V> e; K k;
// tab[i] 是一个节点,并且 hash 和 key 值 是否相同
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
// 判断 tab[i] 是否是 树节点(红黑树)
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);
// 当链表长度大于等于 7 时, 转 红黑树结构
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;
}
}
// 修改原值,返回原值
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount;
// 大于 12 负载因子
if (++size > threshold)
// 扩容
resize();
afterNodeInsertion(evict);
return null;
}