1. 程式人生 > >Java併發(二十):執行緒本地變數ThreadLocal

Java併發(二十):執行緒本地變數ThreadLocal

ThreadLocal是一個本地執行緒副本變數工具類。

主要用於將私有執行緒和該執行緒存放的副本物件做一個對映,各個執行緒之間的變數互不干擾,在高併發場景下,可以實現無狀態的呼叫,特別適用於各個執行緒依賴不同的變數值完成操作的場景。

一、ThreadLocal的核心機制

每個Thread執行緒內部都有一個Map,Tread類的ThreadLocal.ThreadLocalMap屬性

Map裡面儲存執行緒本地物件(key也就是當前的ThreadLoacal物件)和執行緒的變數副本(value)

Thread內部的Map是由ThreadLocal維護的,由ThreadLocal負責向map獲取和設定執行緒的變數值

資料結構:

二、ThreadLocal原始碼分析

 ThreadLocal核心方法:

  • get():返回此執行緒區域性變數的當前執行緒副本中的值。
  • initialValue():返回此執行緒區域性變數的當前執行緒的“初始值”。
  • remove():移除此執行緒區域性變數當前執行緒的值。
  • set(T value):將此執行緒區域性變數的當前執行緒副本中的值設定為指定值。

內部類 ThreadLocalMap:

      static class Entry extends WeakReference<ThreadLocal<?>> {
          /** The value associated with this ThreadLocal. 
*/ Object value; Entry(ThreadLocal<?> k, Object v) { super(k); value = v; } }

Entry繼承自WeakReference(弱引用,生命週期只能存活到下次GC前),但只有Key是弱引用型別的,Value並非弱引用。

ThreadLocalMap的set()方法:

private void set(ThreadLocal<?> key, Object value) {

        ThreadLocal.ThreadLocalMap.Entry[] tab 
= table; int len = tab.length; // 根據 ThreadLocal 的雜湊值,查詢對應元素在陣列中的位置 int i = key.threadLocalHashCode & (len-1); // 採用“線性探測法”,尋找合適位置 for (ThreadLocal.ThreadLocalMap.Entry e = tab[i]; e != null; e = tab[i = nextIndex(i, len)]) { ThreadLocal<?> k = e.get(); // key 存在,直接覆蓋 if (k == key) { e.value = value; return; } // key == null,但是存在值(因為此處的e != null),說明之前的ThreadLocal物件已經被回收了 if (k == null) { // 用新元素替換陳舊的元素 replaceStaleEntry(key, value, i); return; } } // ThreadLocal對應的key例項不存在也沒有陳舊元素,new 一個 tab[i] = new ThreadLocal.ThreadLocalMap.Entry(key, value); int sz = ++size; // cleanSomeSlots 清楚陳舊的Entry(key == null) // 如果沒有清理陳舊的 Entry 並且陣列中的元素大於了閾值,則進行 rehash if (!cleanSomeSlots(i, sz) && sz >= threshold) rehash(); }

ThreadLocalMap中解決Hash衝突的方式並非連結串列的方式,而是採用線性探測的方式,所謂線性探測,就是根據初始key的hashcode值確定元素在table陣列中的位置,如果發現這個位置上已經有其他key值的元素被佔用,則利用固定的演算法尋找一定步長的下個位置,依次判斷,直至找到能夠存放的位置。

ThreadLocalMap解決Hash衝突的方式就是簡單的步長加1或減1,尋找下一個相鄰的位置。

顯然ThreadLocalMap採用線性探測的方式解決Hash衝突的效率很低,如果有大量不同的ThreadLocal物件放入map中時傳送衝突,或者發生二次衝突,則效率很低。

所以這裡引出的建議是:每個執行緒只存一個變數,這樣的話所有的執行緒存放到map中的Key都是相同的ThreadLocal,如果一個執行緒要儲存多個變數,就需要建立多個ThreadLocal,多個ThreadLocal放入Map中時會極大的增加Hash衝突的可能。

get()方法:

步驟:
(1)獲取當前執行緒的ThreadLocalMap物件threadLocals
(2)從map中獲取執行緒儲存的K-V Entry節點。
(3)從Entry節點獲取儲存的Value副本值返回。
(4)map為空的話返回初始值null,即執行緒變數副本為null,在使用時需要注意判斷NullPointerException。
    public T get() {
        // 獲取當前執行緒
        Thread t = Thread.currentThread();

        // 獲取當前執行緒的成員變數 threadLocal
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            // 從當前執行緒的ThreadLocalMap獲取相對應的Entry
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")

                // 獲取目標值        
                T result = (T)e.value;
                return result;
            }
        }
        return setInitialValue();
    }

    ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }

set()方法:

