1. 程式人生 > 程式設計 >Java Unsafe類實現原理及測試程式碼

Java Unsafe類實現原理及測試程式碼

Unsafe類介紹

第一次看到這個類時被它的名字嚇到了,居然還有一個類自名Unsafe?讀完本文,大家也能發現Unsafe類確實有點不那麼安全,它能實現一些不那麼常見的功能。

Unsafe類使Java擁有了像C語言的指標一樣操作記憶體空間的能力,同時也帶來了指標的問題。過度的使用Unsafe類會使得出錯的機率變大,因此Java官方並不建議使用的,官方文件也幾乎沒有。Oracle正在計劃從Java 9中去掉Unsafe類,如果真是如此影響就太大了。

Unsafe類提供了以下這些功能:

一、記憶體管理。包括分配記憶體、釋放記憶體等。

該部分包括了allocateMemory(分配記憶體)、reallocateMemory(重新分配記憶體)、copyMemory(拷貝記憶體)、freeMemory(釋放記憶體 )、getAddress(獲取記憶體地址)、addressSize、pageSize、getInt(獲取記憶體地址指向的整數)、getIntVolatile(獲取記憶體地址指向的整數,並支援volatile語義)、putInt(將整數寫入指定記憶體地址)、putIntVolatile(將整數寫入指定記憶體地址,並支援volatile語義)、putOrderedInt(將整數寫入指定記憶體地址、有序或者有延遲的方法)等方法。getXXX和putXXX包含了各種基本型別的操作。

利用copyMemory方法,我們可以實現一個通用的物件拷貝方法,無需再對每一個物件都實現clone方法,當然這通用的方法只能做到物件淺拷貝。

二、非常規的物件例項化。

allocateInstance()方法提供了另一種建立例項的途徑。通常我們可以用new或者反射來例項化物件,使用allocateInstance()方法可以直接生成物件例項,且無需呼叫構造方法和其它初始化方法。

這在物件反序列化的時候會很有用,能夠重建和設定final欄位,而不需要呼叫構造方法。

三、操作類、物件、變數。

這部分包括了staticFieldOffset(靜態域偏移)、defineClass(定義類)、defineAnonymousClass(定義匿名類)、ensureClassInitialized(確保類初始化)、objectFieldOffset(物件域偏移)等方法。

通過這些方法我們可以獲取物件的指標,通過對指標進行偏移,我們不僅可以直接修改指標指向的資料(即使它們是私有的),甚至可以找到JVM已經認定為垃圾、可以進行回收的物件。

四、陣列操作。

這部分包括了arrayBaseOffset(獲取陣列第一個元素的偏移地址)、arrayIndexScale(獲取陣列中元素的增量地址)等方法。arrayBaseOffset與arrayIndexScale配合起來使用,就可以定位陣列中每個元素在記憶體中的位置。

由於Java的陣列最大值為Integer.MAX_VALUE,使用Unsafe類的記憶體分配方法可以實現超大陣列。實際上這樣的資料就可以認為是C陣列,因此需要注意在合適的時間釋放記憶體。

五、多執行緒同步。包括鎖機制、CAS操作等。

這部分包括了monitorEnter、tryMonitorEnter、monitorExit、compareAndSwapInt、compareAndSwap等方法。

其中monitorEnter、tryMonitorEnter、monitorExit已經被標記為deprecated,不建議使用。

Unsafe類的CAS操作可能是用的最多的,它為Java的鎖機制提供了一種新的解決辦法,比如AtomicInteger等類都是通過該方法來實現的。compareAndSwap方法是原子的,可以避免繁重的鎖機制,提高程式碼效率。這是一種樂觀鎖,通常認為在大部分情況下不出現競態條件,如果操作失敗,會不斷重試直到成功。

六、掛起與恢復。

這部分包括了park、unpark等方法。

將一個執行緒進行掛起是通過park方法實現的,呼叫 park後,執行緒將一直阻塞直到超時或者中斷等條件出現。unpark可以終止一個掛起的執行緒,使其恢復正常。整個併發框架中對執行緒的掛起操作被封裝在 LockSupport類中,LockSupport類中有各種版本pack方法,但最終都呼叫了Unsafe.park()方法。

