jdk1.8下的ConcurrentHashMap源码分析

10 篇文章 0 订阅
7 篇文章 0 订阅

1 、构造函数

    1.1空参构造函数

        
/**
* Creates a new, empty map with the default initial table size (16).
*/
public ConcurrentHashMap() {
}

    1.2指定初始化容量的构造函数

public ConcurrentHashMap(int initialCapacity) {
    if (initialCapacity < 0)//不能指定为复数
        throw new IllegalArgumentException();
    int cap = ((initialCapacity >= (MAXIMUM_CAPACITY >>> 1)) ?
               MAXIMUM_CAPACITY :
               tableSizeFor(initialCapacity + (initialCapacity >>> 1) + 1));
//获取传入参数的向上取2的n次方比如传入5,则实际的初始化容量为8,传入为16,实际的容量就是16,如果传入为17,则实际的容量的大小为32.
    this.sizeCtl = cap;
}

    1.3指定的初始化容量和加载因子的构造函数

public ConcurrentHashMap(int initialCapacity, float loadFactor) {
    this(initialCapacity, loadFactor, 1);//调用的其实是三个参数的构造函数,并发等级为1传入1.4
}

    1.4 指定初始化容量和加载因子和并发等级的构造函数

public ConcurrentHashMap(int initialCapacity,float loadFactor, int concurrencyLevel) {
    if (!(loadFactor > 0.0f) || initialCapacity < 0 || concurrencyLevel <= 0)//检验参数必须都>0
        throw new IllegalArgumentException();
    if (initialCapacity < concurrencyLevel)   // 如果初始容量<并发等级,会将初始化初始容量调整为=并发等级,即保证初始化容量是>=并发等级的
        initialCapacity = concurrencyLevel;   // as estimated threads
    long size = (long)(1.0 + (long)initialCapacity / loadFactor);//下面的代码都是为了计算实际的初始化容量cap,必须是向上取2的n次方的大小
    int cap = (size >= (long)MAXIMUM_CAPACITY) ?
        MAXIMUM_CAPACITY : tableSizeFor((int)size);
    this.sizeCtl = cap;//将数量付给sizeCtl
}

2、put方法

/** Implementation for put and putIfAbsent */
final V putVal(K key, V value, boolean onlyIfAbsent) {
    if (key == null || value == null) throw new NullPointerException();//ConcurrentHashMap不支持Key或value为空的元素(抛出空指针)
    int hash = spread(key.hashCode());//计算hash值(根据key的hashCode)   (h ^ (h >>> 16)) & HASH_BITS;
    int binCount = 0;
    for (Node<K,V>[] tab = table;;) {
        Node<K,V> f; int n, i, fh;
        if (tab == null || (n = tab.length) == 0)
            tab = initTable();//如果当前数组还没有初始化,那么就会调用初始化方法对桶进行初始化
        else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {通过数组角标和hash按位与计算出数组的角标,并且该位置上的链表的头节点或者tree的root节点==null的话
            if (casTabAt(tab, i, null,
                         new Node<K,V>(hash, key, value, null)))//通过CAS原子操作将数组[i]上的位置进行赋值操作,如果成功就跳出循环
                break;                   // no lock when adding to empty bin 当table[i]上的值为null的话,就不会加锁,cas插入到table[i]位置
        }
        else if ((fh = f.hash) == MOVED)//如果该位置上的元素的hash值为-1(MOVED),那么说明正在扩容
            tab = helpTransfer(tab, f);//帮助其他线程进行扩容操作
        else {//table[i]的位置上不是null,并且此时没有其他线程进行扩容
            V oldVal = null;//记录一下匹配上key的旧的value值
            //说明接下来应该是操作链表或者红黑树去遍历,我猜应该是CAS实现线程安全比较困难,所以这里也是通过加锁synchronized锁住table[i]所在对象。
            synchronized (f) {//加锁
                if (tabAt(tab, i) == f) {
                    if (fh >= 0) {//fh是f的hash,如果f的hash是>=0的话说明是一个链表,因为TreeBin的hash是-1,如果是一个树的话这里接收的其实不是TreeNode节点,而是一个TreeBin对
                                    //象,TreeBin对象继承了Node<>对象
                        binCount = 1;//
                        for (Node<K,V> e = f;; ++binCount) {//循环链表
                            K ek;
                            if (e.hash == hash &&
                                ((ek = e.key) == key ||
                                 (ek != null && key.equals(ek)))) {//比较hash+比较key.equals()或者直接比较key的值,如果相等,就覆盖旧的
                                oldVal = e.val;//记录旧的值
                                if (!onlyIfAbsent)
                                    e.val = value;//赋值后结束循环
                                break;
                            }
                            Node<K,V> pred = e;//如果不是当前节点,那就将e指向链表中的下一个节点(其实就是个循环遍历比较key的过程)
                            if ((e = e.next) == null) {//如果找到链表最后一个也没找到相同的key,那就再尾部插入一个新的节点,后结束循环
                                pred.next = new Node<K,V>(hash, key,
                                                          value, null);
                                break;
                            }
                        }
                    }
                    else if (f instanceof TreeBin) {//如果f是TreeBin的话,就调用putTreeVal方法循环遍历红黑树,并且修改val或将新节点插入到指定位置
                        Node<K,V> p;
                        binCount = 2;
                        if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,
                                                       value)) != null) {//如果查找到了相同key的树节点,就替换
                            oldVal = p.val;
                            if (!onlyIfAbsent)
                                p.val = value;//替换值
                        }
                    }
                }
            }//注意这里是同步代码块的结尾,那么它的树化过程怎么保证线程安全呢???其实树化方法里面会重新加锁
            if (binCount != 0) {//binCount要么是链表的长度,要不就是当当前是树结构的时候,会将binCount=2
                if (binCount >= TREEIFY_THRESHOLD)//如果链表的长度是》=8,调用树化方法,但是需要注意的是树化方法也不一定会真实的转为树
                    treeifyBin(tab, i);
                if (oldVal != null)
                    return oldVal;//返回旧的节点值
                break;
            }
        }
    }
    addCount(1L, binCount);//可以先理解为hashmap中的count++,其实过程要复杂一些
    return null;
}

