ConcurrentHashMap源码

主要内容:

  • ConcurrentHashMap数据结构
  • 继承关系、关键属性、构造函数
  • 插入元素

ConcurrentHashMap概述(基于jdk1.7)

ConcurrentHashMap是线程安全的集合,与Hashtable相比ConcurrentHashMap效率高。并发环境下Hashtable效率低下的原因是访问Hashtable的多个线程都竞争同一把锁,当然访问慢了。那我们很容易想到如果有多把锁,每把锁作用于集合的一部分数据,那么访问多线程访问不同数据段的数据时就不会发生锁竞争的情况,那么效率当然就提高了。这其实就是ConcurrentHashMap的锁分离技术

锁分离技术:将数据分成一段一段的存储,每一段数据都有一把锁,当一个线程访问某段数据时,其他段数据可以被其他线程锁访问。

ConcurrentHashMap数据结构

ConcurrentHashMap数据结构.png

底层基于分段的数组和链表实现,由Segment段类型的数组结构和HashEntry类型的数据结构存储数据,将Segement数组声明为final

  • Segment:继承了ReentrantLock,可重入锁。Segment中包含了HashEntry类型的数组。
static final class Segment<K,V> extends ReentrantLock 
    implements Serializable {
        transient volatile HashEntry<K,V>[] table;//HashEntry类型的数组
        transient int count;//Segment中保存的元素数量
        transient int modCount;//段修改的次数
        transient int threshold;//阈值,超过阈值后数组会进行扩容
        final float loadFactor;//负载因子
}
  • HashEntry:链表上的节点,定义了键key、值value、哈希值hash以及指向下一个节点的指针next。这里注意到哈希值、键值是final型的,保证读操作时不需要加锁。值、下一节点指针是volatile型,保证读操作时可以读到最新的值
    static final class HashEntry<K,V> {
        final int hash;
        final K key;
        volatile V value;
        volatile HashEntry<K,V> next;

        HashEntry(int hash, K key, V value, HashEntry<K,V> next) {
            this.hash = hash;
            this.key = key;
            this.value = value;
            this.next = next;
        }

        final void setNext(HashEntry<K,V> n) {
            UNSAFE.putOrderedObject(this, nextOffset, n);
        }

        // Unsafe mechanics
        static final sun.misc.Unsafe UNSAFE;
        static final long nextOffset;
        static {
            try {
                UNSAFE = sun.misc.Unsafe.getUnsafe();
                Class k = HashEntry.class;
                nextOffset = UNSAFE.objectFieldOffset
                    (k.getDeclaredField("next"));
            } catch (Exception e) {
                throw new Error(e);
            }
        }
    }

源码分析

继承关系

