最近看了不少关于HashMap也想自己整理一下。这篇讲jdk1.8里的hashmap但会和jdk1.7的进行对比。主要讲解put方法,和resize方法。
前置知识点
HashMap是一个数组table,table的放的是Node,Node里有一个next的属性,Node是一个链表。但节点长度大于8时会转变成红黑树,但我不讲红黑树,我还没研究到那个。
HashMap有三种构造方式。我们常用的方式可能是HashMap<String,String> map = new HashMap<>();
不显示的设置table的长度,在resize时使用默认的长度创建一个长16的table
不声明长度
public HashMap() {
//仅赋值了loadFactor,下面英文的意思是其他值都默认
this.loadFactor = DEFAULT_LOAD_FACTOR; // all other fields defaulted
}
//很明显这个方法和上面的不传值创建的方法是有点不同的
public HashMap(int initialCapacity, float loadFactor) {
if (initialCapacity < 0)
throw new IllegalArgumentException("Illegal initial capacity: " +
initialCapacity);
if (initialCapacity > MAXIMUM_CAPACITY)
initialCapacity = MAXIMUM_CAPACITY;
if (loadFactor <= 0 || Float.isNaN(loadFactor))
throw new IllegalArgumentException("Illegal load factor: " +
loadFactor);
//在这个构造方法中loadFactor和threshold被赋值了。
this.loadFactor = loadFactor;
this.threshold = tableSizeFor(initialCapacity);
}
//在创建时声明长度,例如HashMap<String,String> map = new HashMap<>(16);
public HashMap(int initialCapacity) {
//调用上面的方法,使用默认的因子
this(initialCapacity, DEFAULT_LOAD_FACTOR);
}
put方法
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
真正的代码
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
Node<K,V>[] tab; Node<K,V> p; int n, i;
// 如果这是第一次调用,也就是table==null,就运行resize生成一个table,默认长度是16
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
//如果这是这个位置第一次添加节点,直接添加
//i=(n-1) & hash是计算位置
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else { //如果找到的这个位置里已经有节点了,我们就需要进入节点中
Node<K,V> e; K k;
//如果第一个节点就是所需要的节点
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
//如果节点是树节点,说明这里放的是红黑树,需要使用专门的方法。
//当链表的长度大于8时会转换为红黑树
else if (p instanceof TreeNode)
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
else {
//到这,说明这个节点是一个链表,开始循环找我们要找的节点
for (int binCount = 0; ; ++binCount) {
//链表到底没找到,说明这是一个要添加的操作,也就是Key没有
if ((e = p.next) == null) {
//添加一个新节点
p.next = newNode(hash, key, value, null);
//这个节点加进入了后,判断一下现在的长度是否超过了TREEIFY_THRESHOLD-1
//TREEIFY_THRESHOLD这是hashmap定义过的属性,值为8
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
treeifyBin(tab, hash);
break;
}
//找到了,说明这是一个修改的操作,key以前被创建过,离开循环
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
//如果e!=null将传进来的value赋值这个节点
if (e != null) { // existing mapping for key
V oldValue = e.value;
//onlyIfAbsent如果是执行put操作的话是false表示可以修改已经存在的值
//如果为true表示不能修改
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
//这是执行了一次在table的某一位置添加一个节点才会执行到的地方
++modCount;
//判断是否需要扩容,threshold=capacity(16)*load_factor(0.75)
if (++size > threshold)
//使用扩容的函数
resize();
afterNodeInsertion(evict);
return null;
}
final Node<K,V>[] resize() {
Node<K,V>[] oldTab = table;
//如果是创建一个table,oldCap=0,否者oldCap=table的长度
int oldCap = (oldTab == null) ? 0 : oldTab.length;
//oldThr=当前阈值,如果这是一个创建table的操作的话且table长度没有被显示赋值
//这里我觉得涉及到hashmap的构造方法,当我们在创建hashmap时传入长度,会对threshold赋值
//如果不传长度一切默认的话此时threshold=0
int oldThr = threshold;
int newCap, newThr = 0;
//如果oldCap>0说明table已经创建过是有长度的
if (oldCap > 0) {
//如果长度大于最大的MAXIMUM_CAPACITY,2的30次。
//直接把阈值赋值成integer的最大值2的31次-1
if (oldCap >= MAXIMUM_CAPACITY) {
threshold = Integer.MAX_VALUE;
return oldTab;
}
//如果不是
else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY &&
oldCap >= DEFAULT_INITIAL_CAPACITY)
//左移一位,可理解成*2,因为左移了一位
newThr = oldThr << 1; // double threshold
}
//threshold已经被赋过值了走这个
else if (oldThr > 0) // initial capacity was placed in threshold
newCap = oldThr;
//我的理解是第一次以默认的形式创建table,应该走这个,oldThr=0
else { // zero initial threshold signifies using defaults
newCap = DEFAULT_INITIAL_CAPACITY;
//下面这个就是16*0.75=12
newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);
}
//这是这个table不是第一次创建,threshold已经有值了
//且使用的可能不是默认值(应该我们可以用默认值16来创建hashmap)
if (newThr == 0) {
//注意下面这句,也是用cap*loadfactor和上面一样,区别是用的不是默认值了
float ft = (float)newCap * loadFactor;
newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ?
(int)ft : Integer.MAX_VALUE);
}
//给threshold赋值
threshold = newThr;
//从下面开始就要进行扩容的操作了,本质上是把oldTable规划起来创建一个新的table
@SuppressWarnings({"rawtypes","unchecked"})
Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];
table = newTab;
//oldTable!=null说明不是第一次创建table
if (oldTab != null) {
//循环这个table数组
for (int j = 0; j < oldCap; ++j) {
Node<K,V> e;
//如果这个节点不为空,执行这个if
if ((e = oldTab[j]) != null) {
oldTab[j] = null;
//如果这个节点的next也就是下一个为null,说明这个链表就一个值
if (e.next == null)
//通过e.hash & (newCap-1)算一下新的位置
newTab[e.hash & (newCap - 1)] = e;
//如果这是一个树节点,说明是红黑树,执行这个if
else if (e instanceof TreeNode)
((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
//执行这个else说明这个一个链表且是有Next的
else { // preserve order
//这个地方本质上是两个节点Node,只不过各分了头尾节点。
//loHead是是当前位置的头节点,hiHead是新的位置的头节点
//接下来的代码做的是对节点进行分割
Node<K,V> loHead = null, loTail = null;
Node<K,V> hiHead = null, hiTail = null;
Node<K,V> next;
do {
next = e.next;
//e.hash & oldCap是用于判断是否需要改变位置也就是进入hiHead新的里面
//如果==0说明不用改变
if ((e.hash & oldCap) == 0) {
//loTail == null 说明这是当前位置要添加的第一个节点,所以用loHeader头部被赋值
if (loTail == null)
loHead = e;
//在当前位置的节点尾部添加
else
loTail.next = e;
loTail = e;
}
//改变的话,就要把这个节点放到新的位置
else {
//和上面的意思差不多
if (hiTail == null)
hiHead = e;
else
hiTail.next = e;
hiTail = e;
}
} while ((e = next) != null);
//判断一下尾节点是否为null
if (loTail != null) {
loTail.next = null;
//把当前的节点赋给当前的位置
newTab[j] = loHead;
}
if (hiTail != null) {
hiTail.next = null;
//把hi那个节点赋给新的位置
newTab[j + oldCap] = hiHead;
}
}
}
}
}
return newTab;
}
1.7出现的线程安全的问题中,有一个死循环的问题是因为1.7的扩容方法中对table的节点再分配是采用头插入法。
而1.8不会出现是因为,采用了头尾节点的方法。当然hashmap本身就是线程不安全的,这里针对1.7的死循环问题。
do {
next = e.next;
if ((e.hash & oldCap) == 0) {
if (loTail == null)
loHead = e;
else
loTail.next = e; //除了第一次给头节点赋值,后面的节点都是添加在尾节点里的,从根本上避开了1.7的头插入死循环
loTail = e;
}
else {
if (hiTail == null)
hiHead = e;
else
hiTail.next = e;
hiTail = e;
}
} while ((e = next) != null);