Java中的HashMap是最常用的数据结构之一,在实际开发中起着至关重要的作用。本文将详细探讨HashMap的工作原理、源码分析、线程安全问题、以及扩容机制等方面。
一、HashMap的基本概念
HashMap是Java集合框架中的一个类,提供了基于哈希表的数据结构。它允许存储键值对,并通过键快速检索对应的值。HashMap允许键和值为null,并且不保证映射的顺序。
二、HashMap的工作原理
HashMap通过哈希函数将键映射到桶(bucket)数组中的一个位置,以实现快速查找。基本操作如put和get的时间复杂度为O(1)。
1. 哈希函数
HashMap使用键的hashCode()方法计算哈希值,然后通过取模运算(hash % array.length
)将哈希值映射到数组的索引位置。例如:
int hash = key.hashCode();
int index = (array.length - 1) & hash;
2. 处理哈希冲突
当两个不同的键有相同的哈希值时,会发生哈希冲突。HashMap使用链地址法(separate chaining)处理冲突,即每个桶存储一个链表或红黑树。当链表长度超过阈值(默认为8)时,链表转换为红黑树,以提高查询效率。
三、源码分析
以下是HashMap的核心代码段,包含put方法和get方法。
1. 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;
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;
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);
if (binCount >= TREEIFY_THRESHOLD - 1)
treeifyBin(tab, hash);
break;
}
if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
if (e != null) {
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}
2. get方法
public V get(Object key) {
Node<K,V> e;
return (e = getNode(hash(key), key)) == null ? null : e.value;
}
final Node<K,V> getNode(int hash, Object key) {
Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
if ((tab = table) != null && (n = tab.length) > 0 && (first = tab[(n - 1) & hash]) != null) {
if (first.hash == hash && ((k = first.key) == key || (key != null && key.equals(k))))
return first;
if ((e = first.next) != null) {
if (first instanceof TreeNode)
return ((TreeNode<K,V>)first).getTreeNode(hash, key);
do {
if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k))))
return e;
} while ((e = e.next) != null);
}
}
return null;
}
3. 线程不安全的原因
上述put和get方法在多线程环境中是不安全的。具体原因如下:
put方法线程不安全分析
-
扩容(resize):当HashMap需要扩容时,可能会导致多个线程同时进行扩容操作。这会导致数据丢失和不一致。
if (++size > threshold) resize();
-
插入节点(newNode):插入节点时,多个线程可能会同时访问同一个桶位置,导致链表或树结构损坏。
if ((p = tab[i = (n - 1) & hash]) == null) tab[i] = newNode(hash, key, value, null);
-
链表操作:在处理哈希冲突时,链表或红黑树的插入操作不是原子的,可能会导致链表结构损坏。
for (int binCount = 0; ; ++binCount) { if ((e = p.next) == null) { p.next = newNode(hash, key, value, null); if (binCount >= TREEIFY_THRESHOLD - 1) treeifyBin(tab, hash); break; } if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) break; p = e; }
get方法线程不安全分析
-
读取不一致:在读取节点时,如果另一个线程正在进行插入或删除操作,可能会导致读取的数据不一致。
if ((tab = table) != null && (n = tab.length) > 0 && (first = tab[(n - 1) & hash]) != null) { if (first.hash == hash && ((k = first.key) == key || (key != null && key.equals(k)))) return first; if ((e = first.next) != null) { if (first instanceof TreeNode) return ((TreeNode<K,V>)first).getTreeNode(hash, key); do { if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) return e; } while ((e = e.next) != null); } }
由于这些原因,HashMap在多线程环境中使用时可能会导致不可预测的问题。因此,在多线程环境中,建议使用ConcurrentHashMap
替代HashMap
。
四、线程安全的解决方案
1. 使用ConcurrentHashMap
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class ConcurrentHashMapExample {
private static final ConcurrentHashMap<Integer, String> map = new ConcurrentHashMap<>();
public static void main(String[] args) {
ExecutorService executorService = Executors.newFixedThreadPool(10);
// 使用多个线程并发访问和修改ConcurrentHashMap
for (int i = 0; i < 100; i++) {
final int key = i;
executorService.execute(() -> map.put(key, "Value" + key));
}
// 读取ConcurrentHashMap中的数据
executorService.execute(() -> {
for (int i = 0; i < 100; i++) {
System.out.println("Key: " + i + ", Value: " + map.get(i));
}
});
executorService.shutdown();
}
}
五、HashMap的初始值设置
在实际开发中,合理设置HashMap的初始容量和负载因子可以提高性能,减少扩容次数。
1. 初始容量
初始容量是HashMap创建时桶数组的大小,默认值为16。初始容量应根据预期的元素数量和负载因子计算:
int initialCapacity = (int) (expectedSize / loadFactor) + 1;
例如,如果预期有100个元素,负载因子为0.75:
int initialCapacity = (int) (100 / 0.75) + 1; // 约等于134
2. 负载因子
负载因子是HashMap在扩容之前允许的最大填充比例,默认值为0.75。负载因子越小,HashMap的空间利用率越低,但查找效率更高。一般情况下,使用默认负载因子即可。
Map<Integer, String> map = new HashMap<>(initialCapacity, 0.75f);
合理设置初始容量和负载因子,可以避免频繁扩容,提高性能。在不确定具体情况时,默认值通常是一个好的选择。
六、HashMap家族中的其他实现
在Java中,除了HashMap,还有其他几个基于哈希表的数据结构实现,它们各自有不同的特点和用途。
1. LinkedHashMap
LinkedHashMap继承自HashMap,并且保留了插入顺序。它使用一个双向链表来维护插入顺序,可以用于需要保持元素顺序的场景。
Map<Integer, String> linkedHashMap = new LinkedHashMap<>();
linkedHashMap.put(1, "one");
linkedHashMap.put(2, "two");
linkedHashMap.put(3, "three");
System.out.println(linkedHashMap); // 输出顺序为1, 2, 3
2. ConcurrentHashMap
ConcurrentHashMap是一个线程安全的HashMap实现,使用了分段锁(segment locking)机制来提高并发性能。适用于高并发场景。
Map<Integer, String> concurrentHashMap = new ConcurrentHashMap<>();
concurrentHashMap.put(1, "one");
concurrentHashMap.put(2, "two");
concurrentHashMap.put(3, "three");
System.out.println(concurrentHashMap);
3. WeakHashMap
WeakHashMap是一种使用弱引用(weak reference)的哈希表实现。其键在没有其他强引用时可以被垃圾回收器回收。适用于缓存和内存敏感的场景。
Map<Integer, String> weakHashMap = new WeakHashMap<>();
Integer key = new Integer(1);
weakHashMap.put(key, "one");
key = null;
System.gc();
System.out.println(weakHashMap); // 可能为空,因为key可能被回收
4. IdentityHashMap
IdentityHashMap使用键的引用相等性(reference equality)而不是键的equals方法来比较键。适用于需要比较对象引用而不是对象内容的场景。
Map<Integer, String> identityHashMap = new IdentityHashMap<>();
identityHashMap.put(new Integer(1), "one");
identityHashMap.put(new Integer(1), "one again");
System.out.println(identityHashMap.size()); // 输出2