【JUC源码】并发容器:ConcurrentHashMap(四)获取 value 源码分析

ConcurrentHashMap 系列:

get()

ConcurrentHashMap 读的话,就比较简单。先获取数组的下标,然后通过判断数组下标的 key 是否和我们的 key 相等,相等的话直接返回,如果下标的槽点是链表或红黑树的话,分别调用相应的查找数据的方法,整体思路和 HashMap 很像,源码如下:

public V get(Object key) {
    Node<K,V>[] tab; Node<K,V> e, p; int n, eh; K ek;
    // 计算hashcode,后面用于判断是什么类型节点,然后用对应的查找方法
    int h = spread(key.hashCode());
    // 判断数组不为空 && 当前索引的槽点数据不是空的
    if ((tab = table) != null && (n = tab.length) > 0 &&
        (e = tabAt(tab, (n - 1) & h)) != null) {
        // 情况一:槽点第一个值和key相等,直接返回
        if ((eh = e.hash) == h) {
            if ((ek = e.key) == key || (ek != null && key.equals(ek)))
                return e.val;
        }
        // 情况二:红黑树或者转移节点,调用 TreeBin 的 find方法
        // 注:TreeBin 已经保存了 root 结点,所以可以直接调用 root.find() 去寻找对应 key 的 value
        else if (eh < 0)
            return (p = e.find(h, key)) != null ? p.val : null;
        // 情况三:如果是链表,遍历查找
        while ((e = e.next) != null) {
            if (e.hash == h &&
                ((ek = e.key) == key || (ek != null && key.equals(ek))))
                return e.val;
        }
    }
    // 否则该key对应的值不存在,返回null
    return null;
}

HashMap 和 ConcurrentHashMap 对比

虽然 ConcurrentHashMap 的底层数据结构,和方法的实现细节和 HashMap 大体一致,但两者在类结构上却没有任何关联:

1)ConcurrentHashMap 和 HashMap 两者的相同之处:

  • 数组、链表结构几乎相同,所以底层对数据结构的操作思路是相同的(只是思路相同,底层实现不同);
  • 都实现了 Map 接口,继承了 AbstractMap 抽象类,所以大多数的方法也都是相同的,HashMap 有的方法,ConcurrentHashMap 几乎都有,所以当我们需要从 HashMap 切换到 ConcurrentHashMap 时,无需关心两者之间的兼容问题。

2)ConcurrentHashMap 和 HashMap 两者的不同之处:

  • 红黑树结构略有不同,HashMap 的红黑树中的节点叫做 TreeNode,TreeNode 不仅仅有属性,还维护着红黑树的结构,比如说查找,新增等等;ConcurrentHashMap 中红黑树被拆分成两块,TreeNode 仅仅维护的属性和查找功能,新增了 TreeBin,来维护红黑树结构,并负责根节点的加锁和解锁;
  • 新增 ForwardingNode (转移)节点,扩容的时候会使用到,通过使用该节点,来保证扩容时的线程安全。

猜你喜欢

转载自blog.csdn.net/qq_33762302/article/details/114421333