深度剖析ConcurrentHashMap源码

概述

你可能会在一些技术类的书籍上看到下面这样一段关于HahsMap和Hashtable的表述：

HashMap是非线程安全的，Hashtable是线程安全的。

不知道大家有什么反应，我当时只是记住了，知道面试的时候能回答上来就行了…至于为什么是线程安全的，内部怎么实现的，却不怎么了解。

今天我们将深入剖析一个比Hashtable性能更优的线程安全的Map类，它就是ConcurrentHashMap，本文基于Java 7的源码做剖析。

ConcurrentHashMap的目的

多线程环境下，使用HashMap进行put操作会引起死循环，导致CPU利用率接近100%。可以通过下面的例子可以得到验证：

    public static void main(String[] args) {
        final HashMap<String, String> map = new HashMap<String, String>(2);
        Thread t = new Thread(new Runnable() {
            @Override
            public void run() {
                for (int i = 0; i < 10000; i++) {
                    new Thread(new Runnable() {
                        @Override
                        public void run() {
                            map.put(UUID.randomUUID().toString(), "");
                        }
                    }, "ftf" + i).start();
                }
            }
        }, "ftf");
        t.start();
        try {
            t.join();
            System.out.println("map.size="+map.size());
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

初始容量为2，启动10000个线程同时执行put操作，结果大概率发生死锁。 运行运行结果如下：

所以在并发情况下不能使用HashMap。虽然已经有一个线程安全的Hashtable，但是Hashtable容器使用synchronized（他的get和put方法的实现代码如下）来保证线程安全，在线程竞争激烈的情况下Hashtable的效率非常低下。因为当一个线程访问Hashtable的同步方法时，访问其他同步方法的线程就可能会进入阻塞或者轮训状态。如线程1使用put进行添加元素，线程2不但不能使用put方法添加元素，并且也不能使用get方法来获取元素，所以竞争越激烈效率越低。

public synchronized V get(Object key) {
    Entry<?,?> tab[] = table;
    int hash = key.hashCode();
    int index = (hash & 0x7FFFFFFF) % tab.length;
    for (Entry<?,?> e = tab[index] ; e != null ; e = e.next) {
        if ((e.hash == hash) && e.key.equals(key)) {
            return (V)e.value;
        }
    }
    return null;
}
public synchronized V put(K key, V value) {
    // Make sure the value is not null
    if (value == null) {
        throw new NullPointerException();
    }

    // Makes sure the key is not already in the hashtable.
    Entry<?,?> tab[] = table;
    int hash = key.hashCode();
    int index = (hash & 0x7FFFFFFF) % tab.length;
    @SuppressWarnings("unchecked")
    Entry<K,V> entry = (Entry<K,V>)tab[index];
    for(; entry != null ; entry = entry.next) {
        if ((entry.hash == hash) && entry.key.equals(key)) {
            V old = entry.value;
            entry.value = value;
            return old;
        }
    }

    addEntry(hash, key, value, index);
    return null;
}

在这么恶劣的环境下，ConcurrentHashMap应运而生。

实现原理

与Hashtable的锁机制不同，ConcurrentHashMap使用分段锁技术，将数据分成一段一段的存储，然后给每一段数据配一把锁，当一个线程占用锁访问其中一个段数据的时候，其他段的数据也能被其他线程访问，能够实现真正的并发访问。如下图是ConcurrentHashMap的内部结构图：