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

我之前的一篇博客:System.gc()和-XX:+DisableExplicitGC启动参数,以及DirectByteBuffer的内存释放 文章末尾处:提到java NIO包是通过sun.misc.Cleaner和PhantomReference来实现堆外内存的自动释放的。现在我们来学习下Cleaner和PhantomReference的使用,自己封装实现堆外内存的自动释放。


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

package direct;
public class FreeMemoryTask implements Runnable
{
	private long address = 0;

	public FreeMemoryTask(long address)
	{
		this.address = address;
	}

	@Override
	public void run()
	{
		System.out.println("runing FreeMemoryTask");

		if (address == 0)
		{
			System.out.println("already released");
		} else
		{
			GetUsafeInstance.getUnsafeInstance().freeMemory(address);
		}

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

public class ObjectInHeapUseCleaner
{
	private long address = 0;

	public ObjectInHeapUseCleaner()
	{
		address = GetUsafeInstance.getUnsafeInstance().allocateMemory(
				2 * 1024 * 1024);
	}

	public static void main(String[] args)
	{
		while (true)
		{
			System.gc();

			ObjectInHeapUseCleaner heap = new ObjectInHeapUseCleaner();

			// 增加heap的虚引用,定义清理的接口FreeMemoryTask
			Cleaner.create(heap, new FreeMemoryTask(heap.address));
		}
	}
}

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

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


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

import java.lang.ref.PhantomReference;
import java.lang.ref.Reference;
import java.lang.ref.ReferenceQueue;
import java.util.HashMap;
import java.util.Map;

public class MyOwnCleaner
{

	private static ReferenceQueue refQueue = new ReferenceQueue();

	private static Map, Runnable> taskMap = new HashMap, Runnable>();

	static
	{
		new CleanerThread().start();
	}

	public static void clear(Object heapObject, Runnable task)
	{
		// 当heapObject没有强引用的时候,reference会自动被JVM加入到引用队列中
		// 不管使用有人持有reference对象的强引用
		PhantomReference reference = new PhantomReference(
				heapObject, refQueue);

		taskMap.put(reference, task);

	}

	// 清理线程
	private static class CleanerThread extends Thread
	{
		@Override
		public void run()
		{
			while (true)
			{
				try
				{
					@SuppressWarnings("unchecked")
					Reference refer = (Reference) refQueue
							.remove();

					Runnable r = taskMap.remove(refer);
					r.run();
				} catch (InterruptedException e)
				{

				}

			}
		}
	}

} 
  

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

private static ReferenceQueue refQueue = new ReferenceQueue(); 
  这个就是引用队列,JVM会自动将幽灵引用PhantomReference加入到队列中。也就是说,我们只要轮询这个队列,就能够知道哪些对象即将被JVM回收(这些对象只存在幽灵引用了)。 
  

public static void clear(Object heapObject, Runnable task)
{
	// 当heapObject没有强引用的时候,reference会自动被JVM加入到引用队列中
	// 不管使用有人持有reference对象的强引用
	PhantomReference reference = new PhantomReference(
			heapObject, refQueue);
	taskMap.put(reference, task);
} 
  这段代码,相当于是我们给堆中的对象heapObject添加了一个监控(注册了一个幽灵引用)。taskMap记录幽灵引用和相应的代码回收逻辑。 
  之后我们在后台开启了一个CleanerThread线程,不断的轮询引用队列,一旦发现队列中有数据(PhantomReference对象), 
  就找出对应的Runnable,调用它的run方法,释放堆对象heapObject中引用的堆外内存。测试代码如下: 
  

public class Test
{
	private long address = 0;

	public Test()
	{
		address = GetUsafeInstance.getUnsafeInstance().allocateMemory(
				2 * 1024 * 1024);
	}

	public static void main(String[] args)
	{
		while (true)
		{
			Test heap = new Test();

			MyOwnCleaner.clear(heap, new FreeMemoryTask(heap.address));

			System.gc();
		}
	}
}
运行测试代码,可以发现也不会报OOM,即正确地实现了堆外内存的自动释放。




你可能感兴趣的:(java高级话题,java)