public class HashMap<K, V> {
/**
* HashMap的主干数组
*/
transient HashMap.Entry<K, V>[] table;
/**
* 实际存储的key-value键值对的个数
*/
transient int size;
/**
* 阈值
*/
int threshold;
/**
* 负载因子,代表了table的填充度有多少,默认是0.75
*/
final float loadFactor;
/**
* 用于快速失败
*/
transient int modCount;
}
public class HashMap<K, V> {
//......
public HashMap(){
this(16, 0.75F);
}
public HashMap(int initialCapacity){
this(initialCapacity, 0.75F);
}
public HashMap(int initialCapacity, float loadFactor){
//一堆校验后调用
this.init();
}
public HashMap(Map<? extends K, ? extends V> map){
this(Math.max((int) (m.size() / DEFAULT_LOAD_FACTOR) + 1,
DEFAULT_INITIAL_CAPACITY), DEFAULT_LOAD_FACTOR);
inflateTable(threshold);
putAllForCreate(m);
}
void init() {
}
}
public V put(K key, V value) {
if (table == EMPTY_TABLE) {
/**
* 如果当前的table没有被分配block则inflate table
*/
inflateTable(threshold);
}
/**
* 划重点,key可以为空
*/
if (key == null)
return putForNullKey(value);
int hash = hash(key);
int i = indexFor(hash, table.length);
/**
* 如果该对应数据已存在,执行覆盖操作。用新value替换旧value,并返回旧value
*/
for (Entry<K,V> e = table[i]; e != null; e = e.next) {
Object k;
if (e.hash == hash && ((k = e.key) == key || key.equals(k))) {
V oldValue = e.value;
e.value = value;
e.recordAccess(this);
return oldValue;
}
}
modCount++;
addEntry(hash, key, value, i);
return null;
}
/**
* 核心是初始化物理存储位置
*/
private void inflateTable(int toSize) {
/**
* Find a power of 2 >= toSize 意思是说找到一个大于等于toSize的2的幂次
*/
int capacity = roundUpToPowerOf2(toSize);
/**
* 此处为threshold赋值,取capacity*loadFactor和MAXIMUM_CAPACITY+1的最小值,capaticy一定不会超过MAXIMUM_CAPACITY,除非loadFactor大于1
*/
threshold = (int) Math.min(capacity * loadFactor, MAXIMUM_CAPACITY + 1);
table = new Entry[capacity];
initHashSeedAsNeeded(capacity);
}
private static int roundUpToPowerOf2(int number) {
// 保证number在[1,MAXIMUM_CAPACITY]区间
return number >= MAXIMUM_CAPACITY
? MAXIMUM_CAPACITY
: (number > 1) ? Integer.highestOneBit((number - 1) << 1) : 1;
}
private V putForNullKey(V value) {
for (Entry<K,V> e = table[0]; e != null; e = e.next) {
if (e.key == null) {
V oldValue = e.value;
e.value = value;
e.recordAccess(this);
return oldValue;
}
}
modCount++;
addEntry(0, null, value, 0);
return null;
}
final int hash(Object k) {
int h = hashSeed;
if (0 != h && k instanceof String) {
return sun.misc.Hashing.stringHash32((String) k);
}
h ^= k.hashCode();
/**
* 对k.hashCode()进一步调整,最终使尽可能均匀分布
*/
h ^= (h >>> 20) ^ (h >>> 12);
return h ^ (h >>> 7) ^ (h >>> 4);
}
static int indexFor(int h, int length) {
return h & (length-1);
}
void addEntry(int hash, K key, V value, int bucketIndex) {
/***
* size>=当前阀值,并且对于的table[bucketIndex]位置不为空,则扩容为当前table.length的两倍
* 关于扩容暂且跳过,等会细聊
*/
if ((size >= threshold) && (null != table[bucketIndex])) {
resize(2 * table.length);
//扩容后重新计算hash
hash = (null != key) ? hash(key) : 0;
//扩容后重新计算bucketIndex
bucketIndex = indexFor(hash, table.length);
}
//创建一个新的条目
createEntry(hash, key, value, bucketIndex);
}
/**
* 创建一个新的Entry,并且Entry中的next指向当前table[bucketIndex]上的值,将新创建的条目作为当前table[bucketIndex]
*/
void createEntry(int hash, K key, V value, int bucketIndex) {
Entry<K,V> e = table[bucketIndex];
table[bucketIndex] = new Entry<>(hash, key, value, e);
size++;
}
/**
* 扩容容量是当前table.length的两倍
*/
void resize(int newCapacity) {
Entry[] oldTable = table;
int oldCapacity = oldTable.length;
if (oldCapacity == MAXIMUM_CAPACITY) {
threshold = Integer.MAX_VALUE;
return;
}
//新建一个容量为newCapacity的bucket
Entry[] newTable = new Entry[newCapacity];
transfer(newTable, initHashSeedAsNeeded(newCapacity));
table = newTable;
//计算新的阀值
threshold = (int)Math.min(newCapacity * loadFactor, MAXIMUM_CAPACITY + 1);
}
/**
* 将table上的数据转移到newTable上
*/
void transfer(Entry[] newTable, boolean rehash) {
int newCapacity = newTable.length;
//循环table桶上所有链条上的条目,重新hash、indexFor
for (Entry<K,V> e : table) {
while(null != e) {
Entry<K,V> next = e.next;
if (rehash) {
e.hash = null == e.key ? 0 : hash(e.key);
}
int i = indexFor(e.hash, newCapacity);
e.next = newTable[i];
newTable[i] = e;
e = next;
}
}
}
public V get(Object key) {
if (key == null)
return getForNullKey();
Entry<K,V> entry = getEntry(key);
return null == entry ? null : entry.getValue();
}
/**
* 获取key为null的值
*/
private V getForNullKey() {
/**
* 避免table没有被初始化造成数组越界问题
*/
if (size == 0) {
return null;
}
/**
* 在table[0]上寻找value
*/
for (Entry<K,V> e = table[0]; e != null; e = e.next) {
if (e.key == null)
return e.value;
}
return null;
}
final Entry<K,V> getEntry(Object key) {
/**
* 避免table没有被初始化造成数组越界问题
*/
if (size == 0) {
return null;
}
/**
* 计算key的hash值
*/
int hash = (key == null) ? 0 : hash(key);
/**
* bucketIndex = indexFor(hash, table.length)
* 循环遍历table[bucketIndex]上的键值对
*/
for (Entry<K,V> e = table[indexFor(hash, table.length)];
e != null;
e = e.next) {
Object k;
/**
* 如果键值对的key和传入key的hash值相同并且equals
* 返回对应的值
*/
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
return e;
}
return null;
}
/***
* 移除hashmap中指定的key
*/
public V remove(Object key) {
Entry<K,V> e = removeEntryForKey(key);
return (e == null ? null : e.value);
}
/**
* 寻找Entry,并移除
*/
final Entry<K,V> removeEntryForKey(Object key) {
if (size == 0) {
return null;
}
/**
* 计算hash
*/
int hash = (key == null) ? 0 : hash(key);
/**
* 计算主干位置
*/
int i = indexFor(hash, table.length);
Entry<K,V> prev = table[i];
Entry<K,V> e = prev;
/**
* 移除链表上的Entry
*/
while (e != null) {
Entry<K,V> next = e.next;
Object k;
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k)))) {
modCount++;
size--;
if (prev == e)
table[i] = next;
else
prev.next = next;
e.recordRemoval(this);
return e;
}
prev = e;
e = next;
}
return e;
}
/***
* 树状节点
*/
static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {
TreeNode<K,V> parent; // red-black tree links
TreeNode<K,V> left;
TreeNode<K,V> right;
TreeNode<K,V> prev; // needed to unlink next upon deletion
boolean red;
}
/**
* 链表结构
*/
static class Node<K,V> implements Map.Entry<K,V> {
final int hash;
final K key;
V value;
Node<K,V> next;
}
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如果没有被初始化,则进行初始化
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
//如果没有发生碰撞,则直接赋值
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else {
Node<K,V> e; K k;
//如果冲突位置上的key和传入的key一样则新的节点
if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k))))
e = p;
//如果冲突位置上的节点是红黑树节点,到红黑树种执行节点插入
else if (p instanceof TreeNode)
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
//如果冲突位置上的节点是红黑树节点,插入红黑树中
else {
for (int binCount = 0; ; ++binCount) {
if ((e = p.next) == null) {
p.next = newNode(hash, key, value, null);
//如果链表的长度大于8是则将链表转换成红黑树
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
treeifyBin(tab, hash);
break;
}
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
//如果节点的key在链表中已存在时,e不为空。需要进行value替换
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}