3、size()方法

/**
* 这里的size方法不是维护成员变量size,而是类似LongAdder类的数量方式获取
*/
public int size() {
    long n = sumCount();
    return ((n < 0L) ? 0 :
            (n > (long)Integer.MAX_VALUE) ? Integer.MAX_VALUE :
            (int)n);
}
//调用的获取sumCount的方法(其实返回值就是整个CHM中的元素个数)
//这里面的sumCount由两部分组成,一是baseCount(基本个数),二是counterCells[]数组中的元素之和

final long sumCount() {
    CounterCell[] as = counterCells; CounterCell a;
    long sum = baseCount;
    if (as != null) {
        for (int i = 0; i < as.length; ++i) {
            if ((a = as[i]) != null)
                sum += a.value;
        }
    }
    return sum;
}

4、树化方法treeifyBin(Node<K,V>[] tab, int index)

/* ---------------- Conversion from/to TreeBins -------------- */


/**
* Replaces all linked nodes in bin at given index unless table is
* too small, in which case resizes instead.
*/
private final void treeifyBin(Node<K,V>[] tab, int index) {//tab是链表,index是table[i]的i即数组中的位置
    Node<K,V> b; int n, sc;
    if (tab != null) {
        if ((n = tab.length) < MIN_TREEIFY_CAPACITY)//这里会判断是否当前数组的容量是否<64,如果小于64的话,那么会尝试扩容,也就是说,只有数组的长度》=64,并且链表中的第九个元素到来到的话才会树化
            tryPresize(n << 1);//扩容1倍
        else if ((b = tabAt(tab, index)) != null && b.hash >= 0) {//length>=64并且当前桶中不是空,并且桶中第一个元素的hash>=0即不是树结构
            synchronized (b) {//这里树化过程会重新加锁
                if (tabAt(tab, index) == b) {//再判断一次是否有其他线程改变了该桶中的结构
                    TreeNode<K,V> hd = null, tl = null;
                    for (Node<K,V> e = b; e != null; e = e.next) {//循环链表
                        TreeNode<K,V> p =
                            new TreeNode<K,V>(e.hash, e.key, e.val,
                                              null, null);//构建一个树节点
                        if ((p.prev = tl) == null)//构建一个双向链表
                            hd = p;
                        else
                            tl.next = p;
                        tl = p;
                    }
                    setTabAt(tab, index, new TreeBin<K,V>(hd));//将root节点封装成TreeBin对象放回table[i]这个封装的过程会真实的把链表转变为一棵红黑树,然后unsafe的putObjectVolatile方法将TreeBin对象放回数组中
                }
            }
        }
    }
}

5、初始化方法initTable()

