HashMap中put方法（jdk8）

桃扇骨 2023-10-01 21:46 29阅读 0赞

## 一、put方法流程 ##

首先看一下网上一个不错的流程图

![watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAY3JhenlLLg_size_20_color_FFFFFF_t_70_g_se_x_16][]

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方法

![watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAY3JhenlLLg_size_20_color_FFFFFF_t_70_g_se_x_16 1][]

方法传入了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博客][HashMap_resize_crazyK._-CSDN]

[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAY3JhenlLLg_size_20_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/17134f88ace944339034b765410631e5.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAY3JhenlLLg==,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAY3JhenlLLg_size_20_color_FFFFFF_t_70_g_se_x_16 1]: https://img-blog.csdnimg.cn/6ba8fb3a4db1449e9219279b6d98ec2f.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAY3JhenlLLg==,size_20,color_FFFFFF,t_70,g_se,x_16
[HashMap_resize_crazyK._-CSDN]: https://blog.csdn.net/weixin_57504474/article/details/124428081?spm=1001.2014.3001.5501