Java并發(fā)之ThreadLocal使用和源碼分析

本篇文章首先給出了ThreadLocal常用的方法采够,并舉例來說明ThreadLocal的使用。然后分析了ThreadLocal的源代碼中的get()方法冰垄,初始化方法蹬癌,ThreadLocalMap的實現(xiàn)等來理解ThreadLocal的原理。

ThreadLocal為每個線程提供一個獨立的對象副本虹茶,每個線程可以單獨訪問自己獨有的對象逝薪,不存在多線程同時訪問一個對象時的共享問題。

一.使用ThreadLocal

ThreadLocal是一個泛型類蝴罪,使用時要指定持有的對象的類型董济。一般會重寫initialValue()方法,來設置線程第一次通過ThreadLocal獲取對象時的初始值要门。

常用方法:

\\1.get()方法 獲取對象引用虏肾,每個線程獲取到的不同
public T get(); 
\\2.set()方法廓啊,修改當前線程在ThreadLocal中存儲的值
public void set(T value);
\\3.initialValue()方法,設置初始值封豪,默認返回null崖瞭,一般被重寫來設置初始值
protected T initialValue();

具體例子

public class ThreadLocalDemo {
    static ThreadLocal<String> threadLocal = new ThreadLocal<String>(){
        @Override
        protected String initialValue() {
            return Thread.currentThread().getName();
        }
    };

    public static void main(String[] args) throws Exception{
        Thread thread1 = new Thread(new Runnable() {
            public void run() {
                System.out.println(threadLocal.get());
            }
        });

        thread1.start();
        thread1.join();
        System.out.println(threadLocal.get());
    }
}

輸出結果:

Thread-0
main

可以看到,主線程和thread1通過threadLocal變量分別持有了各自的String對象撑毛。

二.源碼分析

get方法的主要流程
我們從get方法入手书聚,看看ThreadLocal的實現(xiàn) 原理,先po上源代碼:

