HashMap的put 实现是怎样的?
HashMap 的 put 方法实现了向哈希表中添加键值对的功能。以下是 HashMap 的 put 方法的主要实现步骤:
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
// 第四个参数 onlyIfAbsent 如果是 true,那么只有在不存在该 key 时才会进行 put 操作
// 第五个参数 evict 我们这里不关心
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
Node<K,V>[] tab; Node<K,V> p; int n, i;
// 如果哈希表为空或长度为0,进行初始化或扩容
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
// 根据哈希值计算索引
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)
// 如果当前节点是树节点,调用树节点的 putTreeVal 方法
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);
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) { // 存在相同键的映射
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
// 如果不仅仅是插入或旧值为null,更新节点的值
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount; // 更新修改计数器
if (++size > threshold)
// 如果元素数量超过阈值,进行扩容
resize();
afterNodeInsertion(evict);
return null; // 返回 null 表示没有旧值
}
这段代码展示了 HashMap 的 put 方法的实现,包括计算哈希值、定位桶、处理哈希冲突、插入节点、扩容等关键步骤。该方法确保了键值对能够正确插入到哈希表中,并在需要时进行扩容以维护性能。