天天看点

java中Unsafe使用讲解

前段时间因为看JUC的源码,里面有大量关于unsafe的操作,所以就来看看了.写点笔记总结下(本文基于jdk1.8):

unsafe可以帮我们直接去操作硬件资源,当然了是借助java的jit来进行的,官方不推荐使用,因为不安全,例如你使用unsafe创建一个超级大的数组,但是这个数组jvm是不管理的,只能你自己操作,容易oom,也不利于资源的回收.

好了,下面我们来看代码,

1.获取unsafe

  1. ​//1.最简单的使用方式是基于反射获取Unsafe实例​

  2. ​Field f = Unsafe.class.getDeclaredField("theUnsafe");​

  3. ​f.setAccessible(true);​

  4. ​Unsafe unsafe = (Unsafe) f.get(null);​

2.获取unsafe

  1. ​private static Unsafe unsafe = null;​

  2. ​private static Field getUnsafe = null;​

  3. ​static {​

  4. ​try {​

  5. ​getUnsafe = Unsafe.class.getDeclaredField("theUnsafe");​

  6. ​getUnsafe.setAccessible(true);​

  7. ​unsafe = (Unsafe) getUnsafe.get(null);​

  8. ​} catch (NoSuchFieldException e) {​

  9. ​e.printStackTrace();​

  10. ​} catch (IllegalAccessException e) {​

  11. ​e.printStackTrace();​

  12. ​}​

  13. ​}​

随便只要你高兴,都可以获取到unfase,因为涉及到unfase 的权限问题,所以,我们只能使用这种方式获取,不然就是权限异常,

