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