ConcurrentHashMap 是一个线程安全的Map,它 的数据结跟HashMap一样,是由数组和 链表 加黑红树(JDK 1.8 以后)。
ConcurrentHashMap 首先会默认初始化一个Node[]为16的桶,然后根据key 值的hash值 找到对应的下标,然后放入桶中,当当前链长度大于8,数组长度大于64 的时会 把当前链转化为红黑树。
下面 我们来进行源码分析。ConcurrentHashMap 里面定义了很多变量,直接去看变量 可能看会看不懂,我们直接通过入口
put() 去进行源码解析,边解析源码 边了解这个变量具体有什么用?ConcurrentHashMap 又是如何实现线程安全的呢?
PUT
public V put(K key, V value) {
//一共有3个参数,第一个key,第二个Value 第三个参数不管
return putVal(key, value, false);
}
final V putVal(K key, V value, boolean onlyIfAbsent) {
// 判断key 和value 值 是否为null 其中一个为Null 抛出空指针异常
if (key == null || value == null) throw new NullPointerException();
// 根据 key的hash 值获取 放入桶的下标
int hash = spread(key.hashCode());
//用于当前桶中节点的数
int binCount = 0;
for (Node<K,V>[] tab = table;;) {//进行死循环 跳出循环由里面的代码控制
Node<K,V> f; int n, i, fh;
//下面着有4中情况 ,
if (tab == null || (n = tab.length) == 0) {
//第一种 当tab 等于null 的, 表示Node[] 数组 table 为null,
//初始化这个table, 初始化后面在讲解,继续往下走
tab = initTable();
}
//第二种情况,根据key 的hash值 找了数组对应的下标,判断这个桶是否有值
// 这tabAt(tab, i = (n - 1) & hash) ;这个tabAt 里面的方法是
sun.misc.Unsafe.getObjectVolatile 来获取这节点,这个方法获取的值是用Volatile
来修饰,保证是线程安全的
else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
//这个casTabAt 方法里面只有这一句代码
// U.compareAndSwapObject(tab, ((long)i << ASHIFT) + ABASE, c, v)
// 这个地方是用CAS算法,如果有线程 这个桶的头节点赋值了,另外的线程在来赋值就会失败
if (casTabAt(tab, i, null,
new Node<K,V>(hash, key, value, null)))
break; // no lock when adding to empty bin
}
// 这种情况是在这个Node[] table 这个数组在进行扩容,并且这个下标的桶已经处理过了
//MOVED 这个是ConcurrentHashMap 的一个静态变量值为-1。
//这个时候这个线程就会去帮忙对这个table数组进行扩容,
//这个也是ConcurrentHashMap的一个特性,可以多个线程一起扩容
//这个节点的hash为什么为-1,后面在解析扩容的时候在去了解,可以先跳过,
else if ((fh = f.hash) == MOVED){
//线程去帮忙扩容的方法调用
tab = helpTransfer(tab, f);
}
else {//第四种 情况 是put 方法的核心,
V oldVal = null;
//这里用synchronized 这个关键字,保证了其线程安全
synchronized (f) {
// 判断这个f节点是否是下标为i 的桶的头节点
if (tabAt(tab, i) == f) {
// 判断这个桶是否是链表
if (fh >= 0) {
binCount = 1;
// 遍历这个链表
for (Node<K,V> e = f;; ++binCount) {
K ek;
//判断这个节点的key和当前要插入的key是否相等
//相等就更改这个节点的value值
if (e.hash == hash &&
((ek = e.key) == key ||
(ek != null && key.equals(ek)))) {
oldVal = e.val;
if (!onlyIfAbsent)
e.val = value;
break;
}
Node<