美文网首页
ConcurrentHashMap源码

ConcurrentHashMap源码

作者: 丁木木木木木 | 来源:发表于2017-04-03 11:06 被阅读86次

主要内容:

  • 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;
        }

未完。。。。

相关文章

网友评论

      本文标题:ConcurrentHashMap源码

      本文链接:https://www.haomeiwen.com/subject/qupanttx.html