ConcurrentHashMap源码分析

ConcurrentHashMap中重要属性：

 private static final int MAXIMUM_CAPACITY = 1 << 30; //最大空间
 private static final int DEFAULT_CAPACITY = 16;//默认初始化空间
 static final int TREEIFY_THRESHOLD = 8;//链表和红黑树转换空间
 static final int MIN_TREEIFY_CAPACITY = 64;//链表和红黑树转换最小数组空间

重要方法：

	//添加元素
    public V put(K key, V value) {
    	//真正执行添加元素方法
        return putVal(key, value, false);
    }

    final V putVal(K key, V value, boolean onlyIfAbsent) {
       	//key和value都不允许为null，若任一为null，则抛出异常
        if (key == null || value == null) throw new NullPointerException();
        //获取key的hash值
        int hash = spread(key.hashCode());
        int binCount = 0;
        //相当于无限循环
        for (Node<K,V>[] tab = table;;) {
            Node<K,V> f; int n, i, fh;
            //若table为null,获取长度为0，进行初始化操作
            if (tab == null || (n = tab.length) == 0)
                tab = initTable();//初始化table
            //通过保存数据的key的hash与table长度取摸得到数据应存入table中的位置，得到该位置放的元素赋值给f，将脚标赋值给i,原有位置上数据若为空，则新建node对象放入该位置
            else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
            	//新建Node对象，放入table中
                if (casTabAt(tab, i, null,
                             new Node<K,V>(hash, key, value, null)))
                    //跳出循环
                    break;                   // no lock when adding to empty bin
            }
            //若要放入位置已经存在数据，并且存入数据的hash值为-1，表示正在扩容，该线程也加入帮助迁移数据的工作中
            else if ((fh = f.hash) == MOVED)
                tab = helpTransfer(tab, f);
            else {
            	//放入位置存在数据，
                V oldVal = null;
                //对原有数据加锁
                synchronized (f) {
                	//再次对原有数据进行验证
                    if (tabAt(tab, i) == f) {
                    	//fh大于0,为链表，若转化为树之后，则改为-2
                        if (fh >= 0) {
                            binCount = 1;
                            //遍历链表，每循环一次对binCount加1，记录该位置上链表长度
                            for (Node<K,V> e = f;; ++binCount) {
                                K ek;
                                //遍历链表得到node跟要存入数据key进行比较，如果hash相同并且key相等，则替换，并跳出循环
                                if (e.hash == hash &&
                                    ((ek = e.key) == key ||
                                     (ek != null && key.equals(ek)))) {
                                    oldVal = e.val;
                                    if (!onlyIfAbsent)
                                        e.val = value;
                                    break;
                                }
                                //将正在遍历的node对象赋值给pred
                                Node<K,V> pred = e;
                                //将正在遍历的node对象的next对象赋值给e，并判断是否为空，若为空，则创建新的node对象，并保存在该node对象尾部，跳出循环
                                if ((e = e.next) == null) {
                                    pred.next = new Node<K,V>(hash, key,
                                                              value, null);
                                    break;
                                }
                            }
                        }
                        //f已经是红黑树数据类型
                        else if (f instanceof TreeBin) {
                            Node<K,V> p;
                            binCount = 2;
                            //通过putTreeVal将元素添加到红黑树中
                            if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,
                                                           value)) != null) {
                                oldVal = p.val;
                                if (!onlyIfAbsent)
                                    p.val = value;
                            }
                        }
                    }
                }
                //对该位置上链表长度计数器进行判断，若链表长度大于等于8，则将链表转为红黑树类型
                if (binCount != 0) {
                    if (binCount >= TREEIFY_THRESHOLD)
                        treeifyBin(tab, i);
                    if (oldVal != null)
                        return oldVal;
                    break;
                }
            }
        }
        //对binCount加1
        addCount(1L, binCount);
        return null;
    }

	//初始化table
    private final Node<K,V>[] initTable() {
        Node<K,V>[] tab; int sc;
        //若table为空，或者table长度为0则一直处于循环
        while ((tab = table) == null || tab.length == 0) {
            //将sc赋值为sizeCtl，若小于0,则表示有线程正在执行初始化操作
            if ((sc = sizeCtl) < 0)
                //进入该判断的线程，交出cpu资源，重新回到可运行状态，竞争cpu资源
                Thread.yield(); // lost initialization race; just spin
            //通过无锁方式对sizeCtl进行赋值，赋值为-1,标识有线程正在进行table初始化操作
            else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
                try {
                	//再次确认table是否处于未初始化状态
                    if ((tab = table) == null || tab.length == 0) {
                    	//若sc大于0，表示用户指定了初始化长度，则返回指定长度，否则返回默认大小16
                        int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
                        //通过指定长度，新建Node数组，对table进行初始化
                        @SuppressWarnings("unchecked")
                        Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                        table = tab = nt;
                        sc = n - (n >>> 2);
                    }
                } finally {
                    //对sizeCtl进行赋值
                    sizeCtl = sc;
                }
                break;
            }
        }
        return tab;
    }

	//table容量小于64对table进行扩容到一倍，大于64，对节点进行转换为红黑树
    private final void treeifyBin(Node<K,V>[] tab, int index) {
        Node<K,V> b; int n, sc;
        if (tab != null) {
        	//table长度小于64，进行扩容为长度的一倍
            if ((n = tab.length) < MIN_TREEIFY_CAPACITY)
                //扩容为table长度的一倍
                tryPresize(n << 1);
            //table长度大于或等于64，table指定索引上的node元素赋值给b，b != null && b.hash >= 0
            else if ((b = tabAt(tab, index)) != null && b.hash >= 0) {
            	//同步代码块，将b作为锁
                synchronized (b) {
                	//再次验证指定索引中的元素和b是否相等
                    if (tabAt(tab, index) == b) {
                    	//hd树的头节点
                        TreeNode<K,V> hd = null, tl = null;
                        //循环初始变量e等于b，循环条件是e不等于null，对链表中的元素进行迭代，每次循环都是去next节点
                        for (Node<K,V> e = b; e != null; e = e.next) {
                        	//对链表中的每一个元素，重新创建TreeNode对象
                            TreeNode<K,V> p =
                                new TreeNode<K,V>(e.hash, e.key, e.val,
                                                  null, null);
                            //如果tl为null
                            if ((p.prev = tl) == null)
                                hd = p;
                            else
                                tl.next = p;
                            tl = p;
                        }
                        //把treeNode链表放入TreeBin中
                        setTabAt(tab, index, new TreeBin<K,V>(hd));
                    }
                }
            }
        }
    }

未完待续......