我之见--java多线程 ConcurrentHashMap 源码分析

     jdk 1.5以前,同步的map集合只有Hashtable ,下面我们先来看一下它的常用put方法:

public synchronized V put(K key, V value) {
        if (key == null) {
            throw new NullPointerException("key == null");
        } else if (value == null) {
            throw new NullPointerException("value == null");
        }
        int hash = secondaryHash(key.hashCode());
        HashtableEntry<K, V>[] tab = table;
        int index = hash & (tab.length - 1);
        HashtableEntry<K, V> first = tab[index];
        for (HashtableEntry<K, V> e = first; e != null; e = e.next) {
            if (e.hash == hash && key.equals(e.key)) {
                V oldValue = e.value;
                e.value = value;
                return oldValue;
            }
        }

        // No entry for key is present; create one
        modCount++;
        if (size++ > threshold) {
            rehash();  // Does nothing!!
            tab = doubleCapacity();
            index = hash & (tab.length - 1);
            first = tab[index];

        }
        tab[index] = new HashtableEntry<K, V>(key, value, hash, first);
        return null;
    }
  我们惊奇的发现这里只是简单的对方法加同步:synchronized。这样的方法虽然简单,不过效率上面却是很差:

  1.这里锁着是对象,如果要想获得锁,必须先获得对象,如果能锁住进程就最后的,就像可重入锁一样。

  2.这里锁的粒度比较大,因为它锁住的是整个数组结构。

   jdk 1.5以后,引入新的同步类:ConcurrentHashMap,这里比以前来说是相对比较高效。我们从各个方面对比来看一下:ConcurrentHashMap允许多个修改操作并发进行,其关键在于使用了锁分离技术。它使用了多个锁来控制对hash表的不同部分进行的修改。ConcurrentHashMap内部使用段(Segment)来表示这些不同的部分,每个段其实就是一个小的hash table,它们有自己的锁。只要多个修改操作发生在不同的段上,它们就可以并发进行。有些方法需要跨段,比如size()和containsValue(),它们可能需要锁定整个表而而不仅仅是某个段,这需要按顺序锁定所有段,操作完毕后,又按顺序释放所有段的锁。这里“按顺序”是很重要的,否则极有可能出现死锁,在ConcurrentHashMap内部,段数组是final的,并且其成员变量实际上也是final的,但是,仅仅是将数组声明为final的并不保证数组成员也是final的,这需要实现上的保证。这可以确保不会出现死锁,因为获得锁的顺序是固定的

从这张图里面我们可以清楚看到两个的对比。这里简单的可能大家不一定能理解,下面我们还是从源码上面进一步来了解ConcurrentHashMap的实现,我们来先看一下构造函数:

    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;
        // Find power-of-two sizes best matching arguments
        int sshift = 0;
        int ssize = 1;
        while (ssize < concurrencyLevel) {
            ++sshift;
            ssize <<= 1;
        }
        this.segmentShift = 32 - sshift;
        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;
        while (cap < c)
            cap <<= 1;
        // create segments and 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;
    }
1.这里主要是初始化hashMap的长度大小和锁的大小,concurrencyLevel代表锁的个数,一般用户根据线程访问数来设定,最大为65536,ssize代表比传入的实际大小的最大二进制,比如:实际大小100,那个ssize的值:128,那么segmentMask = 127  segmentShift = 32-7 = 25 下面出现新的数据结构:Segment 类,我们先来看Segment类: 
static final class Segment<K,V> extends ReentrantLock implements Serializable {
        ......................
        transient volatile HashEntry<K,V>[] table;
<span style="white-space:pre">	</span>.................
        Segment(float lf, int threshold, HashEntry<K,V>[] tab) {
            this.loadFactor = lf;
            this.threshold = threshold;
            this.table = tab;
        }