步驟:
(1)獲取當前執行緒的成員變數map
(2)map非空,則重新將ThreadLocal和新的value副本放入到map中。
(3)map空,則對執行緒的成員變數ThreadLocalMap進行初始化建立,並將ThreadLocal和value副本放入map中。
public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}

ThreadLocalMap getMap(Thread t) {
    return t.threadLocals;
}

void createMap(Thread t, T firstValue) {
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

 initialValue()方法:

該方法定義為protected級別且返回為null,很明顯是要子類實現它的,所以我們在使用ThreadLocal的時候一般都應該覆蓋該方法。該方法不能顯示呼叫,只有在第一次呼叫get()或者set()方法時才會被執行,並且僅執行1次。
 protected T initialValue() {
        return null;
    }

三、使用場景

簡單使用示例1:

public class SeqCount {

    private static ThreadLocal<Integer> seqCount = new ThreadLocal<Integer>(){
        // 實現initialValue()
        public Integer initialValue() {
            return 0;
        }
    };

    public int nextSeq(){
        seqCount.set(seqCount.get() + 1);

        return seqCount.get();
    }

    public static void main(String[] args){
        SeqCount seqCount = new SeqCount();

        SeqThread thread1 = new SeqThread(seqCount);
        SeqThread thread2 = new SeqThread(seqCount);
        SeqThread thread3 = new SeqThread(seqCount);
        SeqThread thread4 = new SeqThread(seqCount);

        thread1.start();
        thread2.start();
        thread3.start();
        thread4.start();
    }

    private static class SeqThread extends Thread{
        private SeqCount seqCount;

        SeqThread(SeqCount seqCount){
            this.seqCount = seqCount;
        }

        public void run() {
            for(int i = 0 ; i < 3 ; i++){
                System.out.println(Thread.currentThread().getName() + " seqCount :" + seqCount.nextSeq());
            }
        }
    }
}

 執行結果:

  Thread-1 seqCount :1
  Thread-3 seqCount :1
  Thread-3 seqCount :2
  Thread-3 seqCount :3
  Thread-0 seqCount :1
  Thread-0 seqCount :2
  Thread-0 seqCount :3
  Thread-2 seqCount :1
  Thread-1 seqCount :2
  Thread-1 seqCount :3
  Thread-2 seqCount :2
  Thread-2 seqCount :3

注意:initialValue()方法返回一個物件時,get()和set()方法操作的其實是同一個物件的屬性,不能實現執行緒隔離。

使用場景二:session獲取場景

每個執行緒訪問資料庫都應當是一個獨立的Session會話,如果多個執行緒共享同一個Session會話,有可能其他執行緒關閉連線了,當前執行緒再執行提交時就會出現會話已關閉的異常,導致系統異常。此方式能避免執行緒爭搶Session,提高併發下的安全性。

//獲取Session
public static Session getCurrentSession(){
    Session session =  threadLocal.get();
    //判斷Session是否為空,如果為空,將建立一個session,並設定到本地執行緒變數中
    try {
        if(session ==null&&!session.isOpen()){
            if(sessionFactory==null){
                rbuildSessionFactory();// 建立Hibernate的SessionFactory
            }else{
                session = sessionFactory.openSession();
            }
        }
        threadLocal.set(session);
    } catch (Exception e) {
        // TODO: handle exception
    }

    return session;
}

四、記憶體洩漏問題

ThreadLocalMap使用ThreadLocal的弱引用作為key,如果一個ThreadLocal沒有外部強引用來引用它,那麼系統 GC 的時候,這個ThreadLocal勢必會被回收,這樣一來,ThreadLocalMap中就會出現key為null的Entry,就沒有辦法訪問這些key為null的Entry的value,如果當前執行緒再遲遲不結束的話,這些key為null的Entry的value就會一直存在一條強引用鏈:Thread Ref -> Thread -> ThreaLocalMap -> Entry -> value永遠無法回收,造成記憶體洩漏。

其實,ThreadLocalMap的設計中已經考慮到這種情況,也加上了一些防護措施:在ThreadLocal的get(),set(),remove()的時候都會清除執行緒ThreadLocalMap裡所有key為null的value。

但是這些被動的預防措施並不能保證不會記憶體洩漏:

  使用static的ThreadLocal,延長了ThreadLocal的生命週期,可能導致的記憶體洩漏。
  分配使用了ThreadLocal又不再呼叫get(),set(),remove()方法,那麼就會導致記憶體洩漏。

解決:

每次使用完ThreadLocal,都呼叫它的remove()方法,清除資料。

在使用執行緒池的情況下,沒有及時清理ThreadLocal,不僅是記憶體洩漏的問題,更嚴重的是可能導致業務邏輯出現問題。所以,使用ThreadLocal就跟加鎖完要解鎖一樣,用完就清理。

參考資料 / 相關推薦