使用sun.misc.Cleaner或者PhantomReference实现堆外内存的自动释放

Java NIO包是通过sun.misc.Cleaner和PhantomReference来实现堆外内存的自动释放的。现在我们来学习下Cleaner和PhantomReference的使用，自己封装实现堆外内存的自动释放。

sun.misc.Cleaner是JDK内部提供的用来释放非堆内存资源的API。JVM只会帮我们自动释放堆内存资源，但是它提供了回调机制，通过这个类能方便的释放系统的其他资源。我们先看下如何使用Cleaner。

[java]  view plain  copy

1. package direct;  
2. public class FreeMemoryTask implements Runnable  
3. {  
4.     private long address = 0;  
5.   
6.     public FreeMemoryTask(long address)  
7.     {  
8.         this.address = address;  
9.     }  
10.   
11.     @Override  
12.     public void run()  
13.     {  
14.         System.out.println("runing FreeMemoryTask");  
15.   
16.         if (address == 0)  
17.         {  
18.             System.out.println("already released");  
19.         } else  
20.         {  
21.             GetUsafeInstance.getUnsafeInstance().freeMemory(address);  
22.         }  
23.   
24.     }  
25. }  

这个实现了Runnable接口的类，功能就是释放堆外内存。这是我们必须要做的事，JVM没有办法帮我们做。

[java]  view plain  copy

1. public class ObjectInHeapUseCleaner  
2. {  
3.     private long address = 0;  
4.   
5.     public ObjectInHeapUseCleaner()  
6.     {  
7.         address = GetUsafeInstance.getUnsafeInstance().allocateMemory(  
8.                 2 * 1024 * 1024);  
9.     }  
10.   
11.     public static void main(String[] args)  
12.     {  
13.         while (true)  
14.         {  
15.             System.gc();  
16.   
17.             ObjectInHeapUseCleaner heap = new ObjectInHeapUseCleaner();  
18.   
19.             // 增加heap的虚引用,定义清理的接口FreeMemoryTask  
20.             Cleaner.create(heap, new FreeMemoryTask(heap.address));  
21.         }  
22.     }  
23. }  

运行这段代码，可以发现程序正常运行，不会出现OOM。

 Cleaner.create()需要2个参数：第一个参数：需要监控的堆内存对象，第二个参数：程序释放资源的回调。当JVM进行GC的时候，如果发现我们监控的对象，不存在强引用了（只被Cleaner对象引用，这是个幽灵引用），就会调用第二个参数Runnable.run()方法的逻辑，执行完Runnable.run()的时候（这个时候已经释放了堆外内存），JVM会自动释放堆内存中我们监控的对象。可以看到使用sun.misc.Cleaner很简单。

接下来我们看下，不使用sun.misc.Cleaner的情况下，如何释放资源。

[java]  view plain  copy

1. import java.lang.ref.PhantomReference;  
2. import java.lang.ref.Reference;  
3. import java.lang.ref.ReferenceQueue;  
4. import java.util.HashMap;  
5. import java.util.Map;  
6.   
7. public class MyOwnCleaner  
8. {  
9.   
10.     private static ReferenceQueue<Object> refQueue = new ReferenceQueue<Object>();  
11.   
12.     private static Map<Reference<Object>, Runnable> taskMap = new HashMap<Reference<Object>, Runnable>();  
13.   
14.     static  
15.     {  
16.         new CleanerThread().start();  
17.     }  
18.   
19.     public static void clear(Object heapObject, Runnable task)  
20.     {  
21.         // 当heapObject没有强引用的时候,reference会自动被JVM加入到引用队列中  
22.         // 不管使用有人持有reference对象的强引用  
23.         PhantomReference<Object> reference = new PhantomReference<Object>(  
24.                 heapObject, refQueue);  
25.   
26.         taskMap.put(reference, task);  
27.   
28.     }  
29.   
30.     // 清理线程  
31.     private static class CleanerThread extends Thread  
32.     {  
33.         @Override  
34.         public void run()  
35.         {  
36.             while (true)  
37.             {  
38.                 try  
39.                 {  
40.                     @SuppressWarnings("unchecked")  
41.                     Reference<Object> refer = (Reference<Object>) refQueue  
42.                             .remove();  
43.   
44.                     Runnable r = taskMap.remove(refer);  
45.                     r.run();  
46.                 } catch (InterruptedException e)  
47.                 {  
48.   
49.                 }  
50.   
51.             }  
52.         }  
53.     }  
54.   
55. }  

这里使用到了PhantomReference和ReferenceQueue，这是JVM内部的对象销毁机制。当堆中的对象不存在强引用，只存在幽灵引用的时候，JVM会自动将这个对象的幽灵引用加入到与之相关联的的引用队列中。

[java]  view plain  copy

1. private static ReferenceQueue<Object> refQueue = new ReferenceQueue<Object>();  

这个就是引用队列，JVM会自动将幽灵引用PhantomReference加入到队列中。也就是说，我们只要轮询这个队列，就能够知道哪些对象即将被JVM回收（这些对象只存在幽灵引用了）。

[java]  view plain  copy

1. public static void clear(Object heapObject, Runnable task)  
2. {  
3.     // 当heapObject没有强引用的时候,reference会自动被JVM加入到引用队列中  
4.     // 不管使用有人持有reference对象的强引用  
5.     PhantomReference<Object> reference = new PhantomReference<Object>(  
6.             heapObject, refQueue);  
7.     taskMap.put(reference, task);  
8. }  

这段代码，相当于是我们给堆中的对象heapObject添加了一个监控（注册了一个幽灵引用）。taskMap记录幽灵引用和相应的代码回收逻辑。 之后我们在后台开启了一个CleanerThread线程，不断的轮询引用队列，一旦发现队列中有数据(PhantomReference对象)， 就找出对应的Runnable，调用它的run方法，释放堆对象heapObject中引用的堆外内存。测试代码如下：

[java]  view plain  copy

1. public class Test  
2. {  
3.     private long address = 0;  
4.   
5.     public Test()  
6.     {  
7.         address = GetUsafeInstance.getUnsafeInstance().allocateMemory(  
8.                 2 * 1024 * 1024);  
9.     }  
10.   
11.     public static void main(String[] args)  
12.     {  
13.         while (true)  
14.         {  
15.             Test heap = new Test();  
16.   
17.             MyOwnCleaner.clear(heap, new FreeMemoryTask(heap.address));  
18.   
19.             System.gc();  
20.         }  
21.     }  
22. }  

运行测试代码，可以发现也不会报OOM，即正确地实现了堆外内存的自动释放。