直接上代码
注: 代码来自于 Java 9
put方法
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
- 当调用put(),首先会根据key生成一个 hash值,原理如下:
static final int hash(Object key) {
int h;
//key 是 null 直接返回 0
//key 不是null,先计算key对应的hashCode,赋值给 h
//并将 h 与 h >>> 16 做异或(相同为0 不同为1)运算 ,作为hash值返回
return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
}
下图举例说明了位运算的过程,至于原理解释,参考本文引用
- 拿到了hash值后,调用 putVal(),做了如下操作
- 将对象table赋值给tab,并以tab是否为空作为是否第一次调用此方法的判断,是则resize()并给tab,n赋值;
- 获取tab的第i个元素:根据 (n - 1) & hash 算法 ,计算出i找到,如果为空,调用newNode() ,赋值给tab第i个;
- 如果不为空,可能存在2种情况:hash值重复了,也就是put过程中,发现之前已经有了此key对应的value,则暂时e = p;
至于另外一种情况就是位置冲突了,即根据(n - 1) & hash算法发生了碰撞,再次分情况讨论;
1.以链表的形式存入;
2.如果碰撞导致链表过长(大于等于TREEIFY_THRESHOLD),就把链表转换成红黑树; 最后,如果e不为空,将e添加到table中(e.value 被赋值为 putVal()中的参数 value);
代码如下:
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
Node<K,V>[] tab; Node<K,V> p; int n, i;
//hashmap对象中 tabel属性为空--->第一次put---->resize()
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
//发现tab[i] 没有值,直接存入即可
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else {
//tab[i]取到值了,莫慌,先定义下方2个变量
Node<K,V> e; K k;
//如果是 key 重复了 很简单,直接e = p
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);
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;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}