七、記憶體屏障。

這部分包括了loadFence、storeFence、fullFence等方法。這是在Java 8新引入的,用於定義記憶體屏障,避免程式碼重排序。

loadFence() 表示該方法之前的所有load操作在記憶體屏障之前完成。同理storeFence()表示該方法之前的所有store操作在記憶體屏障之前完成。fullFence()表示該方法之前的所有load、store操作在記憶體屏障之前完成。

測試程式碼

import com.User;
import org.junit.Before;
import org.junit.Test;
import sun.misc.Unsafe;
 
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
 
class User {
 
  public static String USER_CLASS_NAME = "User.class";
  private int age;
  private String name;
 
  public int getAge() {
    return age;
  }
 
  public String getName() {
    return name;
  }
 
  public User(int age,String name) {
    this.age = age;
    this.name = name;
  }
 
  public void setAge(int age) {
    this.age = age;
  }
 
  public void setName(String name) {
    this.name = name;
  }
}
 
public class LockTests {
 
  Unsafe unSafe;
  User u = new User(17,"zhangsan");
 
  @Before
  public void before() throws Exception {
    Field theUnsafeField = Unsafe.class.getDeclaredField("theUnsafe");
    theUnsafeField.setAccessible(true);
    unSafe = (Unsafe) theUnsafeField.get(Unsafe.class);
  }
 
  @Test
  public void objectFieldOffset() throws Exception {
    // unSafe偏底層的一個Java工具類
    java.util.List users = new ArrayList();
    for (int i = 0; i < 10; i++) {
      Field ageField = User.class.getDeclaredField("age");
      User u = new User(18,"daxin");
      users.add(u);
      //使用記憶體獲取User age欄位在記憶體中的 offset
      // 是相對地址,不是一個絕對地址
      long ageOffset = unSafe.objectFieldOffset(ageField);
      // 每次都相同
      System.out.println("ageOffset = " + ageOffset);
    }
  }
 
  @Test
  public void compareAndSwapInt() throws Exception {
 
    // unSafe偏底層的一個Java工具類
    Field ageField = User.class.getDeclaredField("age");
 
    User u = new User(18,"daxin");
 
    //使用記憶體獲取User age欄位在記憶體中的 offset
    long ageOffset = unSafe.objectFieldOffset(ageField);
 
    // 修改之前的值
    System.out.println(u.getAge());
    // 進行CAS更新,由於設定18 因此CAS 會成功
    unSafe.compareAndSwapInt(u,ageOffset,18,20);
    System.out.println(u.getAge());
 
    // 由於age設定20 進行CAS失敗
    unSafe.compareAndSwapInt(u,22);
    System.out.println(u.getAge());
 
  }
 
  @Test
  public void ensureClassInitialized() {
    System.out.println("==== start ====");
    unSafe.ensureClassInitialized(ClassIsLoad.class);
    // 再次 確認不會報錯
    unSafe.ensureClassInitialized(ClassIsLoad.class);
  }
 
  /**
   * AQS 底層的Node連結串列就是基於這個工具實現的 。
   *
   * @throws Exception
   */
  @Test
  public void getValueByFieldOffset() throws Exception {
    for (int i = 0; i < 10; i++) {
      User u = new User(18,UUID.randomUUID().toString().substring(i,20));
      int age = unSafe.getInt(u,12L);
      System.out.println("age = " + age);
 
      // 獲取名字 field offset
      Field nameField = User.class.getDeclaredField("name");
      long nameOffset = unSafe.objectFieldOffset(nameField);
      System.out.println("nameOffset = " + nameOffset);
      String name = unSafe.getObject(u,nameOffset) + "";
      System.out.println("name = " + name);
    }
  }
 
  @Test
  public void pageSize() {
    System.out.println("unSafe.pageSize() = " + unSafe.pageSize());
  }
 
