如何通过 WeakReference 管理内存

在Java中，WeakReference是一种引用类型，它允许你持有对对象的弱引用。当一个对象只被弱引用所指向时，它就成为了垃圾回收的候选对象。这意味着，如果没有其他强引用指向该对象，那么在下一次垃圾回收时，该对象就会被回收。

使用WeakReference可以帮助你管理内存，特别是在缓存等场景中。下面是一些通过WeakReference管理内存的基本步骤：

1. 创建 WeakReference

你可以使用WeakReference类的构造函数来创建一个弱引用。构造函数接受两个参数：要引用的对象和引用队列（可选）。

Object obj = new Object();
WeakReference<Object> weakRef = new WeakReference<>(obj);

2. 获取弱引用对象

你可以使用WeakReference的get()方法来获取它所引用的对象。如果对象已经被垃圾回收，get()方法将返回null。

Object obj = weakRef.get();
if (obj == null) {
    // 对象已被回收
} else {
    // 对象仍然存在
}

3. 使用引用队列（可选）

引用队列允许你在对象被垃圾回收时得到通知。当你创建一个WeakReference时，可以将其与一个引用队列关联。

ReferenceQueue<Object> queue = new ReferenceQueue<>();
WeakReference<Object> weakRef = new WeakReference<>(obj, queue);

// 在某个地方检查引用队列
Reference<? extends Object> ref = queue.poll();
if (ref != null) {
    // 对象已被回收
}

4. 清理不再需要的弱引用

为了避免内存泄漏，你应该定期清理不再需要的弱引用。这可以通过轮询引用队列来实现。

ReferenceQueue<Object> queue = new ReferenceQueue<>();
WeakReference<Object> weakRef = new WeakReference<>(obj, queue);

// 定期清理引用队列
Reference<? extends Object> ref;
while ((ref = queue.poll()) != null) {
    // 处理被回收的对象
}

示例：缓存实现

下面是一个简单的缓存实现，使用WeakReference来管理缓存项：

import java.lang.ref.WeakReference;
import java.util.HashMap;
import java.util.Map;

public class WeakReferenceCache<K, V> {
    private final Map<K, WeakReference<V>> cache = new HashMap<>();

    public void put(K key, V value) {
        cache.put(key, new WeakReference<>(value));
    }

    public V get(K key) {
        WeakReference<V> weakRef = cache.get(key);
        if (weakRef != null) {
            return weakRef.get();
        }
        return null;
    }

    public void remove(K key) {
        cache.remove(key);
    }

    public void clear() {
        cache.clear();
    }
}

在这个示例中，当缓存中的值不再被其他强引用持有时，它将被垃圾回收，并且在下一次访问时，get()方法将返回null。

通过这种方式，你可以有效地管理内存，避免内存泄漏。