JDK源码学习之HashMap的put函数实现

HashMap的put函数

1、首先是计算hash值

使用key的哈希值和key哈希值的低16为进行按位异或操作，得到hash值

    static final int hash(Object key) {
        int h;
        return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
    }

2、put函数接口

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

3、putVal实现

因为哈希桶数组的大小为2的整数次幂，因此 i = (n - 1) & hash就是应该插入的桶的索引。
但是索引位置的桶里面可能不只有一个元素，也有可能哈希桶中存的就是这个数据，所以需要继续进行判断。

如果对应桶里面为null，那就直接创建一个新的桶。
如果不为null,那么来看一下第一个位置是不是当前要插入的值。
如果不是，那就看一下是不是红黑树，如果是，那就进行红黑树的插入操作
如果不是红黑树，那就找一下桶里面有没有自己，如果没有，那就把新的元素放在桶的尾部，也就是链表的尾部。在插入后，看一下是不是达到了转换阈值，如果达到了转换阈值，那就把链表转换为红黑树。

    /**
     * 三个比较有用的变量，hash，key,value.
     * 
     */
    final V putVal(int hash, K key, V value, boolean onlyIfAbsent, boolean evict) {
        Node<K,V>[] tab; Node<K,V> p; int n, i;//这里的tab是哈希桶数组，p代表新值位置的哈希桶(也就是那个链表)
        if ((tab = table) == null || (n = tab.length) == 0)//如果哈希桶数组为null，还没有初始化，那么初始化到默认大小
            n = (tab = resize()).length;
        if ((p = tab[i = (n - 1) & hash]) == null)//计算在哈希桶中的索引，这里 (n - 1) & hash就相当于做了取模运算
            tab[i] = newNode(hash, key, value, null);//如果计算位置的桶为null，那就创建一个新的节点(桶)
        else {//如果不为null（也是就是发生了哈希冲突）
            Node<K,V> e; K k;
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))//如果哈希值和key都一样，那么刚才的数据就是已存在的
                e = p;
            else if (p instanceof TreeNode)//看一下是不是红黑树，如果是，那就按红黑树添加的方法添加
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);//
            else {
                for (int binCount = 0; ; ++binCount) {
                    if ((e = p.next) == null) {//如果下一个元素为null，那么插到它的后面
                        p.next = newNode(hash, key, value, null);
                        if (binCount >= TREEIFY_THRESHOLD - 1) // 如桶里面的元素数量超过阈值，那就转化为红黑树
                            treeifyBin(tab, hash);
                        break;
                    }
                    //看一下新的元素是不是已经在哈希桶数组的这个桶里面，如果是，那就结束了
                    if (e.hash == hash &&((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    p = e;//再找一下桶里的下一个元素
                }
            }
            if (e != null) { // existing mapping for key
                V oldValue = e.value;
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
        ++modCount;
        if (++size > threshold)//哈希桶数组大小超过阈值，进行扩容
            resize();
        afterNodeInsertion(evict);
        return null;
    }