HashMap 之弱引用 - WeakHashMap

■ Java 引用的相關知識

　 1. 強引用

Object o = new Object(); 

• 強引用是Java 默認實現 的引用，JVM會盡量長時間的保留強引用的存在（直到內存溢出）
• 當內存空間不足，Java虛擬機寧願拋出OutOfMemoryError錯誤，使程序異常終止，也不會靠隨意回收具備強引用的對象來解決內存不足的問題：只有當沒有任何對象指向它時JVM將會回收

　2. 軟引用

public class SoftReference<T> extends Reference<T> {...} 

• 軟引用只會在虛擬機 內存不足 的時候纔會被回收
• 軟引用能夠和一個引用隊列（ReferenceQueue）聯合使用，若是軟引用所引用的對象被垃圾回收器回收，Java虛擬機就會把這個軟引用加入到與之關聯的引用隊列中

   3. 弱引用

public class WeakReference<T> extends Reference<T> {...} 

• 弱引用是指當對象沒有任何的強引用存在，在 下次GC回收 的時候它將會被回收
• 在垃圾回收器線程掃描它所管轄的內存區域的過程當中，一旦發現了只具備弱引用的對象，無論當前內存空間足夠與否，都會回收它的內存
• 須要注意的是：因爲垃圾回收器是一個優先級很低的線程，所以不必定會很快發現那些只具備弱引用的對象

 

■ WeakHashMap 的認識：

• WeakHashMap 是存儲鍵值對(key-value)的非同步且無序的散列表，鍵和值都容許爲null，基本跟 HashMap一致
• 特殊之處在於 WeakHashMap 裏的entry可能會被GC自動刪除，即便沒有主動調用 remove() 或者 clear() 方法
• 其鍵爲弱鍵，除了自身有對key的引用外，此key沒有其餘引用那麼此map會自動丟棄此值
• 在《Effective Java》一書中第六條，消除陳舊對象時，提到了weakHashMap，用於短期內就過時的緩存
• 因爲與JDK1.7版本的HashMap實現原理一致（具體請參見筆者的 HashMap一文通），這裏只討論不一樣
• 如有機會寫JVM篇的垃圾回收機制時會再進一步描述 Reference

   1. 類定義

public class WeakHashMap<K,V>
    extends AbstractMap<K,V>
    implements Map<K,V>

  2. 重要的全局變量

/**
  * The default initial capacity -- MUST be a power of two.
  * 默認容量，必須是2次冪
  */
private static final int DEFAULT_INITIAL_CAPACITY = 16;
/**
  * The maximum capacity, used if a higher value is implicitly specified by either of the
  * constructors with arguments. MUST be a power of two <= 1<<30.
  * 最大容量，必須爲2次冪且 <= 1<<30
  */
private static final int MAXIMUM_CAPACITY = 1 << 30;
/**
  * The load factor used when none specified in constructor.
  * 負載因子
  */
private static final float DEFAULT_LOAD_FACTOR = 0.75f;
/**
  * The table, resized as necessary. Length MUST Always be a power of two.
  * 容量必須爲2次冪的數組
  */
Entry<K,V>[] table;
/**
  * The number of key-value mappings contained in this weak hash map.
  * 擁有鍵值對的數量
  */
private int size;
/**
  * The next size value at which to resize (capacity * load factor).
  * 閾值 -- 擴容判斷依據
  */
private int threshold;
/**
  * The load factor for the hash table.
  */
private final float loadFactor;
/**
  * Reference queue for cleared WeakEntries
  * 引用隊列，用於存儲已被GC清除的WeakEntries
  */
private final ReferenceQueue<Object> queue = new ReferenceQueue<>();

  3. 構造器

// 默認構造函數
WeakHashMap()
// 指定"容量大小"的構造函數
WeakHashMap(int capacity)
// 指定"容量大小"和"負載因子"的構造函數
WeakHashMap(int capacity, float loadFactor)
// 包含"子Map"的構造函數
WeakHashMap(Map<? extends K, ? extends V> map)

• 實現跟JDK1.7版本HashMap的實現一致,具體請參見筆者的  HashMap - 基於哈希表和 Map 接口的鍵值對利器 （JDK 1.7）

 4. Entry

/**
  * The entries in this hash table extend WeakReference, using its main ref field as the key.
  * 該Enty繼承WeakReference，從而具有弱引用的特性
  */
private static class Entry<K,V> extends WeakReference<Object> implements Map.Entry<K,V> {
    V value;
    int hash;
    Entry<K,V> next;//鏈表
    /**
      * Creates new entry.
      */
    Entry(Object key, V value,
            ReferenceQueue<Object> queue,
            int hash, Entry<K,V> next) {
        super(key, queue);
        this.value = value;
        this.hash  = hash;
        this.next  = next;
    }
    ....
}  

■ WeakHashMap 的重要方法

• 常規 put(), get(), remove() 等不詳細研究，來看下比較關鍵的 expungeStaleEntries()

/**
  * Expunges stale entries from the table. -- 刪除過期的entry
  * 該方法是實現弱鍵回收的最關鍵方法，也是區分HashMap的根本方法
  * 核心：移除table和queue的並集元素（queue中存儲是已被GC的key，注意是key！！）
  * 效果：key在GC的時候被清除，value在key清除後訪問WeakHashMap被清除
  */
private void expungeStaleEntries() {
    //從隊列中出隊遍歷
    //poll 移除並返問隊列頭部的元素；若是隊列爲空，則返回null
    for (Object x; (x = queue.poll()) != null; ) {
        synchronized (queue) {
            //值得一提的是WeakHashMap是非線程安全的，這裏卻同步了一下
            //大神本來的用意是保證在多線程時能不破壞數據結構，但JavaDoc已經強調這類非安全，以下文
            //http://bugs.java.com/bugdatabase/view_bug.do?bug_id=6425537
            @SuppressWarnings("unchecked")
                Entry<K,V> e = (Entry<K,V>) x;
            //找到該隊列中的元素在數組中的位置，並移除該元素（table和queue都須要移除）    
            int i = indexFor(e.hash, table.length);
            Entry<K,V> prev = table[i];
            Entry<K,V> p = prev;
            while (p != null) {
                Entry<K,V> next = p.next;
                if (p == e) {
                    if (prev == e)
                        table[i] = next;
                    else
                        prev.next = next;
                    // Must not null out e.next;
                    // stale entries may be in use by a HashIterator
                    e.value = null; // Help GC 移除value
                    size--;
                    break;
                }
                prev = p;
                p = next;
            }
        }
    }
}