MK
摩柯社区 - 一个极简的技术知识社区
AI 面试

Java HashMap与WeakHashMap的内存管理

2021-01-243.2k 阅读

Java HashMap的内存管理

在Java开发中，HashMap是一个常用的集合类，用于存储键值对。理解它的内存管理机制对于编写高效且内存友好的代码至关重要。

1. HashMap的基本结构

HashMap内部使用数组和链表（在Java 8及之后引入了红黑树）来实现。数组的每个元素称为桶（bucket），每个桶可以存储一个键值对的链表或红黑树节点。

public class HashMap<K, V> {
    transient Node<K, V>[] table;

    static class Node<K, V> implements Map.Entry<K, V> {
        final int hash;
        final K key;
        V value;
        Node<K, V> next;

        Node(int hash, K key, V value, Node<K, V> next) {
            this.hash = hash;
            this.key = key;
            this.value = value;
            this.next = next;
        }
    }
}

这里的table数组就是用来存放Node节点的，Node类包含了键值对的信息以及指向下一个节点的引用。

2. 内存分配与初始化

当创建一个HashMap实例时，如果没有指定初始容量，它会使用默认初始容量16。这个容量会影响到数组table的大小。

HashMap<String, Integer> map = new HashMap<>();

在上述代码中，map的初始容量为16，意味着table数组初始大小为16，此时内存中会为这个长度为16的数组分配空间。

如果指定了初始容量，HashMap会根据这个容量计算出一个合适的2的幂次方作为实际的初始容量。例如：

HashMap<String, Integer> map = new HashMap<>(32);

这里指定初始容量为32，HashMap内部可能会直接使用32作为数组table的初始大小（实际可能会根据一些规则调整），并为其分配内存。

3. 元素插入与内存变化

当向HashMap中插入一个键值对时，首先会计算键的哈希值，然后根据哈希值确定该键值对应该存储在数组的哪个桶中。

HashMap<String, Integer> map = new HashMap<>();
map.put("key1", 1);

假设键"key1"的哈希值计算后对应的桶为空，那么就会在该桶位置创建一个新的Node节点来存储这个键值对。此时，除了数组table本身占用的内存外，新创建的Node节点也会占用一定内存，包括hash值、键"key1"、值1以及指向下一个节点的next引用（此时为null）。

如果该桶已经有其他节点（发生哈希冲突），新的节点会以链表的形式追加到该桶的链表末尾（在Java 8之前）或根据红黑树的规则插入（Java 8及之后，当链表长度达到一定阈值且数组容量足够时会转换为红黑树）。

map.put("key2", 2);

假设"key2"的哈希值也对应到"key1"所在的桶，那么会在该桶的链表末尾添加一个新的Node节点，新节点的next指向原来的Node节点。这样，内存中除了新节点本身占用的空间外，链表结构的维护也会占用一定内存（通过next引用）。

4. 负载因子与扩容

HashMap有一个负载因子（默认值为0.75），用于衡量HashMap的填满程度。当HashMap中的元素数量达到容量 * 负载因子时，就会触发扩容。

HashMap<String, Integer> map = new HashMap<>(16);
for (int i = 0; i < 13; i++) {
    map.put("key" + i, i);
}

这里初始容量为16，负载因子为0.75，当插入第13个元素时，元素数量达到16 * 0.75 = 12，此时会触发扩容。

扩容时，HashMap会创建一个新的更大的数组，通常是原数组大小的两倍。然后将原数组中的所有元素重新计算哈希值并插入到新数组中。这一过程涉及大量的内存操作，包括新数组的创建、旧元素的重新哈希和插入，因此扩容操作相对耗时且会导致内存的重新分配。

5. 元素删除与内存回收

当从HashMap中删除一个元素时，对应的Node节点会从链表或红黑树中移除。

map.remove("key1");

如果被删除的节点是链表中的唯一节点，那么该节点占用的内存空间就可以被垃圾回收机制回收。如果是链表中的中间节点，那么需要调整链表的next引用，使得前驱节点直接指向后继节点，被删除节点占用的内存也会等待垃圾回收。

在红黑树结构中删除节点相对复杂一些，但最终目的也是使得被删除节点占用的内存可以被回收。垃圾回收机制会在合适的时候扫描内存，标记不再被引用的对象（如被删除的Node节点），并释放其占用的内存。

Java WeakHashMap的内存管理

WeakHashMap是HashMap的一个特殊子类，它在内存管理方面有着与HashMap不同的特性。

1. WeakHashMap的基本原理

WeakHashMap使用弱引用（WeakReference）来存储键。弱引用的特点是，当一个对象只有弱引用指向它时，在垃圾回收机制运行时，这个对象就会被回收，即使此时WeakHashMap中还存在对该键的引用。