Concurrent继承关系
public class ConcurrentHashMap<K, V> extends AbstractMap<K, V>
        implements ConcurrentMap<K, V>, Serializable {
  • 继承AbstractMap抽象类,实现Map接口
  • 实现java.io.Serialization接口,支持序列化
  • 实现ConcurrentMap接口

初始化

    public ConcurrentHashMap(int initialCapacity,
                             float loadFactor, int concurrencyLevel) {
        if (!(loadFactor > 0) || initialCapacity < 0 || concurrencyLevel <= 0)
            throw new IllegalArgumentException();
        if (concurrencyLevel > MAX_SEGMENTS)
            concurrencyLevel = MAX_SEGMENTS;
        //ssize为大于等于concurrencyLevel的最小的2的幂次方
        int sshift = 0;
        int ssize = 1;
        while (ssize < concurrencyLevel) {
            ++sshift;
            ssize <<= 1;
        }
        /**
         * 用于定位Segment,假设ssize为2的n次方,我们可以发现sshift就是n。
         * segmentShift为32-n,segmentMask为2^n-1,二进制表示时各位数都是1
         */
        this.segmentShift = 32 - sshift;//偏移量,定位参与hash运算的位数
        this.segmentMask = ssize - 1;//掩码
        if (initialCapacity > MAXIMUM_CAPACITY)
            initialCapacity = MAXIMUM_CAPACITY;
        int c = initialCapacity / ssize;
        if (c * ssize < initialCapacity)
            ++c;
        int cap = MIN_SEGMENT_TABLE_CAPACITY;//segment中HashEntry数组的长度
        while (cap < c)
            cap <<= 1;
        //创建segments和segments[0]
        Segment<K,V> s0 =
            new Segment<K,V>(loadFactor, (int)(cap * loadFactor),
                             (HashEntry<K,V>[])new HashEntry[cap]);
        Segment<K,V>[] ss = (Segment<K,V>[])new Segment[ssize];
        UNSAFE.putOrderedObject(ss, SBASE, s0); // ordered write of segments[0]
        this.segments = ss;
    }

    //使用指定初始化容量、负载因子和默认并发等级16,创建空ConcurrentHashMap
    public ConcurrentHashMap(int initialCapacity, float loadFactor) {
        this(initialCapacity, loadFactor, DEFAULT_CONCURRENCY_LEVEL);
    }

    //使用指定初始化容量、默认负载因子0.75和默认并发等级16,创建空ConcurrentHashMap
    public ConcurrentHashMap(int initialCapacity) {
        this(initialCapacity, DEFAULT_LOAD_FACTOR, DEFAULT_CONCURRENCY_LEVEL);
    }

    //使用默认初始化容量16、负载因子和默认并发等级16,创建空ConcurrentHashMap
    public ConcurrentHashMap() {
        this(DEFAULT_INITIAL_CAPACITY, DEFAULT_LOAD_FACTOR, DEFAULT_CONCURRENCY_LEVEL);
    }

    public ConcurrentHashMap(Map<? extends K, ? extends V> m) {
        this(Math.max((int) (m.size() / DEFAULT_LOAD_FACTOR) + 1,
                      DEFAULT_INITIAL_CAPACITY),
             DEFAULT_LOAD_FACTOR, DEFAULT_CONCURRENCY_LEVEL);
        putAll(m);
    }

发现构造函数最终调用的都是public ConcurrentHashMap(int initialCapacity, float loadFactor, int concurrencyLevel)这个方法,参数分别是DEFAULT_INITIAL_CAPACITY初始容量、DEFAULT_LOAD_FACTOR默认负载因子以及DEFAULT_CONCURRENCY_LEVEL默认并发级别,表示将ConcurrentHashMap分为几个段。

插入

ConcurrentHashMap用分段Segment来保存数据,所以插入和获取数据时都需要先定位到对应的段上。

     public V put(K key, V value) {
        Segment<K,V> s;
        if (value == null)//value值不能为空
            throw new NullPointerException();
        int hash = hash(key);//求出key的hash值
        /**
         * segmentShift是32-sshift,将计算的hash值无符号右移segmentShift位,其实就是取hash值的高sshift位
         * 然后与segmentMask进行按位与操作
        **/
        int j = (hash >>> segmentShift) & segmentMask;
        if ((s = (Segment<K,V>)UNSAFE.getObject
             (segments, (j << SSHIFT) + SBASE)) == null) //找不到Segment新建一个
            s = ensureSegment(j);
        return s.put(key, hash, value, false);
    }

定位到对应的段上后就需要调用segment的put方法。

    final V put(K key, int hash, V value, boolean onlyIfAbsent) {
            /**
             * tryLock方法尝试去获得锁
             * 获取锁失败的话,调用scanAndLockForPut获得锁
            **/
            HashEntry<K,V> node = tryLock() ? null :
                scanAndLockForPut(key, hash, value);
            V oldValue;
            try {
                HashEntry<K,V>[] tab = table;//当前段的HashEntry类型的数组
                int index = (tab.length - 1) & hash;
                HashEntry<K,V> first = entryAt(tab, index);//定位到对应的桶
                for (HashEntry<K,V> e = first;;) {//遍历HashEntry链表
                    if (e != null) {
                        K k;
                        /**
                         * 要插入的键值对已经存在于HashEntry数组中
                         * onlyIfAbsent为true,表示只能插入不存在的键值对直接退出
                         * onlyIfAbsent为false,用新值替代旧值
                        **/
                        if ((k = e.key) == key ||
                            (e.hash == hash && key.equals(k))) {
                            oldValue = e.value;
                            if (!onlyIfAbsent) {
                                e.value = value;
                                ++modCount;
                            }
                            break;
                        }
                        e = e.next;
                    }
                    else {//要插入的键值对不存在HashEntry数组中                
                        if (node != null)//在scanAndLockForPut获得锁的过程中新建了对应的节点
                            node.setNext(first);//将first节点设置成node的下一个节点
                        else
                            node = new HashEntry<K,V>(hash, key, value, first);//新建HashEntry链表
                        /**
                         * 如果插入键值对后,超过阈值,则进行扩容;
                         * 否则直接插入键值对
                        **/
                        int c = count + 1;
                        if (c > threshold && tab.length < MAXIMUM_CAPACITY)
                            rehash(node);
                        else
                            setEntryAt(tab, index, node);
                        ++modCount;
                        count = c;
                        oldValue = null;
                        break;
                    }
                }
            } finally {
                unlock();//释放锁
            }
            return oldValue;
        }

put方法里首先需要先获得Segment的互斥锁,如果没有获得锁的话会调用scanAndLockForPut这个自旋方法,介绍下这个方法具体做了什么。

        //
        private HashEntry<K,V> scanAndLockForPut(K key, int hash, V value) {
            HashEntry<K,V> first = entryForHash(this, hash);//根据hash值获取对应的HashEntry链表
            HashEntry<K,V> e = first;
            HashEntry<K,V> node = null;
            int retries = -1; //自旋计数器
            while (!tryLock()) {
                HashEntry<K,V> f; // to recheck first below
                if (retries < 0) {
                    if (e == null) {
                        if (node == null) // speculatively create node
                            node = new HashEntry<K,V>(hash, key, value, null);
                        retries = 0;
                    }
                    else if (key.equals(e.key))
                        retries = 0;
                    else
                        e = e.next;
                }
                else if (++retries > MAX_SCAN_RETRIES) {
                    lock();
                    break;
                }
                else if ((retries & 1) == 0 &&
                         (f = entryForHash(this, hash)) != first) {
                    e = first = f; // re-traverse if entry changed
                    retries = -1;
                }
            }
            return node;
        }

未完。。。。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,362评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,330评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,247评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,560评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,580评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,569评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,929评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,587评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,840评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,596评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,678评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,366评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,945评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,929评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,165评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,271评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,403评论 2 342

推荐阅读更多精彩内容