HashMap源码分析(二)

一、HashMap的get()方法:

public V get(Object key) {
    Node<K,V> e;
    //get方法里又调用了getNode方法
    return (e = getNode(hash(key), key)) == null ? null : e.value;
}

 1.getNode():

final Node<K,V> getNode(int hash, Object key) {
        //tab:当前的散列表
        //first:桶位中的头元素
        //e:node节点
        //n:tab表的长度  
        Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
        //初始化过数组，且当前tab表的长度不为零，能根据hash值找到第一个桶位
        if ((tab = table) != null && (n = tab.length) > 0 &&
            (first = tab[(n - 1) & hash]) != null) {
            //第一种情况:当前桶位元素就是要找的元素
            if (first.hash == hash && // always check first node
                ((k = first.key) == key || (key != null && key.equals(k))))
                return first;
            if ((e = first.next) != null) {
                //第二种情况:当前情况是红黑树的情况
                if (first instanceof TreeNode)
                    return ((TreeNode<K,V>)first).getTreeNode(hash, key);
                //第三种情况:当前情况是链表的情况
                do {
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        return e;
                } while ((e = e.next) != null);
            }
        }
        return null;
    }

二、HashMap的remove()方法:

public V remove(Object key) {
    Node<K,V> e;
    //调用了removeNode方法
    return (e = removeNode(hash(key), key, null, false, true)) == null ? null : e.value;
}

 1.removeNode():

final Node<K,V> removeNode(int hash, Object key, Object value, boolean matchValue,boolean movable) {
        //tab:当前的散列表
        //p:当前node节点
        //index:寻址结果  
        Node<K,V>[] tab; Node<K,V> p; int n, index;
        //已经初始化过了
        if ((tab = table) != null && (n = tab.length) > 0 &&
            (p = tab[index = (n - 1) & hash]) != null) {
            //node:查找到的结果
            //e:node的下一个节点
            Node<K,V> node = null, e; K k; V v;
            //第一种情况:当前桶位的元素就是要删除的元素，赋值给node，后续在进行删除操作
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                node = p;
            //e指向p的下一个节点
            else if ((e = p.next) != null) {
                //第二种情况:红黑树的情况
                if (p instanceof TreeNode)
                    node = ((TreeNode<K,V>)p).getTreeNode(hash, key);
                else {
                    //第三种情况:链表的情况
                    do {
                        //找到了元素
                        if (e.hash == hash &&
                            ((k = e.key) == key ||
                             (key != null && key.equals(k)))) {
                            node = e;
                            break;
                        }
                        p = e;
                    } while ((e = e.next) != null);
                }
            }
            //删除操作
            //matchValue:传入的是false表示value也得是正确的
            //判断node不为空则代表找到了
            if (node != null && (!matchValue || (v = node.value) == value ||
                                 (value != null && value.equals(v)))) {
                //红黑树的情况
                if (node instanceof TreeNode)
                    ((TreeNode<K,V>)node).removeTreeNode(this, tab, movable);
                //桶位元素就是要删除的元素
                else if (node == p)
                    tab[index] = node.next;
                //前一个节点的下一个节点指向了node节点的下一个节点就代表删除了node节点
                else
                    p.next = node.next;
                //删除节点也代表修改了一次HashMap
                ++modCount;
                //长度减一
                --size;
                afterNodeRemoval(node);
                //返回node节点
                return node;
            }
        }
        return null;
    }