一、put方法
HashMap的put方法执行流程大致可以分为以下几步
- 计算 key 的哈希值
- 查到 key 在 HashMap 中的位置
- 若查找到 key 在 HashMap 中的位置,将 value 覆盖原值
- 若在 HashMap 中查找不到 key ,新建一个节点存储 key 和 value
话不多说直接看源码
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
可以看出在 put 方法中没有实现逻辑,而是调用了putVal 方法,我们接着往下看
/**
* Implements Map.put and related methods
*
* @param hash key的哈希值
* @param key 输入的key
* @param value 输入的value
* @param onlyIfAbsent true表示如果已经存在key时,不覆盖原来的value
* @param evict 表是否在创造状态,false表示在创造状态
* @return previous value, or null if none
*/
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为空,初始化table
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
//求出 key在 table中的位置
//若该位置为空,则新建一个节点,将key和value存储进table
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else {
//该位置不为空,查找key在HashMap中的位置
Node<K,V> e; K k;
//判断第一个元素是否为key在HashMap中的位置
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 {
//桶结构为链表,遍历链表查找key在HashMap中的位置
for (int binCount = 0; ; ++binCount) {
//链表中查到不到key,则新建一个节点存储key和value
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;
}
}
//若在桶中查找到key,将value覆盖原值
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
//记录的修改次数+1
++modCount;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}
下面这张图能够更直观的体现 HashMap 的 put 流程
图片来源
二、get方法
get 方法和 put 方法的流程差不多。可以说更简单,因为 get 只需要查找到 key 在 HashMap 中的位置
public V get(Object key) {
Node<K,V> e;
return (e = getNode(hash(key), key)) == null ? null : e.value;
}
可以看出,get 方法通过 getNode 方法,得到目标节点。若节点为空,返回 null,否则返回节点的 value。接下来,我们继续看 getNode 方法。
final Node<K,V> getNode(int hash, Object key) {
Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
//若 table为空或者通过hash计算得到的位置为空,直接返回null
if ((tab = table) != null && (n = tab.length) > 0 &&
(first = tab[(n - 1) & hash]) != null) {
//若桶中第一个元素的key和hash和查找的key,hash相等,返回该元素
if (first.hash == hash && // always check first node
((k = first.key) == key || (key != null && key.equals(k))))
return first;
if ((e = first.next) != null) {
//桶结构为红黑树,遍历红黑树查找key的位置
if (first instanceof TreeNode)
return ((TreeNode<K,V>)first).getTreeNode(hash, key);
//桶结构为链表,遍历链表查找key在HashMap中的位置
do {
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
return e;
} while ((e = e.next) != null);
}
}
return null;
}