UnsafeUtil.java

package xerial.larray.buffer;

import sun.misc.Unsafe;

import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.nio.ByteBuffer;

/**
 * Helper methods for using sun.misc.Unsafe.
 *
 * @author Taro L. Saito
 */
public class UnsafeUtil {

    public static Unsafe getUnsafe() {
        try {
            Field f = Unsafe.class.getDeclaredField("theUnsafe");
            f.setAccessible(true);
            return Unsafe.class.cast(f.get(null));
        }
        catch(NoSuchFieldException e) {
            throw new IllegalStateException("sun.misc.Unsafe is not available in this JVM");
        }
        catch(IllegalAccessException e) {
            throw new IllegalStateException("sun.misc.Unsafe is not available in this JVM");
        }
    }

    public static Unsafe unsafe = getUnsafe();

    private static ByteBufferCreator findDirectByteBufferConstructor() {
        try {
            return constructorWithAttStrategy();
        }
        catch(ClassNotFoundException e) {
            throw new IllegalStateException(
                String.format("Failed to find java.nio.DirectByteBuffer: $s", e.getMessage()));
        }
        catch(NoSuchMethodException e) {
            try {
                return constructorWithoutAttStrategy();
            } catch (NoSuchMethodException e2) {
                throw new IllegalStateException(
                    String.format("Failed to find constructor f java.nio.DirectByteBuffer: $s", e2.getMessage()), e2);
            } catch (ClassNotFoundException e2) {
                throw new IllegalStateException(
                    String.format("Failed to find constructor f java.nio.DirectByteBuffer: $s", e2.getMessage()), e2);
            }
        }
    }

    private static ByteBufferCreator byteBufferCreator = findDirectByteBufferConstructor();

    /**
     * Create a new DirectByteBuffer from a given address and size.
     * The returned DirectByteBuffer does not release the memory by itself.
     *
     * @param addr
     * @param size
     * @param att  object holding the underlying memory to attach to the buffer.
     *             This will prevent the garbage collection of the memory area that's
     *             associated with the new <code>DirectByteBuffer</code>
     * @return
     */
    public static ByteBuffer newDirectByteBuffer(long addr, int size, Object att) {
        return byteBufferCreator.newDirectByteBuffer(addr, size, att);
    }

    private static ByteBufferCreator constructorWithAttStrategy()
        throws ClassNotFoundException, NoSuchMethodException {
        final Constructor<? extends ByteBuffer> dbbCC =
            (Constructor<? extends ByteBuffer>) Class.forName("java.nio.DirectByteBuffer")
                .getDeclaredConstructor(Long.TYPE, Integer.TYPE, Object.class);
        return new ByteBufferCreator() {
            @Override
            public ByteBuffer newDirectByteBuffer(long addr, int size, Object att) {
                dbbCC.setAccessible(true);
                try {
                    return dbbCC.newInstance(Long.valueOf(addr), Integer.valueOf(size), att);
                } catch (Exception e) {
                    throw new IllegalStateException(
                        String.format("Failed to create DirectByteBuffer: %s", e.getMessage()), e);
                }
            }
        };
    }

    private static ByteBufferCreator constructorWithoutAttStrategy()
        throws ClassNotFoundException, NoSuchMethodException {
        final Constructor<? extends ByteBuffer> dbbCC =
            (Constructor<? extends ByteBuffer>) Class.forName("java.nio.DirectByteBuffer")
                .getDeclaredConstructor(Long.TYPE, Integer.TYPE);
        return new ByteBufferCreator() {
            @Override
            public ByteBuffer newDirectByteBuffer(long addr, int size, Object att) {
                dbbCC.setAccessible(true);
                try {
                    return dbbCC.newInstance(Long.valueOf(addr), Integer.valueOf(size));
                } catch (Exception e) {
                    throw new IllegalStateException(
                        String.format("Failed to create DirectByteBuffer: %s", e.getMessage()), e);
                }
            }
        };
    }

    private interface ByteBufferCreator {
        ByteBuffer newDirectByteBuffer(long addr, int size, Object att);
    }


}