一、put方法流程
首先看一下网上一个不错的流程图
1.先判断数组是否为空,如果为空则对数组进行resize()扩容
2.然后把插入元素的哈希值和数组长度-1进行与操作,如果table[i] == null,则直接将元素插入到数组对应位置中,否则进行3
3.判断当前元素的key在数组中是否存在(通过hash并equals方法判断),然后判断节点类型,如果是红黑树,则走4;如果是链表,则走5
4.如果节点类型为红黑树,则执行红黑树插入操作
5.如果节点类型为链表,for循环遍历链表直至链表尾部,然后进行尾插,当链表长度>=8时,会进入链表转红黑树的方法,treeifyBin方法中还会判断数组长度,数组长度>=64,链表长度>=8同时满足,才会将链表转为红黑树;在for循环遍历过程中,如果key相同,则直接插入元素
6.put方法是由返回值的,在插入完成后,如果插入之前已经存在这个key,则返回的是插入之前已存在元素的value
7.记录操作次数变量modCount+1,最后再判断当前map中有多少元素,和阈值做对比,如果超过阈值则进行扩容,没有则返回null
二、put方法源码
首先定义一个HashMap,向map中put一个元素
public class HashMapTest {
public static void main(String[] args) {
Map<String,String> map = new HashMap<>();
map.put("k","v");
}
}
按住ctrl进入put方法找到HashMap的pu方法
方法传入了hash值,key和value,然后返回了一个putVal方法,put的方法的核心就是putVal方法
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
直接上源码
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
//初始化变量--> 提高性能 ; p --> 新插入的元素
Node<K,V>[] tab; Node<K,V> p; int n, i;
//判空,初始化数组
if ((tab = table) == null || (n = tab.length) == 0)
//resize() --> 扩容
n = (tab = resize()).length;
//往map中放数据:先判空,如果为空,创建一个新的Node节点
//tab[i = (n - 1) & hash] --> 计算数组下标
// 计算方式: 将元素hash值和数组长度-1做与操作
// i 就是 key值 范围要在 0 - (n-1) 之间
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else {//如果要插入的元素在这个位置有元素了,执行以下操作
Node<K,V> e; K k;
//比较两个元素的key是否相同
//p --> 插入新元素之前该位置已经存在的元素
// 比较 哈希值 : 引用地址 : key
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
//如果相等则老元素地址指向新元素
e = p;
//不相等则判断节点类型:树 TreeNode是Node的一个子类
else if (p instanceof TreeNode)
//红黑树的插入操作
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
//节点类型:链表
else {
//for循环:1.遍历到链表尾部,进行尾插
//2.判断链表长度,超过8将链表改为红黑树
for (int binCount = 0; ; ++binCount) {
//判断节点的下一个节点为空,遍历到链表尾部,进行尾插
if ((e = p.next) == null) {
//生成一个Node对象,将Node对象作为新节点插入到链表(p.next)
p.next = newNode(hash, key, value, null);
//如果链表长度 >= TREEIFY_THRESHOLD-1 = 7 因为是从0开始遍历,所以此时链表长度为8
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
//进入转红黑树的方法,treeifyBin方法中还会判断数组长度,数组长度>=64,链表长度>=8同时满足,才会将链表转为红黑树
treeifyBin(tab, hash);
break;
}
//key相同时同样插入返回
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
if (e != null) { // existing mapping for key
//记录已经存在元素key的value
V oldValue = e.value;
//onlyIfAbsent put方法的参数,默认为false;如果调用热put方法putIfAbsent,默认为true
if (!onlyIfAbsent || oldValue == null)
//新元素的值将老元素的值覆盖掉
e.value = value;
//HashMap提供给子类的方法
afterNodeAccess(e);
//put操作有返回值,返回的是插入之前已经存在的元素的value值
return oldValue;
}
}
//增加修改次数
++modCount;
//统计当前map中有多少元素,和阈值对比,判断是否需要扩容
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}
HashMap扩容resize详解
:
HashMap扩容操作resize_crazyK.的博客-CSDN博客