HsahMap源码分析

  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)
            //进行table扩容
            n = (tab = resize()).length;
        if ((p = tab[i = (n - 1) & hash]) == null) //(n - 1) & hash 表示获取数组中的位置
            //p == null 表示p在当前数组中位置不存在,则直接存放到数组当前位置
            tab[i] = newNode(hash, key, value, null);
        else { //表示新元素在数组中存在
            Node<K,V> e; K k;
            //判断当前存入元素的k和数组中当前位置已经存在元素的k值是否相同
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p; //把当前数组中存在的元素赋值给e
            else if (p instanceof TreeNode) //判断数组中当前位置元素是否为tree的类型
                //把数据转变成树结构
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            else {
                //以上条件都不满足,则以链表的形式存储
                for (int binCount = 0; ; ++binCount) {
                    //如果e==null当前的p节点为链表的头节点
                    if ((e = p.next) == null) {
                        //直接把新数据插入到链表头结点后面
                        p.next = newNode(hash, key, value, null);
                   //TREEIFY_THRESHOLD=8;TREEIFY_THRESHOLD表示链表转成红黑树的判断条件
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            //把数组中当前位置的链表结构转成红黑树
                            treeifyBin(tab, hash);
                        break;
                    }
                //如果新插入的k值和链表中下个节点已经存在的k值相同,则e p.next并退出循环
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    p = e;  //把链表下个节点赋值给p
                }
            }
            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;
    }
    
    
    
    
    
    
    
    
final TreeNode<K,V> putTreeVal(HashMap<K,V> map, Node<K,V>[] tab,
                                       int h, K k, V v) {
           // map  当前Hashmap对象
           //tab   Hashmap对象中的table数组
           //h      hash值
           // K     key
           // V     value
            Class<?> kc = null;
            boolean searched = false;   //标识是否被收索过
            TreeNode<K,V> root = (parent != null) ? root() : this; // 找到root根节点
            for (TreeNode<K,V> p = root;;) {    //从根节点开始遍历循环
                int dir, ph; K pk;
                // 根据hash值 判断方向
                if ((ph = p.hash) > h)
                // 大于放左侧
                    dir = -1;
                else if (ph < h)
                // 小于放右侧
                    dir = 1;
                    // 如果key 相等  直接返回该节点的引用 外边的方法会对其value进行设置
                else if ((pk = p.key) == k || (k != null && k.equals(pk)))
                    return p;
                
                /**
                *下面的步骤主要就是当发生冲突 也就是hash相等的时候
                * 检验是否有hash相同 并且equals相同的节点。
                * 也就是检验该键是否存在与该树上
                */
                //说明进入下面这个判断的条件是 hash相同 但是equal不同
                // 没有实现Comparable<C>接口或者 实现该接口 并且 k与pk Comparable比较结果相同
                else if ((kc == null &&
                          (kc = comparableClassFor(k)) == null) ||
                         (dir = compareComparables(kc, k, pk)) == 0) {
                    //在左右子树递归的寻找 是否有key的hash相同  并且equals相同的节点
                    if (!searched) {
                        TreeNode<K,V> q, ch;
                        searched = true;
                        if (((ch = p.left) != null &&
                             (q = ch.find(h, k, kc)) != null) ||
                            ((ch = p.right) != null &&
                             (q = ch.find(h, k, kc)) != null))
                             //找到了 就直接返回
                            return q;
                    }


                    //说明红黑树中没有与之equals相等的  那就必须进行插入操作
                    //打破平衡的方法的 分出大小 结果 只有-1 1 
                    dir = tieBreakOrder(k, pk);
                }
                //下列操作进行插入节点
                //xp 保存当前节点
                TreeNode<K,V> xp = p;
                //找到要插入节点的位置
                if ((p = (dir <= 0) ? p.left : p.right) == null) {
                    Node<K,V> xpn = xp.next;
                    //创建出一个新的节点
                    TreeNode<K,V> x = map.newTreeNode(h, k, v, xpn);
                    if (dir <= 0)
                    //小于父亲节点  新节点放左孩子位置
                        xp.left = x;
                    else
                    //大于父亲节点  放右孩子位置
                        xp.right = x;
                    
                    //维护双链表关系
                    xp.next = x;
                    x.parent = x.prev = xp;
                    if (xpn != null)
                        ((TreeNode<K,V>)xpn).prev = x;
                    //将root移到table数组的i 位置的第一个节点
                    //插入操作过红黑树之后 重新调整平衡。
                    moveRootToFront(tab, balanceInsertion(root, x));
                    return null;
                }
            }
        }
添加树形节点,与添加双链表节点个过程类似:

1、从root节点开始寻找 如果目标k.hash 小于 当前节点的 hash ,那么到左树寻找,大于那么从右树寻找。如果找到key相同并且equals相同的节点 p 那就直接返回。
2、如果hash相同 但是equal不同 进而通过Comparable接口,进行比较,如果比较的结果如果还是相等,则在左右子树递归的寻找是否有与要插入的key equals相同的元素。如果有那么直接return返回。
(也即是没实现Comparable接口,大小由hash判定。实现了,则由Comparable接口的比较方法判定)
3、如果遍历完所有的节点 并未找到equals相同的节点。那就需要插入该新节点。必须分出大小,所以通过执行tieBreakOrder方法,该方法的返回值是-1,1。如果是-1则插入到左边节点,1就插入到右边节点。
4、插入完成之后,需要重新移动root节点 到table数组的i位置的第一个节点上 并且需重新平衡红黑树。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。