HashMap源码分析

对部分HashMap源码进行分析,然后逐步分析每一步的用意,结合一些HashMap数据结构的文章视频来理解源码的用意,总结经验就是有时候不能太注重每一步的细节,从大的结构开始模糊得了解,逐步到细节的实现的了解,这样才能全方位地理解。下面展示的是HashMap的结构图:整体来说是个数组链表结构,链表长度超过8会变成红黑树

put方法调用的putVal方法源码分析

/**
 * 实现 Map.put 和有关方法
 *
 * hash来自hash(key)方法不细究,是key经过运算的hash值
 * key:key值
 * value:value值
 * onlyIfAbsent:如果为true,请不要更改现有值
 * evict:如果为false,则表处于创建模式。
 * 返回先前的值;如果没有,则为null
 */
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
               boolean evict) {
    Node<K,V>[] tab; Node<K,V> p; int n, i;
    //如果table为空的时候就调用resize()方法初始化table
    if ((tab = table) == null || (n = tab.length) == 0)
        n = (tab = resize()).length;
    //通过(n - 1) & hash的hash运算来算出键值对在tab的位置,
    //如果是null说明当前算出的tab数组位置没有元素,就直接newNode一个Node放到tab数组对应位置中
    if ((p = tab[i = (n - 1) & hash]) == null)
        tab[i] = newNode(hash, key, value, null);
    //如果上面判断算出不为null,说明当前算出的tab数组位置有元素,进行下面的数据处理
    else {
        Node<K,V> e; K k;
        //如果发现上面算出的键值对和当前位置的键是一样的则直接赋值Node<K,V>的p给e,后面再进行赋值
        if (p.hash == hash &&
            ((k = p.key) == key || (key != null && key.equals(k))))
            e = p;
        //如果p是TreeNode类型,直接进入putTreeVal方法去操作这个树,赋值给e,后面再进行赋值
        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) {
                    p.next = newNode(hash, key, value, null);
                    //如果发现链表元素总数超过了TREEIFY_THRESHOLD就将链表转化为树
                    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不为null,说明上面的有判断生效,e被赋值,直接替换当前位置的value值
        if (e != null) { // existing mapping for key
            V oldValue = e.value;
            if (!onlyIfAbsent || oldValue == null)
                e.value = value;
            afterNodeAccess(e);
            return oldValue;
        }
    }
    ++modCount;
    //如果集合大于阈值则做resize重排扩容操作(阈值是在初始化集合的时候resize方法赋值的,默认大小是16*0.75)
    if (++size > threshold)
        resize();
    afterNodeInsertion(evict);
    return null;
}

resize方法源码分析

