Java中ThreadLocal執行緒變數的實現原理

2022-06-26 14:01:02

ThreadLocal是什麼?

ThreadLocal 使得我們可以建立執行緒私有的變數, 這個變數相對於其他執行緒來說是不可見的,ThreadLocal為變數在每個執行緒中都建立了一個副本 , 每個執行緒可以存取自己私有的執行緒變數,程式碼範例如下 :

public class ThreadLocalDemo {
 
    //建立一個ThreadLocal物件,用來為每個執行緒會複製儲存一份變數,實現執行緒封閉
    private  static ThreadLocal<Integer> localNum = new ThreadLocal<Integer>(){
        @Override
        protected Integer initialValue() {
            return 1;
        }
    };
 
    public static void main(String[] args) {
          //執行緒0
          new Thread(){
              @Override
              public void run() {
                   localNum.set(1);
                  try {
                      Thread.sleep(2000);
                  } catch (InterruptedException e) {
                      e.printStackTrace();
                  }
                  localNum.set(localNum.get()+10);
                  System.out.println(Thread.currentThread().getName()+":"+localNum.get());//11
              }
          }.start();
        //執行緒1
        new Thread(){
            @Override
            public void run() {
                localNum.set(3);
                try {
                    Thread.sleep(2000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                localNum.set(localNum.get()+20);
                System.out.println(Thread.currentThread().getName()+":"+localNum.get());//23
            }
        }.start();
        
        System.out.println(Thread.currentThread().getName()+":"+localNum.get());//0
    }
}

如上所述, 算上main執行緒與新建的兩個執行緒 ,總共三個執行緒 , 每個執行緒都包含自己的私有變數,此處我們設定值1 , set() 和 get() 方法用來設定值和獲得值, 執行結果如下 :

ThreadLocal實現原理分析

ThreadLocal是一個泛型類 , 可以接受任何型別的物件 , 其內部維護了一個ThreadLocalMap 的靜態內部類, 我們使用的 get(), set()等其實都來自這個類, 每次都會為當前執行緒建立一個ThreadLocalMap物件, 用來記錄私有的值

先看 set() 方法

public void set(T value) {
    //拿到當前執行緒
    Thread t = Thread.currentThread();
    //拿到當前執行緒map
    ThreadLocalMap map = getMap(t);
    if (map != null)
        //存在設定值
        map.set(this, value);
    else
        //不存在則建立
        createMap(t, value); 
}

void createMap(Thread t, T firstValue) {
    //threadLocals屬性即為此map
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

接著是get() 方法

public T get() {
    //拿到當前執行緒
    Thread t = Thread.currentThread();
    //拿到當前執行緒對應的map
    ThreadLocalMap map = getMap(t);
    //如果已有map
    if (map != null) {
        //取值操作, 拿到對應的Entry
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
           @SuppressWarnings("unchecked")
           T result = (T)e.value;
           return result;
        }
    }
    //沒有map, 則去建立初始化一個map
    return setInitialValue();
}

private T setInitialValue() {
    //initialValue()方法返回的value為null
    T value = initialValue();
    //拿到當前執行緒去建立對應的map
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
    return value;
}

ThreadLocal可以理解為對ThreadLocalMap的封裝

ThreadLocal記憶體漏失問題

在ThreadLocalMap中 , 使用 ThreadLocal 的弱參照作為 key

這樣的話, 如果一個ThreadLocal不存在外部強參照時, 那麼key註定要被GC回收 , 這樣導致ThreadLocalMap 中key為null , 而value還存在著強參照鏈

一個執行緒可以同時擁有多個ThreadLocal, 如果作為弱參照的key被回收後, value還不能被回收,那麼這就導致此ThreadLocal的生命週期和此執行緒是一樣長的(因為執行緒執行完畢後此value的強參照鏈才會斷), 如果執行緒一直不結束, 堆積的value也一直無法被回收, 那麼就會產生記憶體漏失問題

這裡解決問題的方式是 : 每次使用完ThreadLocal後都呼叫它的remove()方法清除資料

public void remove() {
    ThreadLocalMap m = getMap(Thread.currentThread());
    if (m != null)
       m.remove(this);
}

這裡我們再來看一下key作為強弱參照的區別

如果key作為強參照, 那麼它的生命週期和執行緒一樣長,存在穩定的強參照鏈,無法被回收,產生記憶體漏失問題, 而如果作為弱參照, GC則會自動的去回收它們, 在後續的remove()方法中也可以更好的去回收value , 所以我們一般將ThreadLocal設計成 private static 的, 在使用完後用remove()方法去手動刪除它們

到此這篇關於Java中ThreadLocal執行緒變數的實現的文章就介紹到這了,更多相關Java ThreadLocal執行緒變數內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Java中ThreadLocal執行緒變數的實現原理

目錄

ThreadLocal是什麼?

ThreadLocal實現原理分析

ThreadLocal記憶體漏失問題

熱門文章