  /**
   * AtomicInteger 底層是基於getAndAddInt實現
   */
  @Test
  public void getAndAddInt() throws InterruptedException {
 
    User u = new User(17,"zhangsan");
    CountDownLatch downLatch = new CountDownLatch(10);
    System.out.println("u.getAge() = " + u.getAge());
    for (int i = 0; i < 10; i++) {
 
      new Thread(new Runnable() {
        @Override
        public void run() {
          downLatch.countDown();
          int val = unSafe.getAndAddInt(u,12L,1);
          System.out.println(Thread.currentThread().getName() + " val = " + val);
        }
      }).start();
 
    }
    Thread.sleep(5000);
    System.out.println("u.getAge() = " + u.getAge());
  }
 
  @Test
  public void getAndSetInt() throws InterruptedException {
    User u = new User(17,"zhangsan");
    CountDownLatch downLatch = new CountDownLatch(10);
    System.out.println("u.getAge() = " + u.getAge());
    for (int i = 0; i < 10; i++) {
 
      new Thread(new Runnable() {
        @Override
        public void run() {
          downLatch.countDown();
          int val = unSafe.getAndSetInt(u,10);
          System.out.println(Thread.currentThread().getName() + " val = " + val);
        }
      }).start();
 
    }
    Thread.sleep(5000);
    System.out.println("u.getAge() = " + u.getAge());
  }
 
 
  @Test
  public void getIntVolatile() {
 
    for (int i = 0; i < 10; i++) {
      u.setAge(i);
      /**
       * @param obj  the object containing the field to modify.
       * @param offset the offset of the integer field within <code>obj</code>.
       * @return
       */
      int age = unSafe.getIntVolatile(u,12L);
      System.out.println("age = " + age);
    }
  }
 
  // 系統負載取樣的介面
  @Test
  public void getLoadAverage() {
    double[] nums = new double[8];
    int val = unSafe.getLoadAverage(nums,8);
    System.out.println(val);
  }
 
 
  /**
   * //記憶體屏障,禁止load操作重排序。屏障前的load操作不能被重排序到屏障後,屏障後的load操作不能被重排序到屏障前
   * public native void loadFence();
   * <p>
   * <p>
   * 參見:https://tech.meituan.com/2019/02/14/talk-about-java-magic-class-unsafe.html
   */
 
  @Test
  public void loadFence() {
    //java.util.concurrent.locks.StampedLock.validate
    unSafe.loadFence();
  }
 
  /**
   * //記憶體屏障,禁止store操作重排序。屏障前的store操作不能被重排序到屏障後,屏障後的store操作不能被重排序到屏障前
   * public native void storeFence();
   * 參見:https://tech.meituan.com/2019/02/14/talk-about-java-magic-class-unsafe.html
   */
  @Test
  public void storeFence() {
  }
 
  /**
   * //記憶體屏障,禁止load、store操作重排序
   * public native void fullFence();
   * 參見:https://tech.meituan.com/2019/02/14/talk-about-java-magic-class-unsafe.html
   */
  @Test
  public void fullFence() {
  }
 
 
  @Test
  public void shouldBeInitialized() {
    boolean shouldBeInitialized = unSafe.shouldBeInitialized(String.class);
    System.out.println(shouldBeInitialized);
    shouldBeInitialized = unSafe.shouldBeInitialized(User.class);
    System.out.println(shouldBeInitialized);
  }
 
  /**
   * synchronized 的一種實現獲取鎖
   *
   * @throws InterruptedException
   */
  @Test
  public void monitorEnter() throws InterruptedException {
 
    unSafe.monitorEnter(u);
    new Thread(new Runnable() {
      @Override
      public void run() {
        synchronized (u) {
          System.out.println("==u lock got ==");
        }
      }
    }).start();
 
    Thread.sleep(2000);
    unSafe.monitorExit(u);
  }
 
  @Test
  public void compareAndSwap() {
//    unSafe.compareAndSwapInt(物件,物件中的欄位偏移,期望值,設定值)
//    unSafe.compareAndSwapLong(物件,設定值)
//    unSafe.compareAndSwapObject(物件,設定值)
  }
 
  @Test
  public void t() {
    // 方法簽名
    // public void copyMemory(Object srcBase,long srcOffset,Object destBase,long destOffset,long bytes)
    // unSafe.copyMemory();
  }
 
 
}
 
 
class ClassIsLoad {
 
  static {
    System.out.println("ClassIsLoad class Is Load !");
  }
}

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。