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));
}
}
}
}
}
未完待续......