public class WeakHashMap<K, V> extends AbstractMap<K, V>
    implements Map<K, V> {
    private static class Entry<K, V> extends WeakReference<Object>
        implements Map.Entry<K, V> {
        V value;
        Entry<K, V> next;

        Entry(K key, V value,
              ReferenceQueue<Object> queue,
              int hash, Entry<K, V> next) {
            super(key, queue);
            this.value = value;
            this.next = next;
            this.hash = hash;
        }
    }
}

这里的Entry类继承自WeakReference，用来存储键值对，其中键通过弱引用存储。

2. 内存分配与初始化

WeakHashMap的初始化过程与HashMap类似，可以使用默认构造函数创建一个初始容量为16的实例，也可以指定初始容量。

WeakHashMap<String, Integer> weakMap = new WeakHashMap<>();

上述代码创建了一个初始容量为16的WeakHashMap，会为其内部的数组table分配内存，与HashMap类似。

3. 元素插入与内存变化

当向WeakHashMap中插入一个键值对时，过程与HashMap相似，会计算键的哈希值确定存储位置。

WeakHashMap<String, Integer> weakMap = new WeakHashMap<>();
String key = new String("key1");
weakMap.put(key, 1);

这里创建了一个String对象key并插入到WeakHashMap中。由于WeakHashMap使用弱引用存储键，内存中除了WeakHashMap内部结构和值对象1占用的空间外，键key对应的String对象会被弱引用关联。

4. 弱引用与内存回收

WeakHashMap的关键特性在于其对键的弱引用。假设在上述代码之后，没有其他强引用指向key对象：

key = null;
System.gc();

当调用System.gc()（实际垃圾回收不一定立即执行）时，如果垃圾回收机制运行，由于key对象只有WeakHashMap中的弱引用指向它，key对象就会被回收。同时，WeakHashMap中对应的Entry节点也会被标记为无效（因为其键已经被回收），在下一次WeakHashMap进行清理操作（如插入新元素或调用get方法时），这些无效的节点会被移除，从而释放相关的内存。

5. 清理机制

WeakHashMap内部有一个清理机制，在进行某些操作（如插入新元素、获取元素等）时，会检查并移除那些键已经被回收的无效Entry节点。

WeakHashMap<String, Integer> weakMap = new WeakHashMap<>();
String key = new String("key1");
weakMap.put(key, 1);
key = null;
System.gc();
// 调用get方法触发清理
Integer value = weakMap.get("key1");

在调用get方法时，WeakHashMap会检查内部的无效Entry节点并移除它们，这样可以保证WeakHashMap占用的内存不会因为无效节点而一直膨胀。

HashMap与WeakHashMap内存管理对比

  1. 键的引用类型
    • HashMap使用强引用存储键，只要HashMap实例存在且键值对未被移除，键对象就不会被垃圾回收。
    • WeakHashMap使用弱引用存储键，当键对象没有其他强引用时，可能会被垃圾回收机制回收。
  2. 内存释放时机
    • 在HashMap中，只有当键值对被明确移除（通过remove方法）或者HashMap实例本身被回收时，相关的内存才会被释放。
    • WeakHashMap在键对象被垃圾回收后，会在后续的清理操作中移除对应的无效Entry节点，释放相关内存。这使得WeakHashMap在处理大量临时对象作为键时，能更及时地释放内存。
  3. 应用场景
    • HashMap适用于需要长期持有键值对，且键对象的生命周期与HashMap实例的生命周期紧密相关的场景，比如缓存一些不轻易变化的数据。
    • WeakHashMap适用于那些希望在键对象不再被其他地方强引用时，能自动释放相关内存的场景，例如实现缓存机制，当缓存对象不再被其他部分使用时，自动清理缓存以节省内存。
  4. 性能影响
    • HashMap在插入、删除和查找操作时，由于不需要处理弱引用相关的逻辑，通常性能相对稳定。
    • WeakHashMap在每次操作时可能需要额外检查并清理无效节点，特别是在数据量较大且键频繁被回收的情况下，可能会对性能产生一定影响。但如果能合理利用其特性，在内存管理方面能带来很大优势。

总结与实践建议

在实际开发中，选择HashMap还是WeakHashMap取决于具体的需求。如果对内存非常敏感，且有大量临时对象作为键，并且希望在这些对象不再被强引用时自动释放内存，那么WeakHashMap是一个不错的选择。但需要注意其可能带来的性能影响，特别是在高并发读写的场景下。

而如果数据需要长期稳定存储，且对性能有较高要求，HashMap则是更合适的选择。同时，在使用WeakHashMap时，要了解其清理机制，合理触发清理操作，以保证内存的有效管理。

在进行大规模数据处理和内存敏感的应用开发中，深入理解HashMap和WeakHashMap的内存管理机制，能够帮助开发者优化代码，提高系统的性能和稳定性。无论是选择哪种集合类，都需要根据实际场景进行权衡和测试，以达到最佳的内存使用和性能表现。