HashMap中put方法（jdk8）

一、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博客