//ConcurrentHashMap map = new ConcurrentHashMap();map.put("key","value");就会调用初始化方法
private final Node<K,V>[] initTable() {
    Node<K,V>[] tab; int sc;
    while ((tab = table) == null || tab.length == 0) {
        if ((sc = sizeCtl) < 0)//sizeCtl默认是0,除非其他线程在当前线程之前将sizeCtrl改为了-1,即else if中的逻辑
            Thread.yield(); // lost initialization race; just spin 放弃cpu执行权
        else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {//多线程争取去更改sizeCtl,有点类似有争取锁的过程,只不过区别是这里的逻辑可以避免线程争取锁过程中的阻塞状态
            try {
                if ((tab = table) == null || tab.length == 0) {//更新成功,可以理解为抢占到锁了
                    int n = (sc > 0) ? sc : DEFAULT_CAPACITY;//初始长度为16
                    @SuppressWarnings("unchecked")
                    Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                    table = tab = nt;
                    sc = n - (n >>> 2);//取容量的3/4
                }
            } finally {
                sizeCtl = sc;//改为这为正数后推出循环
            }
            break;
        }
    }
    return tab;
}//执行完该方法之后table的容量为16,sizeCtl为12 = 16*(3/4)

6、计数+1和扩容逻辑 addCount(1L, binCount);

/**
* Adds to count, and if table is too small and not already
* resizing, initiates transfer. If already resizing, helps
* perform transfer if work is available.  Rechecks occupancy
* after a transfer to see if another resize is already needed
* because resizings are lagging additions.
*
* @param x the count to add
* @param check if <0, don't check resize, if <= 1 only check if uncontended
*/
private final void addCount(long x, int check) {
    CounterCell[] as; long b, s;
    if ((as = counterCells) != null ||
        !U.compareAndSwapLong(this, BASECOUNT, b = baseCount, s = b + x)) {
        CounterCell a; long v; int m;
        boolean uncontended = true;
        if (as == null || (m = as.length - 1) < 0 ||
            (a = as[ThreadLocalRandom.getProbe() & m]) == null ||
            !(uncontended =
              U.compareAndSwapLong(a, CELLVALUE, v = a.value, v + x))) {
            fullAddCount(x, uncontended);
            return;
        }
        if (check <= 1)
            return;
        s = sumCount();
    }
    if (check >= 0) {
        Node<K,V>[] tab, nt; int n, sc;
        while (s >= (long)(sc = sizeCtl) && (tab = table) != null &&
               (n = tab.length) < MAXIMUM_CAPACITY) {
            int rs = resizeStamp(n);
            if (sc < 0) {
                if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||
                    sc == rs + MAX_RESIZERS || (nt = nextTable) == null ||
                    transferIndex <= 0)
                    break;
                if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1))
                    transfer(tab, nt);
            }
            else if (U.compareAndSwapInt(this, SIZECTL, sc,
                                         (rs << RESIZE_STAMP_SHIFT) + 2))
                transfer(tab, null);
            s = sumCount();
        }
    }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
ConcurrentHashMap是Java中线程安全的哈希表实现,它提供了比Hashtable和SynchronizedMap更好的并发性能和可伸缩性。 ConcurrentHashMap的实现原理主要包括以下几个方面: 1. 分段锁设计 ConcurrentHashMap将整个Map分成了多个Segment,每个Segment默认是一个长度为16的数组(可以通过构造函数指定长度)。每个Segment维护了一个HashEntry数组和一个Segment锁。在ConcurrentHashMap中,所有的操作都是在Segment的粒度上进行的,不同的Segment之间可以并发操作。 2. Hash表和Hash算法 ConcurrentHashMap的底层数据结构是一个哈希表,每个元素是一个HashEntry,其中包含了key、value和next指针。在ConcurrentHashMap中,使用了一种称为“位运算”的Hash算法,可以在不需要锁的情况下计算出每个元素的位置,从而提高了并发性能。 3. CAS操作 在ConcurrentHashMap中,使用了CAS(Compare And Swap)操作来保证线程安全,而不是像Hashtable和SynchronizedMap一样使用全局锁。通过CAS操作,可以在不加锁的情况下对某个元素进行修改,从而提高了并发性能。 4. Jdk1.8的改进 在Jdk1.8中,ConcurrentHashMap的实现进行了一些改进,主要包括了以下几个方面: - Segment被废弃了,使用Node数组代替; - 使用了红黑树来优化链表的查找性能; - 使用了批量操作来减少CAS操作的次数,提高了并发性能。 总的来说,ConcurrentHashMap的实现原理是通过分段锁、哈希表和CAS操作来保证线程安全,同时利用了JDK1.8的一些改进来提高并发性能。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值