public T get() {
    //先獲取當前線程
    Thread t = Thread.currentThread();   
    //獲取當前線程的ThreadLocalMap
    ThreadLocalMap map = getMap(t);   
    if (map != null) {
        //從ThreadLocalMap中獲取到要的值藻雌,注意這里使用的是當前的ThreadLocal對象作為key的
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    //當前線程的ThreadLocalMap為null雌续,或者map里沒有保存和當前ThreadLocal相關的值時
    return setInitialValue();                   
}

從代碼中可以總結出TheadLocal.get的流程:

  • 先獲取到當前的線程t,
  • 獲取到t中保存的ThreaaLocalMap胯杭,如果沒有則轉到最后一步
  • 以當前ThreadLocal作為鍵值驯杜,從上述map中獲取到要的變量
  • 如果獲取到變量為null,則轉到最后一步進行初始化
  • 初始化thread的map或保存的變量值

查看getMap(t):從線程中獲取ThreadLocalMap的代碼發(fā)現(xiàn):Thread類中有一個名為threadLocals的ThreadLocalMap類型的變量做个,初始值為null鸽心,在第一次ThreadLocal.get時初始化。getMap會獲取這個變量居暖。

setInitialValue方法分析
下面再分析 一下setInitialValue()

private T setInitialValue() {
    //獲取初始值顽频,這里調(diào)用的方法常常被重寫,來設置初始值太闺。默認返回Null
    T value = initialValue();
    //獲取當前線程
    Thread t = Thread.currentThread();
    //先嘗試獲取map糯景,因為也不知道是因為沒有map還是因為map中沒有值才進來這個方法的
    ThreadLocalMap map = getMap(t);
    //map存在,放入value值省骂,注意這里的key是當前ThreadLocal對象蟀淮;和get中的map.getEntry相呼應
    if (map != null)
        map.set(this, value);
    else
        //map不存在,創(chuàng)建map
        createMap(t, value);
    return value;
}

看一下createMap的代碼:

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

直接調(diào)用了ThreadLocalMap的構造方法钞澳,將其初始化并返回怠惶。這個構造方法不再細究了,只要記住轧粟,傳入的兩個參數(shù)是用來作為初始化后的第一個鍵值對存入該map的即可策治。
總結一下setInitialValue()方法:

  • 獲取value的初始化值,從initialValue方法逃延,這里就看出initialValue的作用了
  • 如果Thread有ThreadLocalMap览妖,則直接使用轧拄,存入當前ThreadLocal->value的鍵值對
  • 如果沒有map揽祥,則創(chuàng)建map,并同時存入鍵值對
  • 返回value

ThreadLocalMap簡析

ThreadLocalMap是ThreadLocal的一個內(nèi)部類檩电。在每一個Thread都持有一個ThreadLocalMap變量(如果該線程沒用的ThreadLocal則為null)拄丰。它是一個map府树,key是ThreadLocal類型,value是保存的值料按。

這樣設計就保證了一個Thread可以存儲多個ThreadLocal奄侠。

ThreadLocalMap的Entry設計:
一個map中肯定有一個Entry來存儲鍵值對,ThreadLocal的Entry的設計的較為特殊:

static class Entry extends WeakReference<ThreadLocal<?>> {
    /** The value associated with this ThreadLocal. */
    Object value;

    Entry(ThreadLocal<?> k, Object v) {
        super(k);
        value = v;
    }
}

可以看到它繼承了WeakReference载矿。我們知道WeakReference是四大引用的一種叫做弱引用垄潮,它的含義是當對象只被弱引用引用時,下次GC時該對象就會被回收闷盔。

Q1:ThreadLocalMap的Entry為什么繼承WeekReference?
(個人理解)這樣當ThreadLocal變量生命周期結束后弯洗,對應的線程的ThreadLocalMap中存儲的相應鍵值對也會被回收,不會造成內(nèi)存泄露逢勾。

參考資料

本文是Java并發(fā)專題(歡迎大家關注)的一篇牡整。
以下是完整的目錄:
Java并發(fā)之基礎知識
Java并發(fā)之volatile關鍵字
Java并發(fā)之synchronized關鍵字
Java并發(fā)之原子類
Java并發(fā)之線程池
Java并發(fā)之并發(fā)工具類
Java并發(fā)之AQS原理
Java并發(fā)之ThreadLocal使用和源碼分析


最后編輯于
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市溺拱,隨后出現(xiàn)的幾起案子逃贝,更是在濱河造成了極大的恐慌,老刑警劉巖迫摔,帶你破解...
    沈念sama閱讀 206,311評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件沐扳,死亡現(xiàn)場離奇詭異,居然都是意外死亡句占,警方通過查閱死者的電腦和手機迫皱,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,339評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來辖众,“玉大人卓起,你說我怎么就攤上這事“颊ǎ” “怎么了戏阅?”我有些...
    開封第一講書人閱讀 152,671評論 0 342
  • 文/不壞的土叔 我叫張陵,是天一觀的道長啤它。 經(jīng)常有香客問我奕筐,道長,這世上最難降的妖魔是什么变骡? 我笑而不...
    開封第一講書人閱讀 55,252評論 1 279
  • 正文 為了忘掉前任离赫,我火速辦了婚禮,結果婚禮上塌碌,老公的妹妹穿的比我還像新娘渊胸。我一直安慰自己,他們只是感情好台妆,可當我...
    茶點故事閱讀 64,253評論 5 371
  • 文/花漫 我一把揭開白布翎猛。 她就那樣靜靜地躺著胖翰,像睡著了一般。 火紅的嫁衣襯著肌膚如雪切厘。 梳的紋絲不亂的頭發(fā)上萨咳,一...
    開封第一講書人閱讀 49,031評論 1 285
  • 那天,我揣著相機與錄音疫稿,去河邊找鬼培他。 笑死,一個胖子當著我的面吹牛遗座,可吹牛的內(nèi)容都是我干的靶壮。 我是一名探鬼主播,決...
    沈念sama閱讀 38,340評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼员萍,長吁一口氣:“原來是場噩夢啊……” “哼腾降!你這毒婦竟也來了?” 一聲冷哼從身側響起碎绎,我...
    開封第一講書人閱讀 36,973評論 0 259
  • 序言:老撾萬榮一對情侶失蹤螃壤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后筋帖,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體奸晴,經(jīng)...
    沈念sama閱讀 43,466評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,937評論 2 323
  • 正文 我和宋清朗相戀三年日麸,在試婚紗的時候發(fā)現(xiàn)自己被綠了寄啼。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,039評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡代箭,死狀恐怖墩划,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情嗡综,我是刑警寧澤乙帮,帶...
    沈念sama閱讀 33,701評論 4 323
  • 正文 年R本政府宣布,位于F島的核電站极景,受9級特大地震影響察净,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜盼樟,卻給世界環(huán)境...
    茶點故事閱讀 39,254評論 3 307
  • 文/蒙蒙 一氢卡、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧晨缴,春花似錦译秦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,259評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至延都,卻和暖如春雷猪,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背晰房。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評論 1 262
  • 我被黑心中介騙來泰國打工求摇, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人殊者。 一個月前我還...
    沈念sama閱讀 45,497評論 2 354
  • 正文 我出身青樓与境,卻偏偏與公主長得像,于是被迫代替她去往敵國和親猖吴。 傳聞我的和親對象是個殘疾皇子摔刁,可洞房花燭夜當晚...
    茶點故事閱讀 42,786評論 2 345

推薦閱讀更多精彩內(nèi)容