看完这篇ConcurrentHashMap源码解析,我又觉得能手撕面试官了(中)

简介: 看完这篇ConcurrentHashMap源码解析,我又觉得能手撕面试官了(中)

3 构造方法

3.1 无参

  • 使用默认的初始表大小(16)创建一个新的空map
  • image.png

3.2 有参

  • 创建一个新的空map,其初始表大小可容纳指定数量的元素,而无需动态调整大小。
  • image.png
  • -创建一个与给定map具有相同映射的新map
  • image.png
  • 注意 sizeCtl 会暂先维护一个2的幂次方的值的容量.

实例化ConcurrentHashMap时带参数时,会根据参数调整table的大小,假设参数为100,最终会调整成256,确保table的大小总是2的幂次方

tableSizeFor

  • 对于给定的所需容量,返回2的幂的表大小
  • image.png

table 的延迟初始化

ConcurrentHashMap在构造函数中只会初始化sizeCtl值,并不会直接初始化table,而是延缓到第一次put操作table初始化.但put是可以并发执行的,是如何保证 table 只初始化一次呢?

private final Node<K,V>[] initTable() {
    Node<K,V>[] tab; int sc;
    // 进入自旋
    while ((tab = table) == null || tab.length == 0) {
        // 若某线程发现sizeCtl<0,意味着其他线程正在初始化,当前线程让出CPU时间片
        if ((sc = sizeCtl) < 0) 
            Thread.yield(); // 失去初始化的竞争机会; 直接自旋
        else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
            try {
                // 有可能执行至此时,table 已经非空,所以做双重检验
                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;
}

执行第一次put操作的线程会执行Unsafe.compareAndSwapInt方法修改sizeCtl为-1,有且只有一个线程能够修改成功,而其它线程只能通过Thread.yield()让出CPU时间片等待table初始化完成。

4 put

table已经初始化完成,put操作采用CAS+synchronized实现并发插入或更新操作.

final V putVal(K key, V value, boolean onlyIfAbsent) {
    if (key == null || value == null) throw new NullPointerException();
    // 计算hash
    int hash = spread(key.hashCode());
    int binCount = 0;
    // 自旋保证可以新增成功
    for (Node<K,V>[] tab = table;;) {
        Node<K,V> f; int n, i, fh;
        // step1. table 为 null或空时进行初始化
        if (tab == null || (n = tab.length) == 0)
            tab = initTable();
        // step 2. 若当前数组索引无值,直接创建
        else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
            // CAS 在索引 i 处创建新的节点,当索引 i 为 null 时,即能创建成功,结束循环,否则继续自旋
            if (casTabAt(tab, i, null,
                         new Node<K,V>(hash, key, value, null)))
                break;                   // no lock when adding to empty bin
        }
        // step3. 若当前桶为转移节点,表明该桶的点正在扩容,一直等待扩容完成
        else if ((fh = f.hash) == MOVED)
            tab = helpTransfer(tab, f);
        // step4. 当前索引位置有值
        else {
            V oldVal = null;
            // 锁定当前槽点,保证只会有一个线程能对槽点进行修改
            synchronized (f) {
                // 这里再次判断 i 位置数据有无被修改
                // binCount 被赋值,说明走到了修改表的过程
                if (tabAt(tab, i) == f) {
                    // 链表
                    if (fh >= 0) {
                        binCount = 1;
                        for (Node<K,V> 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<K,V> pred = e;
                            // 将新增的元素赋值到链表的最后,退出自旋
                            if ((e = e.next) == null) {
                                pred.next = new Node<K,V>(hash, key,
                                                          value, null);
                                break;
                            }
                        }
                    }
                    // 红黑树,这里没有使用 TreeNode,使用的是 TreeBin,TreeNode 只是红黑树的一个节点
                    // TreeBin 持有红黑树的引用,并且会对其加锁,保证其操作的线程安全
                    else if (f instanceof TreeBin) {
                        Node<K,V> p;
                        binCount = 2;
                        // 满足if的话,把老的值给oldVal
                        // 在putTreeVal方法里面,在给红黑树重新着色旋转的时候
                        // 会锁住红黑树的根节点
                        if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,
                                                       value)) != null) {
                            oldVal = p.val;
                            if (!onlyIfAbsent)
                                p.val = value;
                        }
                    }
                }
            }
            // binCount不为空,并且 oldVal 有值的情况,说明已新增成功
            if (binCount != 0) {
                // 链表是否需要转化成红黑树
                if (binCount >= TREEIFY_THRESHOLD)
                    treeifyBin(tab, i);
                if (oldVal != null)
                    return oldVal;
                // 槽点已经上锁,只有在红黑树或者链表新增失败的时候
                // 才会走到这里,这两者新增都是自旋的,几乎不会失败
                break;
            }
        }
    }
    // step5. check 容器是否需要扩容,如果需要去扩容,调用 transfer 方法扩容
    // 如果已经在扩容中了,check有无完成
    addCount(1L, binCount);
    return null;
}

4.2 执行流程

  1. 若数组空,则初始化,完成之后,转2
  2. 计算当前桶位是否有值
  • 无,则 CAS 创建,失败后继续自旋,直到成功
  • 有,转3
  1. 判断桶位是否为转移节点(扩容ing)
  • 是,则一直自旋等待扩容完成,之后再新增
  • 否,转4
  1. 桶位有值,对当前桶位加synchronize锁
  • 链表,新增节点到链尾
  • 红黑树,红黑树版方法新增
  1. 新增完成之后,检验是否需要扩容

通过自旋 + CAS + synchronize 锁三板斧的实现很巧妙,给我们设计并发代码提供了最佳实践!

目录
相关文章
|
3天前
|
前端开发 JavaScript
Web前端开发之面试题全解析 一,2024年最新面经牛客
Web前端开发之面试题全解析 一,2024年最新面经牛客
|
3天前
|
移动开发 前端开发 JavaScript
Web前端开发之面试题全解析 一(3),前端面试题背不下来怎么办
Web前端开发之面试题全解析 一(3),前端面试题背不下来怎么办
|
3天前
|
前端开发 算法 搜索推荐
Web前端开发之面试题全解析 一(1),2024年最新前端组件化面试
Web前端开发之面试题全解析 一(1),2024年最新前端组件化面试
|
3天前
|
Linux 网络安全
CentOS系统openssh-9,网络安全大厂面试真题解析大全
CentOS系统openssh-9,网络安全大厂面试真题解析大全
|
3天前
|
Linux 网络安全 Windows
网络安全笔记-day8,DHCP部署_dhcp搭建部署,源码解析
网络安全笔记-day8,DHCP部署_dhcp搭建部署,源码解析
|
3天前
|
Android开发
Android高级开发面试题以及笞案整理,实战解析
Android高级开发面试题以及笞案整理,实战解析
|
3天前
|
Android开发
71,字节跳动历年校招Android面试真题解析
71,字节跳动历年校招Android面试真题解析
|
4天前
HuggingFace Tranformers 源码解析(4)
HuggingFace Tranformers 源码解析
6 0
|
4天前
HuggingFace Tranformers 源码解析(3)
HuggingFace Tranformers 源码解析
7 0
|
4天前
|
开发工具 git
HuggingFace Tranformers 源码解析(2)
HuggingFace Tranformers 源码解析
7 0

推荐镜像

更多