Java 理論與實踐: 用弱引用堵住內存泄漏
要讓垃圾收集(GC)回收程序不再使用的對象,對象的邏輯 生命周期(應用程序使用它的時間)和對該對象擁有的引用的實際 生命周期必須是相同的。在大多數時候,好的軟件工程技術保證這是自動實現的,不用我們對對象生命周期問題花費過多心思。但是偶爾我們會創(chuàng)建一個引用,它在內存中包含對象的時間比我們預期的要長得多,這種情況稱為無意識的對象保留(unintentional object retention)。
無意識對象保留最常見的原因是使用 Map 將元數據與臨時對象(transient object)相關聯。假定一個對象具有中等生命周期,比分配它的那個方法調用的生命周期長,但是比應用程序的生命周期短,如客戶機的套接字連接。需要將一些元數據與這個套接字關聯,如生成連接的用戶的標識。在創(chuàng)建 Socket 時是不知道這些信息的,并且不能將數據添加到 Socket 對象上,因為不能控制 Socket 類或者它的子類。這時,典型的方法就是在一個全局 Map 中存儲這些信息,如清單 1 中的 SocketManager 類所示:
- public class SocketManager {
- private Map< Socket,User> m = new HashMap< Socket,User>();
- public void setUser(Socket s, User u) {
- m.put(s, u);
- }
- public User getUser(Socket s) {
- return m.get(s);
- }
- public void removeUser(Socket s) {
- m.remove(s);
- }
- }
- SocketManager socketManager;
- ...
- socketManager.setUser(socket, user);
這種方法的問題是元數據的生命周期需要與套接字的生命周期掛鉤,但是除非準確地知道什么時候程序不再需要這個套接字,并記住從 Map 中刪除相應的映射,否則,Socket 和 User 對象將會永遠留在 Map 中,遠遠超過響應了請求和關閉套接字的時間。這會阻止 Socket 和 User 對象被垃圾收集,即使應用程序不會再使用它們。這些對象留下來不受控制,很容易造成程序在長時間運行后內存爆滿。除了最簡單的情況,在幾乎所有情況下找出什么時候 Socket 不再被程序使用是一件很煩人和容易出錯的任務,需要人工對內存進行管理。
程序有內存泄漏的第一個跡象通常是它拋出一個 OutOfMemoryError,或者因為頻繁的垃圾收集而表現出糟糕的性能。幸運的是,垃圾收集可以提供能夠用來診斷內存泄漏的大量信息。如果以 -verbose:gc 或者 -Xloggc 選項調用 JVM,那么每次 GC 運行時在控制臺上或者日志文件中會打印出一個診斷信息,包括它所花費的時間、當前堆使用情況以及恢復了多少內存。記錄 GC 使用情況并不具有干擾性,因此如果需要分析內存問題或者調優(yōu)垃圾收集器,在生產環(huán)境中默認啟用 GC 日志是值得的。
有工具可以利用 GC 日志輸出并以圖形方式將它顯示出來,JTune 就是這樣的一種工具(請參閱 參考資料)。觀察 GC 之后堆大小的圖,可以看到程序內存使用的趨勢。對于大多數程序來說,可以將內存使用分為兩部分:baseline 使用和 current load 使用。對于服務器應用程序,baseline 使用就是應用程序在沒有任何負荷、但是已經準備好接受請求時的內存使用,current load 使用是在處理請求過程中使用的、但是在請求處理完成后會釋放的內存。只要負荷大體上是恒定的,應用程序通常會很快達到一個穩(wěn)定的內存使用水平。如果在應用程序已經完成了其初始化并且負荷沒有增加的情況下,內存使用持續(xù)增加,那么程序就可能在處理前面的請求時保留了生成的對象。
清單 2 展示了一個有內存泄漏的程序。MapLeaker 在線程池中處理任務,并在一個 Map 中記錄每一項任務的狀態(tài)。不幸的是,在任務完成后它不會刪除那一項,因此狀態(tài)項和任務對象(以及它們的內部狀態(tài))會不斷地積累。
- public class MapLeaker {
- public ExecutorService exec = Executors.newFixedThreadPool(5);
- public Map< Task, TaskStatus> taskStatus
- = Collections.synchronizedMap(new HashMap< Task, TaskStatus>());
- private Random random = new Random();
- private enum TaskStatus { NOT_STARTED, STARTED, FINISHED };
- private class Task implements Runnable {
- private int[] numbers = new int[random.nextInt(200)];
- public void run() {
- int[] temp = new int[random.nextInt(10000)];
- taskStatus.put(this, TaskStatus.STARTED);
- doSomeWork();
- taskStatus.put(this, TaskStatus.FINISHED);
- }
- }
- public Task newTask() {
- Task t = new Task();
- taskStatus.put(t, TaskStatus.NOT_STARTED);
- exec.execute(t);
- return t;
- }
- }
圖 1 顯示 MapLeaker GC 之后應用程序堆大小隨著時間的變化圖。上升趨勢是存在內存泄漏的警示信號。(在真實的應用程序中,坡度不會這么大,但是在收集了足夠長時間的 GC 數據后,上升趨勢通常會表現得很明顯。)
確信有了內存泄漏后,下一步就是找出哪種對象造成了這個問題。所有內存分析器都可以生成按照對象類進行分解的堆快照。有一些很好的商業(yè)堆分析工具,但是找出內存泄漏不一定要花錢買這些工具 —— 內置的 hprof 工具也可完成這項工作。要使用 hprof 并讓它跟蹤內存使用,需要以 -Xrunhprof:heap=sites 選項調用 JVM。
清單 3 顯示分解了應用程序內存使用的 hprof 輸出的相關部分。(hprof 工具在應用程序退出時,或者用 kill -3 或在 Windows 中按 Ctrl+Break 時生成使用分解。)注意兩次快照相比,Map.Entry、Task 和 int[] 對象有了顯著增加。
- SITES BEGIN (ordered by live bytes) Fri Oct 28 16:30:48 2005
- percent live alloc'ed stack class
- rank self accum bytes objs bytes objs trace name
- 1 70.13% 70.13% 5694888 13909 5694888 13909 300305 int[]
- 2 18.27% 88.40% 1483976 68 278273632 13908 300321 int[]
- 3 4.11% 92.51% 333816 13909 333816 13909 300310 java.util.HashMap$Entry
- 4 2.74% 95.25% 222544 13909 222544 13909 300304 com.quiotix.dummy.MapLeaker$Task
- 5 2.42% 97.67% 196640 2 262192 11 300325 java.util.HashMap$Entry[]
- 6 0.66% 98.33% 53680 3355 222464 13904 300324 java.util.concurrent.LinkedBlockingQueue$Node
- SITES END
- SITES BEGIN (ordered by live bytes) Fri Oct 28 16:31:32 2005
- percent live alloc'ed stack class
- rank self accum bytes objs bytes objs trace name
- 1 77.07% 77.07% 41176024 100020 41176024 100020 301069 int[]
- 2 12.98% 90.05% 6933768 359 2001885688 100020 301093 int[]
- 3 4.49% 94.55% 2400480 100020 2400480 100020 301082 java.util.HashMap$Entry
- 4 3.00% 97.54% 1600320 100020 1600320 100020 301068 com.quiotix.dummy.MapLeaker$Task
- 5 1.96% 99.50% 1048592 1 2097248 14 301104 java.util.HashMap$Entry[]
- 6 0.05% 99.55% 25936 1621 1600240 100015 301101 java.util.concurrent.LinkedBlockingQueue$Node
- SITES END
清單 4 展示了 hprof 輸出的另一部分,給出了 Map.Entry 對象的分配點的調用堆棧信息。這個輸出告訴我們哪些調用鏈生成了 Map.Entry 對象,并帶有一些程序分析,找出內存泄漏來源一般來說是相當容易的。
- TRACE 300446:
- java.util.HashMap$Entry.< init>(< Unknown Source>:Unknown line)
- java.util.HashMap.addEntry(< Unknown Source>:Unknown line)
- java.util.HashMap.put(< Unknown Source>:Unknown line)
- java.util.Collections$SynchronizedMap.put(< Unknown Source>:Unknown line)
- com.quiotix.dummy.MapLeaker.newTask(MapLeaker.java:48)
- com.quiotix.dummy.MapLeaker.main(MapLeaker.java:64)
SocketManager 的問題是 Socket-User 映射的生命周期應當與 Socket 的生命周期相匹配,但是語言沒有提供任何容易的方法實施這項規(guī)則。這使得程序不得不使用人工內存管理的老技術。幸運的是,從 JDK 1.2 開始,垃圾收集器提供了一種聲明這種對象生命周期依賴性的方法,這樣垃圾收集器就可以幫助我們防止這種內存泄漏 —— 利用弱引用。
#p#
弱引用是對一個對象(稱為 referent)的引用的持有者。使用弱引用后,可以維持對 referent 的引用,而不會阻止它被垃圾收集。當垃圾收集器跟蹤堆的時候,如果對一個對象的引用只有弱引用,那么這個 referent 就會成為垃圾收集的候選對象,就像沒有任何剩余的引用一樣,而且所有剩余的弱引用都被清除。(只有弱引用的對象稱為弱可及(weakly reachable)。)
WeakReference 的 referent 是在構造時設置的,在沒有被清除之前,可以用 get() 獲取它的值。如果弱引用被清除了(不管是 referent 已經被垃圾收集了,還是有人調用了 WeakReference.clear()),get() 會返回 null。相應地,在使用其結果之前,應當總是檢查 get() 是否返回一個非 null 值,因為 referent 最終總是會被垃圾收集的。
用一個普通的(強)引用拷貝一個對象引用時,限制 referent 的生命周期至少與被拷貝的引用的生命周期一樣長。如果不小心,那么它可能就與程序的生命周期一樣 —— 如果將一個對象放入一個全局集合中的話。另一方面,在創(chuàng)建對一個對象的弱引用時,完全沒有擴展 referent 的生命周期,只是在對象仍然存活的時候,保持另一種到達它的方法。
弱引用對于構造弱集合最有用,如那些在應用程序的其余部分使用對象期間存儲關于這些對象的元數據的集合 —— 這就是 SocketManager 類所要做的工作。因為這是弱引用最常見的用法,WeakHashMap 也被添加到 JDK 1.2 的類庫中,它對鍵(而不是對值)使用弱引用。如果在一個普通 HashMap 中用一個對象作為鍵,那么這個對象在映射從 Map 中刪除之前不能被回收,WeakHashMap 使您可以用一個對象作為 Map 鍵,同時不會阻止這個對象被垃圾收集。清單 5 給出了 WeakHashMap 的 get() 方法的一種可能實現,它展示了弱引用的使用:
- public class WeakHashMap< K,V> implements Map< K,V> {
- private static class Entry< K,V> extends WeakReference< K>
- implements Map.Entry< K,V> {
- private V value;
- private final int hash;
- private Entry< K,V> next;
- ...
- }
- public V get(Object key) {
- int hash = getHash(key);
- Entry< K,V> e = getChain(hash);
- while (e != null) {
- K eKey= e.get();
- if (e.hash == hash && (key == eKey || key.equals(eKey)))
- return e.value;
- e = e.next;
- }
- return null;
- }
調用 WeakReference.get() 時,它返回一個對 referent 的強引用(如果它仍然存活的話),因此不需要擔心映射在 while 循環(huán)體中消失,因為強引用會防止它被垃圾收集。WeakHashMap 的實現展示了弱引用的一種常見用法 —— 一些內部對象擴展 WeakReference。其原因在下面一節(jié)討論引用隊列時會得到解釋。
在向 WeakHashMap 中添加映射時,請記住映射可能會在以后“脫離”,因為鍵被垃圾收集了。在這種情況下,get() 返回 null,這使得測試 get() 的返回值是否為 null 變得比平時更重要了。
用 WeakHashMap 堵住泄漏
在 SocketManager 中防止泄漏很容易,只要用 WeakHashMap 代替 HashMap 就行了,如清單 6 所示。(如果 SocketManager 需要線程安全,那么可以用 Collections.synchronizedMap() 包裝 WeakHashMap)。當映射的生命周期必須與鍵的生命周期聯系在一起時,可以使用這種方法。不過,應當小心不濫用這種技術,大多數時候還是應當使用普通的 HashMap 作為 Map 的實現。
- public class SocketManager {
- private Map< Socket,User> m = new WeakHashMap< Socket,User>();
- public void setUser(Socket s, User u) {
- m.put(s, u);
- }
- public User getUser(Socket s) {
- return m.get(s);
- }
- }
WeakHashMap 用弱引用承載映射鍵,這使得應用程序不再使用鍵對象時它們可以被垃圾收集,get() 實現可以根據 WeakReference.get() 是否返回 null 來區(qū)分死的映射和活的映射。但是這只是防止 Map 的內存消耗在應用程序的生命周期中不斷增加所需要做的工作的一半,還需要做一些工作以便在鍵對象被收集后從 Map 中刪除死項。否則,Map 會充滿對應于死鍵的項。雖然這對于應用程序是不可見的,但是它仍然會造成應用程序耗盡內存,因為即使鍵被收集了,Map.Entry 和值對象也不會被收集。
可以通過周期性地掃描 Map,對每一個弱引用調用 get(),并在 get() 返回 null 時刪除那個映射而消除死映射。但是如果 Map 有許多活的項,那么這種方法的效率很低。如果有一種方法可以在弱引用的 referent 被垃圾收集時發(fā)出通知就好了,這就是引用隊列 的作用。
引用隊列是垃圾收集器向應用程序返回關于對象生命周期的信息的主要方法。弱引用有兩個構造函數:一個只取 referent 作為參數,另一個還取引用隊列作為參數。如果用關聯的引用隊列創(chuàng)建弱引用,在 referent 成為 GC 候選對象時,這個引用對象(不是 referent)就在引用清除后加入 到引用隊列中。之后,應用程序從引用隊列提取引用并了解到它的 referent 已被收集,因此可以進行相應的清理活動,如去掉已不在弱集合中的對象的項。(引用隊列提供了與 BlockingQueue 同樣的出列模式 —— polled、timed blocking 和 untimed blocking。)
WeakHashMap 有一個名為 expungeStaleEntries() 的私有方法,大多數 Map 操作中會調用它,它去掉引用隊列中所有失效的引用,并刪除關聯的映射。清單 7 展示了 expungeStaleEntries() 的一種可能實現。用于存儲鍵-值映射的 Entry 類型擴展了 WeakReference,因此當 expungeStaleEntries() 要求下一個失效的弱引用時,它得到一個 Entry。用引用隊列代替定期掃描內容的方法來清理 Map 更有效,因為清理過程不會觸及活的項,只有在有實際加入隊列的引用時它才工作。
- private void expungeStaleEntries() {
- ry< K,V> e;
- while ( (e = (Entry< K,V>) queue.poll()) != null) {
- int hash = e.hash;
- Entry< K,V> prev = getChain(hash);
- Entry< K,V> cur = prev;
- while (cur != null) {
- Entry< K,V> next = cur.next;
- if (cur == e) {
- if (prev == e)
- setChain(hash, next);
- else
- prev.next = next;
- break;
- }
- prev = cur;
- cur = next;
- }
- }
- }
弱引用和弱集合是對堆進行管理的強大工具,使得應用程序可以使用更復雜的可及性方案,而不只是由普通(強)引用所提供的“要么全部要么沒有”可及性。
【編輯推薦】