final Node<K, V>[] resize(){
    //原先的Node数组
    Node<K, V>[] oldTab = table;
    //原先的容量就是table的长度
    int oldCap = (oldTab == null) ? 0 : oldTab.length;
    //原先的阈值
    int oldThr = threshold;
    int newCap, newThr = 0;
    if (oldCap > 0) {
        //如果原先的容量大于最大容量,则将阈值赋值为int的最大值
        if (oldCap >= MAXIMUM_CAPACITY) {
            threshold = Integer.MAX_VALUE;
            return oldTab;
        }
        //如果原先容量的两倍(同时将新的容量扩容为原先容量的两倍)小于最大容量并且原先容量大于等于默认初始容量,
        // 则新的阈值等于原先阈值的两倍
        else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY && oldCap >= DEFAULT_INITIAL_CAPACITY) {
            newThr = oldThr << 1;
        }
    }
    //如果原先的阈值大于0,则容量就等于阈值
    else if (oldThr > 0) {
        newCap = oldThr;
    }
    //上面都不满足说明是新初始化map,新的容量等于默认初始化容量16;阈值=初始容量*负载系数
    else {
        newCap = DEFAULT_INITIAL_CAPACITY;
        newThr =(int)(DEFAULT_INITIAL_CAPACITY * DEFAULT_LOAD_FACTOR);
    }
    //如果新的容量等于0,说明上面的if代码没有赋值newThr,说明原先阈值大于0,则容量就等于阈值,
    //这时计算新的阈值=容量*负载系数(此处的负载系数loadFactor的值由初始化HashMap所调用的构造函数决定)
    //如果容量和这个计算出来的阈值只要有一个大于最大容量MAXIMUM_CAPACITY,就给阈值赋为int的最大值
    if (newThr == 0) {
        float ft = (float) newCap * loadFactor;
        newThr = (newCap < MAXIMUM_CAPACITY && ft < (float) MAXIMUM_CAPACITY ? (int)ft : Integer.MAX_VALUE);
    }
    //将newThr赋值给全局阈值
    threshold = newThr;
    @SuppressWarnings({"rawtypes", "unchecked"})
    Node<K, V>[] newTab = (Node<K, V>[]) new Node[newCap];
    table = newTab;
    //如果oldTab不为null,说明触发了重排序进入下面步骤,否则初始化无元素的话直接返回newTab
    if (oldTab != null) {
        //循环遍历原先的链表数组oldTab
        for (int j = 0; j < oldCap; j++) {
            Node<K, V> e;
            if ((e = oldTab[j]) != null) {
                oldTab[j] = null;
                //oldTab[j]位置的node链表如果没有next元素,说明只有一个元素,直接赋值给newTab数组妥当位置即可
                if (e.next == null) {
                    newTab[e.hash & (newCap - 1)] = e;
                }
                //如果oldTab[j]位置是红黑树,使用TreeNode的split()方法处理oldTab[j]位置的元素
                else if (e instanceof TreeNode) {
                    ((TreeNode<K, V>) e).split(this, newTab, j, oldCap);
                } else {
                    //不需要移动的链表头和尾
                    Node<K, V> loHead = null, loTail = null;
                    //需要移动的链表头和尾
                    Node<K, V> hiHead = null, hiTail = null;
                    Node<K, V> next;
                    do {
                        next = e.next;
                        //如果还是原来的索引值
                        if ((e.hash & oldCap) == 0) {
                            if (loTail == null) {
                                loHead = e;
                            } else {
                                loTail.next = e;
                            }
                            loTail = e;
                        }
                        //不是原来的索引值
                        else {
                            if (hiTail == null) {
                                hiHead = e;
                            } else {
                                hiTail.next = e;
                            }
                            hiTail = e;
                        }
                    } while ((e = next) != null);
                    //把以loHead为首的链表放到数组的原位置
                    if (loTail != null) {
                        loTail.next = null;
                        newTab[j] = loHead;
                    }
                    //把以hiHead为首的链表放到原位置+oldCap的位置
                    if (hiTail != null) {
                        hiTail.next = null;
                        newTab[j + newCap] = hiHead;
                    }
                }
            }
        }
    }
    return newTab;
}

get方法调用的getNode方法源码分析

final Node<K,V> getNode(int hash, Object key) {
    Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
    //如果table不为null&&长度大于0
    if ((tab = table) != null && (n = tab.length) > 0 &&
        (first = tab[(n - 1) & hash]) != null) {
        //根据hash与运算算出的tab对应元素的Node为first,key为要找的key就直接返回
        if (first.hash == hash && // always check first node
            ((k = first.key) == key || (key != null && key.equals(k))))
            return first;
        //上面的判断没进如则判断下一个元素位置
        if ((e = first.next) != null) {
            //如果发现根据hash与运算算出的tab对应元素位置是树结构数据
            //则进入getTreeNode方法查找对应元素
            if (first instanceof TreeNode)
                return ((TreeNode<K,V>)first).getTreeNode(hash, key);
            //如果不是树结构数据,则一直查找下一个元素位置数据,
            //判断对应key是否与要找的相同,直到找到相同的返回结果
            do {
                if (e.hash == hash &&
                    ((k = e.key) == key || (key != null && key.equals(k))))
                    return e;
            } while ((e = e.next) != null);
        }
    }
    return null;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注