CurrentHashMap 源码分析 initTable(),get() 方法

在jdk8中，集合类 ConcurrentHashMap 经 Doug Lea 大师之手，借助volatile语义以及CAS操作进行优化，使得该集合类更好地发挥出了并发的优势。与jdk7中相比，在原有段锁（Segment）的基础上，引入了数组＋链表＋红黑树的存储模型，在查询效率上花费了不少心思。

ConcurrentHashMap内存存储结构图大致如下：

/**
 * Initializes table, using the size recorded in sizeCtl.
 */
private final Node<K,V>[] initTable() {
    Node<K,V>[] tab; int sc;
    while ((tab = table) == null || tab.length == 0) {
        if ((sc = sizeCtl) < 0)
            Thread.yield(); // lost initialization race; just spin
        else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
            try {
                //此处为什么要再次判断
                if ((tab = table) == null || tab.length == 0) {
                    int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
                    @SuppressWarnings("unchecked")
                    Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                    table = tab = nt;
                    sc = n - (n >>> 2);
                }
            } finally {
                sizeCtl = sc;
            }
            break;
        }
    }
    return tab;
}

Q： 为什么在else代码中要重复判断table是否为空？
A： 在多线程中，有可能多个线程进入到else代码块的。
举个例子：

• 假如线程a，b同时 执行 initTable方法:
  此时sizeCtl为0，a，b同时执行U.compareAndSwapInt(this,SIZECTL, sc, -1 )，但只有一个线程判断为true进入到else中去。

• 假如线程a成功，此时原子操作将sizeCtl设置为－1，线程b则重新执行while循环（此时table未初始化），此时sizeCtl为－1，则线程b让出cpu。

• 继续来看线程a ,在执行到finally{ sizeCtl ＝ sc } 代码前，sizeCtl一直为－1，线程b只能永远执行Thread.yield() 让出cpu。

• 但在线程a执行sizeCtl = sc 之后，线程b可能会进入运行态 此时src = sizeCtl (等于 n - (n >>> 2));
  所以线程b会重新执行U.compareAndSwapInt(this, SIZECTL, sc, -1 )判断， 线程b将 sizeCtl 设置为－1. 此时判断table != null。执行finally代码，将sizeCtl重新置为sc。(如果此时不判断table是否为null,则会再初始化一个table)

---

---

get方法思想：

1. 计算出该节点的hash值，并算出该node在table中的位置 e = tabAt(tab, (n - 1) & h)。
2. 判断首节点是不是要找的节点，是，直接返回。
3. 如果该节点hash值小于0 则可能是forwadingNode也可能是Treebin节点,直接调用forwadingNode.find或者Treebin节点的find方法。
4. 如果该节点hash值大于等于0，则是普通节点，通过遍历链表查到要找的节点。

 public V get(Object key) {
        Node<K,V>[] tab; Node<K,V> e, p; int n, eh; K ek;
        //计算出该节点的hash值 
        int h = spread(key.hashCode());
        if ((tab = table) != null && (n = tab.length) > 0 &&
            (e = tabAt(tab, (n - 1) & h)) != null) {
            //判断首节点是不是要找的节点
            if ((eh = e.hash) == h) {
                if ((ek = e.key) == key || (ek != null && key.equals(ek)))
                    return e.val;
            }
            //forwadingNode或者Treebin
            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;
            }
        }
        return null;
    }

参考：https://bentang.me/tech/2016/12/01/jdk8-concurrenthashmap-1/