HashMap的put()方法源码分析

jak版本
java version “1.8.0_181”
查看put()源码 先调用hash()

public V put(K key, V value) {
    return putVal(hash(key), key, value, false, true);
}

进去hash(key)，这里是计算key的hash值

static final int hash(Object key) {
    int h;
    // 如果key是null，直接返回0，就是放在数组的第一个
    // 若key不是null，先计算key的hashCode()得到h临时值，再h>>>16，两个结果再异或，这样写会降低hash冲突
    // 〉〉〉
    return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
}

再进去putVal(hash(key), key, value, false, true);方法

final V putVal(int hash, K key, V value, boolean onlyIfAbsent,boolean evict) {
    // n是数组的容量，i是元素要存储在数组的下标
    Node<K,V>[] tab; Node<K,V> p; int n, i;
    // 如果数组是null，或者数组的长度是0，就扩容
    if ((tab = table) == null || (n = tab.length) == 0)
        n = (tab = resize()).length;
    // 数组下标对应的位置没有元素
    // n是集合的容量，一直都是2的幂次方
    // 若是n&hash，则碰撞的几率比较高，因为如16的二进制是10000，和很多数&后结果是一样的，碰撞率高
    // n-1,就是基数，如16-1=15。二进制是1111，和很多数&后的结果是不一样的，碰撞率低
    // 两个数&后的结果吧两个值都小，可以作为数组的下标
    if ((p = tab[i = (n - 1) & hash]) == null)    
        // 创建一个数组的结点,是个Node,存储key的hash值,key,value,下一个结点指向,暂时为null                          
        tab[i] = newNode(hash, key, value, null);
    else {                                        // 有哈希冲突                                          
        Node<K,V> e; K k;
        // 判断新存入元素和数组中的元素是否key相等 若一样，那新增结点的value覆盖原结点value
        if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k))))     
            e = p;
        // 结点是TreeNode类型(红黑树)，意思就是当数组容量>64且链表长度>8，链表成为了红黑树，
        // 那么直接向树中插入元素    
        else if (p instanceof TreeNode)                                    
            e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
        // 有哈希冲突，是链表 
        else {
            // 有哈希冲突，是链表 ，若链表中有结点相同，那么会覆盖
            // 若链表长度>8，那么会进行数组容量判断，
            for (int binCount = 0; ; ++binCount) {
                // 若是最后一个节点，就把新节点放在这个节点后面
                if ((e = p.next) == null) {
                    p.next = newNode(hash, key, value, null);
                    // 若循环次数大于等于7，从0开始，也就是链表长度>8，
                    // 进入treeifyBin()进行判断数组容量,再决定是否转红黑树
                    if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                        // 进去之后判断数组容量，如果大于64，那么会转换成红黑树
                        treeifyBin(tab, hash);
                    break;
                }
                // 在遍历的过程中，若有结点与新增结点相同，在后面把value覆盖了
                if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k))))
                    break;
                p = e;
            }
        }
        // e结点是和新添加的元素key相同的节点,把新添加的元素的value设置到e的value上
        // 调用put()添加元素时,onlyIfAbsent传入是false,
        if (e != null) { // existing mapping for key
            V oldValue = e.value;
            if (!onlyIfAbsent || oldValue == null)
                e.value = value;
            // 给linkedHashMap用的    
            afterNodeAccess(e);
            // 返回旧的value
            return oldValue;
        }
    }
    // 加1 保证遍历时数据不会变
    ++modCount;
    // 添加完元素后判断是否大于阈值，大于就扩容
    if (++size > threshold)
        resize();
    // 给LinkedHashMap用的    
    afterNodeInsertion(evict);
    return null;
}

---

treeifyBin()是否要转红黑树

final void treeifyBin(Node<K,V>[] tab, int hash) {
    int n, index; Node<K,V> e;
    // 若数组长度小于64，只扩容
    if (tab == null || (n = tab.length) < MIN_TREEIFY_CAPACITY)
        resize();
    // 若数组长度大于等于64，转红黑树
    else if ((e = tab[index = (n - 1) & hash]) != null) {
        TreeNode<K,V> hd = null, tl = null;
        do {
            TreeNode<K,V> p = replacementTreeNode(e, null);
            if (tl == null)
                hd = p;
            else {
                p.prev = tl;
                tl.next = p;
            }
            tl = p;
        } while ((e = e.next) != null);
        if ((tab[index] = hd) != null)
            hd.treeify(tab);
    }
}