        final V put(K key, int hash, V value, boolean onlyIfAbsent) {
            HashEntry<K,V> node = tryLock() ? null :
                scanAndLockForPut(key, hash, value);
            V oldValue;
            try {
                HashEntry<K,V>[] tab = table;
                int index = (tab.length - 1) & hash;
                HashEntry<K,V> first = entryAt(tab, index);
                for (HashEntry<K,V> e = first;;) {
                    if (e != null) {
                        K k;
                        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 {
                        if (node != null)
                            node.setNext(first);
                        else
                            node = new HashEntry<K,V>(hash, key, value, first);
                        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;
        }............
由于继承ReentrantLock所以Segment据有可重入锁的功能,也就是说这里锁住的是线程,这和一般的hashMap没有什么区别:只是对HashEntry的访问进行了锁着。我们接着上面的代码分析: Segment<K,V>[] ss = (Segment<K,V>[])new Segment<?,?>[ssize];主要根据锁的大小生成 Segment对象,UNSAFE.putOrderedObject(ss, SBASE, s0)这里主要利用java原语进行操作,主要是对ss数组第0个进行赋值。   UNSAFE其实是java对数据最底层的操作,也就是说这样的操作都是原子性,所有没有多线程问题,以后我们也会学习这样的操作,可以实现无阻塞的进程访问。

   接下来我们来看一下:

 public V put(K key, V value) {
        Segment<K,V> s;
        if (value == null)
            throw new NullPointerException();
        int hash = hash(key.hashCode());
        int j = (hash >>> segmentShift) & segmentMask;
        if ((s = (Segment<K,V>)UNSAFE.getObject          // nonvolatile; recheck
             (segments, (j << SSHIFT) + SBASE)) == null) //  in ensureSegment
            s = ensureSegment(j);
        return s.put(key, hash, value, false);
    }
1.先对对象进行hashcode值计算,然后检测还有没有剩余锁,如果有则重新获取锁,下面我们来看一下:
private Segment<K,V> ensureSegment(int k) {
        final Segment<K,V>[] ss = this.segments;
        long u = (k << SSHIFT) + SBASE; // raw offset
        Segment<K,V> seg;
        if ((seg = (Segment<K,V>)UNSAFE.getObjectVolatile(ss, u)) == null) {
            Segment<K,V> proto = ss[0]; // use segment 0 as prototype
            int cap = proto.table.length;
            float lf = proto.loadFactor;
            int threshold = (int)(cap * lf);
            HashEntry<K,V>[] tab = (HashEntry<K,V>[])new HashEntry<?,?>[cap];
            if ((seg = (Segment<K,V>)UNSAFE.getObjectVolatile(ss, u))
                == null) { // recheck
                Segment<K,V> s = new Segment<K,V>(lf, threshold, tab);
                while ((seg = (Segment<K,V>)UNSAFE.getObjectVolatile(ss, u))
                       == null) {
                    if (UNSAFE.compareAndSwapObject(ss, u, null, seg = s))
                        break;
                }
            }
        }
        return seg;
    }
这里主要根据当初设定锁内有多少个Entry!则生成对应的数组:HashEntry<K,V>[] tab = (HashEntry<K,V>[])new HashEntry<?,?>[cap];然后再生成Segment对象同时添加到ss对象数组。我们还是通过图来理解:

   其实这里一把锁锁对两个hashMap的键值。这里一种更小粒度的锁,当多线程访问的时候,第一次hash之后我们先定位锁的位置,如果大家都是不一样的锁也就不会阻塞,下面再定位index的位置。

     我们接着看上面的put方法:

final V put(K key, int hash, V value, boolean onlyIfAbsent) {
            HashEntry<K,V> node = tryLock() ? null :
                scanAndLockForPut(key, hash, value);
            V oldValue;
            try {
                HashEntry<K,V>[] tab = table;
                int index = (tab.length - 1) & hash;
                HashEntry<K,V> first = entryAt(tab, index);
                for (HashEntry<K,V> e = first;;) {
                    if (e != null) {
                        K k;
                        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 {
                        if (node != null)
                            node.setNext(first);
                        else
                            node = new HashEntry<K,V>(hash, key, value, first);
                        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;
        }

1. HashEntry<K,V> node = tryLock() ? null : scanAndLockForPut(key, hash, value);尝试去获取锁,如果获取成功node = null,我们先看一下获取成功的。

2.int index = (tab.length - 1) & hash;这里主要是定位Segment的tab数组的下标值。

3.for循环里面主要是遍历index下面的entry链表,如果发现key值存在,则更新value值,如果没有则测新生成一个node对象,它的next是指向原来的头节点,同时加的到链接前面,这里采用的头插法,上面还有一点是对容量的检测,如果发现容量不够的时候先扩容。

4.上面马上获取锁的情况分析完了,下面我们再分析一下,没有获取锁的时候,scanAndLockForPut方法再去扫描尝试获取锁。

private HashEntry<K,V> scanAndLockForPut(K key, int hash, V value) {
            HashEntry<K,V> first = entryForHash(this, hash);
            HashEntry<K,V> e = first;
            HashEntry<K,V> node = null;
            int retries = -1; // negative while locating node
            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;
        }
1.直接获取链表的头节点: HashEntry<K,V> first = entryForHash(this, hash);

2.while 循环里面尝试获取锁,如果没有获取成功,则继续尝试获取,同时会把尝试次数加1.直到获取成功为 止。



  到这里加锁的原理大概明白,总结一下:ConcurrentHashMap为什么高效的原因:

  1.采用ReentrantLock可重入锁,比一般的synchronized关健字加锁效率要高。

  2. 锁的粒度更小了,相比较原来的hashTable 使用synchronized关健字,实际上面锁的整张hashMap表,ConcurrentHashMap采用锁分离技术,只对Segment进行加        锁,而且Segment本身可以锁定多个entry对象,为什么要锁定多个entry对象,我们个人认为是效率的原因,如果每个entry对象都加一把锁的话,会造成锁的频繁切      换。所以就算是Segment对象锁住多个entry对象,也比锁住整张hashMap表粒度更小,而且Segment也是采用可重入锁的机制,因此对同一个线程频繁的操作也是很高效的,

 3. 多个线程可以同时进行读写操作。什么这样说呢!假如:A线程在进程写入操作,它操作的Segment对象的index号是10 ,这个时候B线程来了,它要读11号的Segment对象,因为是不同的对象,所以也不存在共享的问题,线程绝对是安全的。





评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值