HashMap是我们最常用的类之一,它实现了hash算法,虽然使用很简单,但是其实现有很多值得研究的地方。
HashMap存储的是key-value形式的键值对,这个键值对在实现中使用一个静态内部类Entry来表示,它存储了key、value、hash值、以及在hash冲突时链表中下一个元素的引用。
HashMap底层实现使用了一个数组来存储元素。它的初始容量默认是16,而且必须容量必须是2的整数次幂,最大容量是1<<30(10.7亿+),同时还使用一个加载因子(load factor)来控制这个map的这个hash表的扩容,默认为0.75,即当容量达到初始容量3/4时会扩容(当然不只这样,后面会说明)。
在往HashMap中添加元素时,会计算key的hashCode,然后基于这个hashCode和数组大小来确定它在数组中的存储位置,当遇到hash冲突时,会以链表的形式存储在数组中。
下面具体看看源码,首先看构造方法
public HashMap(int initialCapacity, float loadFactor) {
// 初始容量不能小于0,否则会抛出异常
if (initialCapacity < 0)
throw new IllegalArgumentException("Illegal initial capacity: " +
initialCapacity);
// 控制初始容量不能大于最大容量1<<30
if (initialCapacity > MAXIMUM_CAPACITY)
initialCapacity = MAXIMUM_CAPACITY;
// 检查加载因子的合法性,不能小于0,且必须是数值
if (loadFactor <= 0 || Float.isNaN(loadFactor))
throw new IllegalArgumentException("Illegal load factor: " +
loadFactor);
this.loadFactor = loadFactor;
threshold = initialCapacity;
// 这个init方法是留给子类扩展
init();
}
可以看到在创建HashMap时,并不分配内存空间,而是在真正往map中添加数据时才会分配,可以从put方法中看到:
public V put(K key, V value) {
// 创建时未分配空间,所以检查如果还是空表的话,就分配内存空间
if (table == EMPTY_TABLE) {
inflateTable(threshold);
}
// 对null的key进行的特殊处理
if (key == null)
return putForNullKey(value);
// 计算key的hashCode
int hash = hash(key);
// 根据hashCode和当前容量来确定元素在hash表中的位置,即hash桶的位置
int i = indexFor(hash, table.length);
// 检查key是否已经存在,如果已经存在,则替换旧值为新值,并返回旧值
for (Entry<K,V> e = table[i]; e != null; e = e.next) {
Object k;
// 这里可以看到是根据hashCode和equals方法来判断一个key是否已经存在
if (e.hash == hash && ((k = e.key) == key || key.equals(k))) {
V oldValue = e.value;
e.value = value;
e.recordAccess(this);
return oldValue;
}
}
// 增加map的修改次数,这用于实现fail-fast机制
modCount++;
// 真正把元素添加到hash表中指定的索引位置处理(也叫hash桶)
addEntry(hash, key, value, i);
// 返回null表示key之前不存在
return null;
}
void addEntry(int hash, K key, V value, int bucketIndex) {
// 判断是否需要扩容,当前容量达到阙值,并且产生了hash冲突(指定hash桶已经有元素存在)
if ((size >= threshold) && (null != table[bucketIndex])) {
// 容量扩展为之前的2倍
resize(2 * table.length);
hash = (null != key) ? hash(key) : 0;
// 重新计算存储的hash桶位置
bucketIndex = indexFor(hash, table.length);
}
// 创建Entry并存储到hash表中
createEntry(hash, key, value, bucketIndex);
}
void createEntry(int hash, K key, V value, int bucketIndex) {
// 取出之前已经存在的元素
Entry<K,V> e = table[bucketIndex];
// 把新元素放到链表的开头,即让新元素的next引用指向之前已经存在的元素
table[bucketIndex] = new Entry<>(hash, key, value, e);
// 修改元素计数
size++;
}
从代码中可以看到, 扩容需要满足以下两个条件:
- 达到加载因子指定的阙值
- put当前值时发生hash冲突(即当前桶的位置已经存在有元素了)
只是当前容器中key value数量超过阙值是不会进行扩容的。就是说,比如初始容量为16,当达到阙值以前发生大量的hash冲突,而后添加的元素又很少发生hash冲突,那么有可能key value的数量超过16*0.75=12甚至超过16都不进行扩容,所以hash算法必须保证分布均匀,尽量减少hash冲突。
上面是添加元素的实现,这里再看看它是如何初始化并分配内存的:
private void inflateTable(int toSize) {
// 保证容量是2的整数次幂
int capacity = roundUpToPowerOf2(toSize);
// 在初始化的时候就把扩容的阙值计算好并保存,避免每次都重新计算
threshold = (int) Math.min(capacity * loadFactor, MAXIMUM_CAPACITY + 1);
// 这里才会真正的分配内存
table = new Entry[capacity];
// 初始化hash种子
initHashSeedAsNeeded(capacity);
}
/**
* 保证容量是2的整数次幂,并且不超过最大容量。
* 比如:传入的是15,值变成16,传入的是17,则会变成32,
* 即大于当前值且与最接近2的整数次幂的数
*/
private static int roundUpToPowerOf2(int number) {
// 保证容量是2的整数次幂,并且不超过最大容量
return number >= MAXIMUM_CAPACITY
? MAXIMUM_CAPACITY
: (number > 1) ? Integer.highestOneBit((number - 1) << 1) : 1;
}
对null key的特殊处理:
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;
}
}
// 增加map的修改次数,这用于实现fail-fast机制
modCount++;
// null key的hashCode固定为0,并且桶的位置也固定为0
addEntry(0, null, value, 0);
return null;
}
再来看如何确定非null key的位置
static int indexFor(int h, int length) {
return h & (length-1);
}
h是key的hashCode,length是当前hash表的最大长度,h & (length-1)与h % length等价,只是前者使用位运算,而位运算比取模运算速度更快。这里为什么可以用&运算代替取模运算呢?因为length是2的整数次幂,而它减1,低位正好全是1,与另一个数进行&运算,结果肯定不会超过length,与%运算的效果一样。如果length不是2的整数次幂,那么是不能这样做的,所以这里运用的非常巧妙。
下面看看最核心的生成hashCode的hash方法:
final int hash(Object k) {
int h = hashSeed;
if (0 != h && k instanceof String) {
return sun.misc.Hashing.stringHash32((String) k);
}
// 调用key的hashCode()方法得到hashCode
h ^= k.hashCode();
// 对hashCode进行一系列的位移与异或运算并把结果作为hashCode返回
h ^= (h >>> 20) ^ (h >>> 12);
return h ^ (h >>> 7) ^ (h >>> 4);
}
这里
为什么要进行这一系列的位移与异或运算呢?主要是经过它这里的运算之后,能够使这个hashCode中的bit 0和1均匀分布,从而减少hash冲突,从而提高整个HashMap的效率。
扩容时的rehash:
void resize(int newCapacity) {
Entry[] oldTable = table;
int oldCapacity = oldTable.length;
if (oldCapacity == MAXIMUM_CAPACITY) {
threshold = Integer.MAX_VALUE;
return;
}
// 重新创建底层数组
Entry[] newTable = new Entry[newCapacity];
// 对已经存在的元素进行重新hash放到新的hash桶中
transfer(newTable, initHashSeedAsNeeded(newCapacity));
table = newTable;
// 更新扩容阙值
threshold = (int)Math.min(newCapacity * loadFactor, MAXIMUM_CAPACITY + 1);
}
void transfer(Entry[] newTable, boolean rehash) {
int newCapacity = newTable.length;
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;
}
}
}
由于hash表长度变化了,所以对于已经存在的元素,需要重新计算hashCode并放到新的hash桶中。这是一个比较耗时的操作,所以在创建HashMap时,如果对数据量有个预期值,那么,应该设置更合适的初始容量,以避免添加数据的过程中不断的扩容造成的性能损失。
下面再来看看get操作
public V get(Object key) {
// null key进行特殊操作
if (key == null)
return getForNullKey();
// 获取key对应的Entry
Entry<K,V> entry = getEntry(key);
// 如果存在则返回key对应的值,不存在则返回null
return null == entry ? null : entry.getValue();
}
final Entry<K,V> getEntry(Object key) {
// size为0表示没有元素,所以直接返回null
if (size == 0) {
return null;
}
// 获取key的hashCode
int hash = (key == null) ? 0 : hash(key);
// 获取key对应的hash桶中的元素,并对链表进行迭代返回相应的value
for (Entry<K,V> e = table[indexFor(hash, table.length)];
e != null;
e = e.next) {
Object k;
// 根据hashCode和equalse()方法来确定key
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
return e;
}
// 如果不存在,返回null
return null;
}
对于加载因子,默认为0.75,这是一个折衷的值, 我们可以通过构造方法来改变这个值,但是需要注意,
加载因子越大,查询数据的开销可能越大。因为加载因子越大,意味着map中存放的元素越多,所以hash冲突的可能性越大,根据hashCode计算出的hash桶的位置相同,则保存为链表,而链表的查询操作会遍历整个链表,所以查询效率不高。而在get和put时都要查询元素,所以提高查询效率就提高了hashmap的效率。这是一种用空间换取时间的策略。
为什么HashMap很高效呢?HashMap通过以下几点保证了它的效率:
- 高效的hash算法,使其不易产生hash冲突
- 基于数组存储,实现了元素的快速存取
- 可通过加载因子,使用空间换取时间