文章目录

一、put方法运行流程图

二、put方法源码分析

三、总结


相关文章

HashMap 扩容机制

一、put方法运行流程图

二、put方法源码分析

    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) {
        Node<K,V>[] tab; Node<K,V> p; int n, i;
        // 判断数组是否未初始化
        if ((tab = table) == null || (n = tab.length) == 0)
            // 未初始化,调用resize()方法,进行初始化操作
            n = (tab = resize()).length;
        // 通过 & 计算出该数据(key)的数组下标并判断该下标位置是否有数据
        if ((p = tab[i = (n - 1) & hash]) == null)
            // 如果没有,直接将数据放在该下标位置
            tab[i] = newNode(hash, key, value, null);
        else {// 该数组下标有数据的情况
            Node<K,V> e; K k;
            // 判断该位置数据的key是否和待插入Key匹配
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                /*
                 * 如果一样,证明为修改操作,该节点的数据赋值给e,后面会用
                 * 注:
                 *    变量 e 是 Node<K,V> 类型,用于存储匹配到的旧节点
                 *    若 e != null,说明 Key 已存在,后续会覆盖旧值;若 e == null,说明 Key 不存在,会新增节点
                 * */
                e = p;
            else if (p instanceof TreeNode)// 判断是否为红黑树
                // 如果为红黑树,执行红黑树的操作
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            else {
                // 待插入的key与当前位置数据的key不相同,也不是红黑树节点,则为链表,
                // 遍历链表
                for (int binCount = 0; ; ++binCount) {
                    // 判断next节点,如果为空的话,证明遍历到链表尾部了
                    if ((e = p.next) == null) {
                        // 把新值放入到链表尾部
                        p.next = newNode(hash, key, value, null);
                        // 因为插入了一条数据,所以判断链表长度是不是大于等于8
                        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;
                }
            }
            // 判断e是否为空(变量 e 是 Node<K,V> 类型,用于存储匹配到的旧节点)
            if (e != null) { // existing mapping for key
                // 不为空的话证明是修改操作,取出旧值
                V oldValue = e.value;
                // 一定会执行(因为onlyIfAbsent 传进方法来的是false)
                if (!onlyIfAbsent || oldValue == null)
                    // 将新值赋值当前节点
                    e.value = value;
                afterNodeAccess(e);
                // 返回旧值
                return oldValue;
            }
        }
        // 计数器,计算当前节点的修改次数
        ++modCount;
        // 当前数组中的数据数量如果大于扩容阈值
        if (++size > threshold)
            // 进行扩容操作
            resize();
        // 一个空方法,方法中无代码
        afterNodeInsertion(evict);
        // 添加操作时,返回空值
        return null;
    }

三、总结

  1. 判断键值对数组table是否为空或为null,否则执行resize()进行扩容(初始化操作)
  2. 根据键值key计算hash值得到数组索引
  3. 判断table[i]==null,条件成立,直接新建节点添加
  4. 如果table[i]==null,不成立
    1. 判断table[i]的首个元素是否和key一样,如果相同直接覆盖value
    2. 判断table[i]是否为treeNode,即table[i]是否为红黑树,如果是红黑树,则直接在树中插入键值对
    3. 遍历table[i],链表的尾部插入数据,然后判断链表长度是否大于8,大于8则把链表转换为红黑树,在红黑树中执行插入操作,遍历过程中若发现key已经存在直接覆盖value
      1. :当然转换红黑树还要数组的容量>64,put方法源码判断大于8会执行 treeifyBin方法,在treeifyBin方法中会判断数组长度小于64会执行扩容操作
    4. 插入成功后,判断实际存在的键值对数量size是否超多了最大容量threshold(数组长度*0.75),如果超过,进行扩容
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