浅入浅出 Java ConcurrentHashMap
共 5597字，需浏览 12分钟
·
2020-09-02 07:16
HashMap 是 Java 中非常强大的数据结构，使用频率非常高，几乎所有的应用程序都会用到它。但 HashMap 不是线程安全的，不能在多线程环境下使用，该怎么办呢？
1）Hashtable，一个老掉牙的同步哈希表，t 竟然还是小写的，一看就非常不专业：
public class Hashtable<K,V>
extends Dictionary<K,V>
implements Map<K,V>, Cloneable, java.io.Serializable {
public synchronized V put(K key, V value) {}
public synchronized int size() {}
public synchronized V get(Object key) {}
}
里面的方法全部是 synchronized，同步的力度非常大，对不对？这样的话，性能就没法保证了。pass。
2）Collections.synchronizedMap(new HashMap，可以把一个 HashMap 包装成同步的 SynchronizedMap：
private static class SynchronizedMap<K,V>
implements Map<K,V>, Serializable {
public int size() {
synchronized (mutex) {return m.size();}
}
public V get(Object key) {
synchronized (mutex) {return m.get(key);}
}
public V put(K key, V value) {
synchronized (mutex) {return m.put(key, value);}
}
}
可以看得出，SynchronizedMap 确实比 Hashtable 改进了，synchronized 不再放在方法上，而是放在方法内部，作为同步块出现，但仍然是对象级别的同步锁，读和写操作都需要获取锁，本质上，仍然只允许一个线程访问，其他线程被排斥在外。
3）ConcurrentHashMap，本篇的主角，唯一正确的答案。Concurrent 这个单词就是并发、并行的意思，所以 ConcurrentHashMap 就是一个可以在多线程环境下使用的 HashMap。
ConcurrentHashMap 一直在进化，Java 7 和 Java 8 就有很大的不同。Java 7 版本的 ConcurrentHashMap 是基于分段锁的，就是将内部分成不同的 Segment（段），每个段里面是 HashEntry 数组。
来看一下 Segment：
static final class Segment<K,V> extends ReentrantLock implements Serializable {
transient volatile HashEntry[] table;
transient int count;
transient int modCount;
transient int threshold;
final float loadFactor;
}
再来看一下 HashEntry：
static final class HashEntry<K,V> {
final K key; // 声明 key 为 final 型
final int hash; // 声明 hash 值为 final 型
volatile V value; // 声明 value 为 volatile 型
final HashEntry next; // 声明 next 为 final 型
HashEntry(K key, int hash, HashEntry next, V value) {
this.key = key;
this.hash = hash;
this.next = next;
this.value = value;
}
}
和 HashMap 非常相似，唯一的区别就是 value 是 volatile 的，保证 get 时候的可见性。
Segment 继承自 ReentrantLock，所以不会像 Hashtable 那样不管是 put 还是 get 都需要 synchronized，锁的力度变小了，每个线程只锁一个 Segment，对其他线程访问的 Segment 没有影响。
Java 8 和之后的版本在此基础上做了很大的改进，不再采用分段锁的机制了，而是利用 CAS（Compare and Swap，即比较并替换，实现并发算法时常用到的一种技术）和 synchronized 来保证并发，虽然内部仍然定义了 Segment，但仅仅是为了保证序列化时的兼容性，代码注释上就可以看得出来：
/**
* Stripped-down version of helper class used in previous version,
* declared for the sake of serialization compatibility.
*/
static class Segment<K,V> extends ReentrantLock implements Serializable {
final float loadFactor;
Segment(float lf) { this.loadFactor = lf; }
}
底层结构和 Java 7 也有所不同，更接近 HashMap（数组+双向链表+红黑树）：
来看一下新版 ConcurrentHashMap 定义的关键字段：
public class ConcurrentHashMap<K,V> extends AbstractMap<K,V>
implements ConcurrentMap<K,V>, Serializable {
transient volatile Node[] table;
private transient volatile Node[] nextTable;
private transient volatile int sizeCtl;
}
1）table，默认为 null，第一次 put 的时候初始化，默认大小为 16，用来存储 Node 节点，扩容时大小总是 2 的幂次方。
顺带看一下 Node 的定义：
static class Node<K,V> implements Map.Entry<K,V> {
final int hash;
final K key;
volatile V val;
volatile Node next;
// …
}
hash 和 key 是 final 的，和 HashMap 的 Node 一样，因为 key 是不会发生变化的。val 和 next 是 volatile 的，保证多线程环境下的可见性。
2）nextTable，默认为 null，扩容时新生成的数组，大小为原数组的两倍。
3）sizeCtl，默认为 0，用来控制 table 的初始化和扩容操作。-1 表示 table 正在初始化；-(1+线程数) 表示正在被多个线程扩容。
Map 最重要的方法就是 put，ConcurrentHashMap 也不例外：
public V put(K key, V value) {
return putVal(key, value, false);
}
final V putVal(K key, V value, boolean onlyIfAbsent) {
if (key == null || value == null) throw new NullPointerException();
int hash = spread(key.hashCode());
int binCount = 0;
for (Node[] tab = table;;) {
Node f; int n, i, fh;
if (tab == null || (n = tab.length) == 0)
tab = initTable();
else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
if (casTabAt(tab, i, null, new Node(hash, key, value, null)))
break; // no lock when adding to empty bin
}
else if ((fh = f.hash) == MOVED)
tab = helpTransfer(tab, f);
...省略部分代码
}
addCount(1L, binCount);
return null;
}
1）spread() 是一个哈希算法，和 HashMap 的 hash() 方法类似：
static final int spread(int h) {
return (h ^ (h >>> 16)) & HASH_BITS;
}
2）如果是第一次 put 的话，会调用 initTable() 对 table 进行初始化。
private final ConcurrentHashMap.Node[] initTable() {
ConcurrentHashMap.Node[] tab; int sc;
while ((tab = table) == null || tab.length == 0) {
if ((sc = sizeCtl) < 0)
Thread.yield(); // lost initialization race; just spin
else if (U.compareAndSetInt(this, SIZECTL, sc, -1)) {
try {
if ((tab = table) == null || tab.length == 0) {
int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
@SuppressWarnings("unchecked")
ConcurrentHashMap.Node[] nt = (ConcurrentHashMap.Node[])new ConcurrentHashMap.Node[n];
table = tab = nt;
sc = n - (n >>> 2);
}
} finally {
sizeCtl = sc;
}
break;
}
}
return tab;
}
外层用了一个 while 循环，如果发现 sizeCtl 小于 0 的话，就意味着其他线程正在初始化，yield 让出 CPU。
第一次 put 的时候会执行 U.compareAndSetInt(this, SIZECTL, sc, -1)，把 sizeCtl 赋值为 -1，表示当前线程正在初始化。
private static final Unsafe U = Unsafe.getUnsafe();
private static final long SIZECTL
= U.objectFieldOffset(ConcurrentHashMap.class, "sizeCtl");
U 是一个 Unsafe（可以提供硬件级别的原子操作，可以获取某个属性在内存中的位置，也可以修改对象的字段值）对象，compareAndSetInt() 是 Unsafe 的一个本地（native）方法，它就负责把 ConcurrentHashMap 的 sizeCtl 修改为指定的值（-1）。
初始化后的 table 大小为 16（DEFAULT_CAPACITY）。
不是第一次 put 的话，会调用 tabAt() 取出 key 位置（(n - 1) & hash）上的值（f）：
static final ConcurrentHashMap.Node tabAt(ConcurrentHashMap.Node[] tab, int i) {
return (ConcurrentHashMap.Node)U.getReferenceAcquire(tab, ((long)i << ASHIFT) + ABASE);
}
U.getReferenceAcquire() 会调用 Unsafe 的 本地方法 getReferenceVolatile() 获取指定内存中的数据，保证每次拿到的数据都是最新的。
如果 f 为 null，说明 table 中这个位置上是第一次 put 元素，调用 casTabAt() 插入 Node。
static final boolean casTabAt(ConcurrentHashMap.Node[] tab, int i,
ConcurrentHashMap.Node c, ConcurrentHashMap.Node v) {
return U.compareAndSetReference(tab, ((long)i << ASHIFT) + ABASE, c, v);
}
如果 CAS 成功，说明 Node 插入成功，执行 addCount() 方法检查是否需要扩容。
如果失败，说明有其他线程提前插入了 Node，进行下一轮 for 循环继续尝试，俗称自旋。
如果 f 的 hash 为 MOVED（-1），意味着有其他线程正在扩容，执行 helpTransfer() 一起扩容。
否则，把 Node 按链表或者红黑树的方式插入到合适的位置，这个过程是通过 synchronized 块实现的。
synchronized (f) {
if (tabAt(tab, i) == f) {
if (fh >= 0) {
binCount = 1;
for (Node e = f;; ++binCount) {
K ek;
if (e.hash == hash &&
((ek = e.key) == key ||
(ek != null && key.equals(ek)))) {
oldVal = e.val;
if (!onlyIfAbsent)
e.val = value;
break;
}
Node pred = e;
if ((e = e.next) == null) {
pred.next = new Node(hash, key,
value, null);
break;
}
}
}
else if (f instanceof TreeBin) {
Node p;
binCount = 2;
if ((p = ((TreeBin)f).putTreeVal(hash, key,
value)) != null) {
oldVal = p.val;
if (!onlyIfAbsent)
p.val = value;
}
}
}
}
1）插入之前，再次调用 tabAt(tab, i) == f 来判断 f 是否被其他线程修改。
2）如果 fh（f 的哈希值） >= 0，说明 f 是链表的头节点，遍历链表，找到对应的 Node，更新值，否则插入到末尾。
3）如果 f 是红黑树，则按照红黑树的方式插入或者更新节点。
分析完 put() 方法后，再来看 get() 方法：
public V get(Object key) {
ConcurrentHashMap.Node[] tab; ConcurrentHashMap.Node e, p; int n, eh; K ek;
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;
}
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;
}
是不是简单很多？
1）如果哈希值相等（(eh = e.hash) == h），直接返回 table 数组中的元素。
2）如果是红黑树（eh < 0），按照红黑树的方式 find 返回。
3）如果是链表，进行遍历，然后根据 key 获取 value。
最后，来写一个 ConcurrentHashMap 的应用实例吧！
/**
* @author 沉默王二，一枚有趣的程序员
*/
public class ConcurrentHashMapDemo {
public final static int THREAD_POOL_SIZE = 5;
public static void main(String[] args) throws InterruptedException {
Map map = new ConcurrentHashMap<>();
long startTime = System.nanoTime();
ExecutorService crunchifyExServer = Executors.newFixedThreadPool(THREAD_POOL_SIZE);
for (int j = 0; j < THREAD_POOL_SIZE; j++) {
crunchifyExServer.execute(new Runnable() {
@SuppressWarnings("unused")
@Override
public void run() {
for (int i = 0; i < 500000; i++) {
map.put("itwanger"+i, "沉默王二");
}
}
});
}
crunchifyExServer.shutdown();
crunchifyExServer.awaitTermination(Long.MAX_VALUE, TimeUnit.DAYS);
long entTime = System.nanoTime();
long totalTime = (entTime - startTime) / 1000000L;
System.out.println(totalTime + "ms");
}
}
给同学们留一道作业题，感兴趣的话可以尝试下，把 ConcurrentHashMap 换成 SynchronizedMap，比较一下两者性能上的差异，差距还是挺明显的。
公众号：沉默王二（ID：cmower）
CSDN：沉默王二
bilibili：沉默王二
知乎：沉默王二
这是一个有颜值却假装靠才华苟且的程序员，你知道，他的文章风趣幽默，读起来就好像花钱一样爽快。
长按下图二维码关注，你将感受到一个有趣的灵魂，且每篇文章都有干货。
------------------
原创不易，莫要白票，如果觉得有点用的话，请毫不留情地素质三连吧，分享、点赞、在看，我不挑，因为这将是我写作更多优质文章的最强动力。