JAVA源码剖析之HashMap（五）：get()方法

get()方法

get()方法比较简单，首先通过hash值定位指定key的table[]数组索引，其次优先比较第一个元素是否相等，如果不相等则走红黑树的getTreeNode()逻辑或者链表遍历查找比较。

public V get(Object key) {
    Node<K,V> e;
    return (e = getNode(hash(key), key)) == null ? null : e.value;
}

final Node<K,V> getNode(int hash, Object key) {
    Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
    if ((tab = table) != null && (n = tab.length) > 0 &&
        (first = tab[(n - 1) & hash]) != null) {
        if (first.hash == hash && // 总是检查第一个节点
            ((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;
}

红黑树的查找

思路如下：底层通过find()函数实现，通过二分查找，hash值小于父节点的进入左节点查找，hash值大于父节点的进入右节点查找，如果不通过hash值比较，则通过比较器返回值判断进入左右节点查找，比较器也无法比较的则先进入右节点查找，找不到再进入左节点查找。

final TreeNode<K,V> getTreeNode(int h, Object k) {
    return ((parent != null) ? root() : this).find(h, k, null);
}
final TreeNode<K,V> find(int h, Object k, Class<?> kc) {
    TreeNode<K,V> p = this;
    do {
        int ph, dir; K pk;
        TreeNode<K,V> pl = p.left, pr = p.right, q;
        if ((ph = p.hash) > h) // 如果给定哈希值小于当前节点的哈希值，进入左节点
            p = pl;
        else if (ph < h) // 如果大于，进入右结点
            p = pr;
        // 如果哈希值相等，且对象实例相等或者equals比较相等，则返回当前节点
        else if ((pk = p.key) == k || (k != null && k.equals(pk)))
            return p;
        else if (pl == null)
            p = pr;
        else if (pr == null)
            p = pl;
        // 如果不按哈希值排序，而是按照比较器排序，则通过比较器返回值决定进入左右结点
        else if ((kc != null ||
                  (kc = comparableClassFor(k)) != null) &&
                 (dir = compareComparables(kc, k, pk)) != 0)
            p = (dir < 0) ? pl : pr;
        else if ((q = pr.find(h, k, kc)) != null) // 进入右节点查找
            return q;
        else
            p = pl; // 进入左节点查找
    } while (p != null);
    return null;
}