操作方法:

  1. ​/**​

  2. ​* 操作数组:​

  3. ​* 可以获取数组的在内容中的基本偏移量(arrayBaseOffset),获取数组内元素的间隔(比例),​

  4. ​* 根据数组对象和偏移量获取元素值(getObject),设置数组元素值(putObject),示例如下。​

  5. ​*/​

  6. ​String[] strings = new String[]{"1", "2", "3"};​

  7. ​long i = unsafe.arrayBaseOffset(String[].class);​

  8. ​System.out.println("string[] base offset is :" + i);​

  9. ​//every index scale​

  10. ​long scale = unsafe.arrayIndexScale(String[].class);​

  11. ​System.out.println("string[] index scale is " + scale);​

  12. ​//print first string in strings[]​

  13. ​System.out.println("first element is :" + unsafe.getObject(strings, i));​

  14. ​//set 100 to first string​

  15. ​unsafe.putObject(strings, i + scale * 0, "100");​

  16. ​//print first string in strings[] again​

  17. ​System.out.println("after set ,first element is :" + unsafe.getObject(strings, i + scale * 0));​

  1. ​/**​

  2. ​* 对象操作​

  3. ​* 实例化Data​

  4. ​*​

  5. ​* 可以通过类的class对象创建类对象(allocateInstance),获取对象属性的偏移量(objectFieldOffset)​

  6. ​* ,通过偏移量设置对象的值(putObject)​

  7. ​*​

  8. ​* 对象的反序列化​

  9. ​* 当使用框架反序列化或者构建对象时,会假设从已存在的对象中重建,你期望使用反射来调用类的设置函数,​

  10. ​* 或者更准确一点是能直接设置内部字段甚至是final字段的函数。问题是你想创建一个对象的实例,​

  11. ​* 但你实际上又不需要构造函数,因为它可能会使问题更加困难而且会有副作用。​

  12. ​*​

  13. ​*/​

  14. ​//调用allocateInstance函数避免了在我们不需要构造函数的时候却调用它​

  15. ​Data data = (Data) unsafe.allocateInstance(Data.class);​

  16. ​data.setId(1L);​

  17. ​data.setName("unsafe");​

  18. ​System.out.println(data);​

  19. ​//返回成员属性在内存中的地址相对于对象内存地址的偏移量​

  20. ​Field nameField = Data.class.getDeclaredField("name");​

  21. ​long fieldOffset = unsafe.objectFieldOffset(nameField);​

  22. ​//putLong,putInt,putDouble,putChar,putObject等方法,直接修改内存数据(可以越过访问权限)​

  23. ​unsafe.putObject(data,fieldOffset,"这是新的值");​

  24. ​System.out.println(data.getName());​

  25. ​/**​

  26. ​* 我们可以在运行时创建一个类,比如从已编译的.class文件中。将类内容读取为字节数组,​

  27. ​* 并正确地传递给defineClass方法;当你必须动态创建类,而现有代码中有一些代理, 这是很有用的​

  28. ​*/​

  29. ​File file = new File("C:\\workspace\\idea2\\disruptor\\target\\classes\\com\\onyx\\distruptor\\test\\Data.class");​

  30. ​FileInputStream input = new FileInputStream(file);​

  31. ​byte[] content = new byte[(int)file.length()];​

  32. ​input.read(content);​

  33. ​Class c = unsafe.defineClass(null, content, 0, content.length,null,null);​

  34. ​c.getMethod("getId").invoke(c.newInstance(), null);​

  35. ​/**​

  36. ​* 内存操作​

  37. ​* 可以在Java内存区域中分配内存(allocateMemory),设置内存(setMemory,用于初始化),​

  38. ​* 在指定的内存位置中设置值(putInt\putBoolean\putDouble等基本类型)​

  39. ​*/​

  40. ​//分配一个8byte的内存​

  41. ​long address = unsafe.allocateMemory(8L);​

  42. ​//初始化内存填充1​

  43. ​unsafe.setMemory(address, 8L, (byte) 1);​

  44. ​//测试输出​

  45. ​System.out.println("add byte to memory:" + unsafe.getInt(address));​

  46. ​//设置0-3 4个byte为0x7fffffff​

  47. ​unsafe.putInt(address, 0x7fffffff);​

  48. ​//设置4-7 4个byte为0x80000000​

  49. ​unsafe.putInt(address + 4, 0x80000000);​

  50. ​//int占用4byte​

  51. ​System.out.println("add byte to memory:" + unsafe.getInt(address));​

  52. ​System.out.println("add byte to memory:" + unsafe.getInt(address + 4));​

  1. ​/**​

  2. ​* CAS操作​

  3. ​* Compare And Swap(比较并交换),当需要改变的值为期望的值时,那么就替换它为新的值,是原子​

  4. ​* (不可在分割)的操作。很多并发框架底层都用到了CAS操作,CAS操作优势是无锁,可以减少线程切换耗费​

  5. ​* 的时间,但CAS经常失败运行容易引起性能问题,也存在ABA问题。在Unsafe中包含compareAndSwapObject、​

  6. ​* compareAndSwapInt、compareAndSwapLong三个方法,compareAndSwapInt的简单示例如下。​

  7. ​*/​

  8. ​Data data = new Data();​

  9. ​data.setId(1L);​

  10. ​Field id = data.getClass().getDeclaredField("id");​

  11. ​long l = unsafe.objectFieldOffset(id);​

  12. ​id.setAccessible(true);​

  13. ​//比较并交换,比如id的值如果是所期望的值1,那么就替换为2,否则不做处理​

  14. ​unsafe.compareAndSwapLong(data,1L,1L,2L);​

  15. ​System.out.println(data.getId());​

  1. ​/**​

  2. ​* 常量获取​

  3. ​*​

  4. ​* 可以获取地址大小(addressSize),页大小(pageSize),基本类型数组的偏移量​

  5. ​* (Unsafe.ARRAY_INT_BASE_OFFSET\Unsafe.ARRAY_BOOLEAN_BASE_OFFSET等)、​

  6. ​* 基本类型数组内元素的间隔(Unsafe.ARRAY_INT_INDEX_SCALE\Unsafe.ARRAY_BOOLEAN_INDEX_SCALE等)​

  7. ​*/​

  8. ​//get os address size​

  9. ​System.out.println("address size is :" + unsafe.addressSize());​

  10. ​//get os page size​

  11. ​System.out.println("page size is :" + unsafe.pageSize());​

  12. ​//int array base offset​

  13. ​System.out.println("unsafe array int base offset:" + Unsafe.ARRAY_INT_BASE_OFFSET);​

  14. ​/**​

  15. ​* 线程许可​

  16. ​* 许可线程通过(park),或者让线程等待许可(unpark),​

  17. ​*/​

  18. ​Thread packThread = new Thread(() -> {​

  19. ​long startTime = System.currentTimeMillis();​

  20. ​//纳秒,相对时间park​

  21. ​unsafe.park(false,3000000000L);​

  22. ​//毫秒,绝对时间park​

  23. ​//unsafe.park(true,System.currentTimeMillis()+3000);​

  24. ​System.out.println("main thread end,cost :"+(System.currentTimeMillis()-startTime)+"ms");​

  25. ​});​

  26. ​packThread.start();​

  27. ​TimeUnit.SECONDS.sleep(1);​

  28. ​//注释掉下一行后,线程3秒数后进行输出,否则在1秒后输出​

  29. ​unsafe.unpark(packThread);​

  1. ​/**​

  2. ​* Java数组大小的最大值为Integer.MAX_VALUE。使用直接内存分配,我们创建的数组大小受限于堆大小;​

  3. ​* 实际上,这是堆外内存(off-heap memory)技术,在java.nio包中部分可用;​

  4. ​*​

  5. ​* 这种方式的内存分配不在堆上,且不受GC管理,所以必须小心Unsafe.freeMemory()的使用。​

  6. ​* 它也不执行任何边界检查,所以任何非法访问可能会导致JVM崩溃​

  7. ​*/​

  8. ​public class SuperArray {​

  9. ​private static Unsafe unsafe = null;​

  10. ​private static Field getUnsafe = null;​

  11. ​static {​

  12. ​try {​

  13. ​getUnsafe = Unsafe.class.getDeclaredField("theUnsafe");​

  14. ​getUnsafe.setAccessible(true);​

  15. ​unsafe = (Unsafe) getUnsafe.get(null);​

  16. ​} catch (NoSuchFieldException e) {​

  17. ​e.printStackTrace();​

  18. ​} catch (IllegalAccessException e) {​

  19. ​e.printStackTrace();​

  20. ​}​

  21. ​}​

  22. ​private final static int BYTE = 1;​

  23. ​private long size;​

  24. ​private long address;​

  25. ​public SuperArray(long size) {​

  26. ​this.size = size;​

  27. ​address = unsafe.allocateMemory(size * BYTE);​

  28. ​}​

  29. ​public void set(long i, byte value) {​

  30. ​unsafe.putByte(address + i * BYTE, value);​

  31. ​}​

  32. ​public int get(long idx) {​

  33. ​return unsafe.getByte(address + idx * BYTE);​

  34. ​}​

  35. ​public long size() {​

  36. ​return size;​

  37. ​}​

  38. ​public static void main(String[] args) {​

  39. ​long SUPER_SIZE = (long)Integer.MAX_VALUE * 2;​

  40. ​SuperArray array = new SuperArray(SUPER_SIZE);​

  41. ​System.out.println("Array size:" + array.size()); // 4294967294​

  42. ​int sum=0;​

  43. ​for (int i = 0; i < 100; i++) {​

  44. ​array.set((long)Integer.MAX_VALUE + i, (byte)3);​

  45. ​sum += array.get((long)Integer.MAX_VALUE + i);​

  46. ​}​

  47. ​System.out.println(sum);​

  48. ​}​

  49. ​}​