揭秘ThreadLocal應(yīng)用

ThreadLocal

ThreadLocal類用來提供線程內(nèi)部的局部變量邻奠,不同的線程之間不會相互干擾套鹅,這種變量在線程的生命周期內(nèi)起作用讹挎,減少同一個線程內(nèi)多個函數(shù)或組件之間一些公共變量的傳遞的復(fù)雜度玫镐。說白了,ThreadLocal就是創(chuàng)建了能夠拿到以自己為key存在當(dāng)前線程的內(nèi)容煤伟,來達到對當(dāng)前線程中的所有方法共享,減少了參數(shù)的多層傳遞木缝。
這么說可能不明顯便锨,我來舉個例子吧:

@Service
public class CommonService {

    public void toWork(User user){
        bus(user);
    }

    public void bus(User user){
        metro(user);
    }

    public void metro(User user){
        System.out.println("到達公司");
    }
}

比如一個接口叫工作,需要調(diào)用去工作的方法我碟,而去工作不管是公交還是地鐵都需要個人信息放案。傳統(tǒng)做法就是如上代碼,將user信息傳遞下去矫俺。這和ThreadLocal有什么關(guān)系呢吱殉?下面來看:

@Service
public class CommonService {
    private ThreadLocal<User> local = new ThreadLocal<>();

    public void toWork(User user){
        local.set(user);
        bus();
    }

    public void bus(){
        User user = local.get();
        metro();
    }

    public void metro(){
        User user = local.get();
        System.out.println("到達公司");
    }
}

這樣在該類中掸冤,user都是可以共享的,如果要跨類友雳,就得將local移動到工具類中稿湿,給不同的類調(diào)用,我就不多加寫了押赊。特別注意饺藤,一個線程對同一個threadlocal來說只能塞入一個value,再次塞入就會更新之前塞入的值考杉。

到這里就會有人問了策精,為什么不將user定義成全部變量呢?
你想想崇棠,每個請求的人都不一樣咽袜,如何定義全局變量。

那ThreadLocal又是如何做到讓一個local變量就set進去就能拿到不同用戶訪問時的用戶信息呢?
這就得從源碼來解釋了枕稀。

public T get() {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        return setInitialValue();
 }
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;
}

每一個請求其實就是一個線程询刹,不管是get還是set都是操作當(dāng)前線程的threadLocals變量,也就是說和當(dāng)前線程threadLocals起到了全局變量的作用萎坷,但只對當(dāng)前線程有用凹联。
下面來詳細講講線程存儲的ThreadLocalMap吧:

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

            // We don't use a fast path as with get() because it is at
            // least as common to use set() to create new entries as
            // it is to replace existing ones, in which case, a fast
            // path would fail more often than not.

            Entry[] tab = table;
            int len = tab.length;
            int i = key.threadLocalHashCode & (len-1);

            for (Entry e = tab[i];
                 e != null;
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();

                if (k == key) {
                    e.value = value;
                    return;
                }

                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }

            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
}

每個ThreadLocal對象都有一個hash值threadLocalHashCode,每初始化一個ThreadLocal對象哆档,hash值就增加一個固定的大小0x61c88647蔽挠。

在插入過程中,根據(jù)ThreadLocal對象的hash值瓜浸,定位到table中的位置i澳淑,過程如下:
1、如果當(dāng)前位置是空的插佛,那么正好杠巡,就初始化一個Entry對象放在位置i上;
2雇寇、不巧氢拥,位置i已經(jīng)有Entry對象了,如果這個Entry對象的key正好是即將設(shè)置的key锨侯,那么重新設(shè)置Entry中的value嫩海;
3、很不巧识腿,位置i的Entry對象出革,和即將設(shè)置的key沒關(guān)系,那么只能找下一個空位置渡讼;

這樣的話骂束,在get的時候耳璧,也會根據(jù)ThreadLocal對象的hash值,定位到table中的位置展箱,然后判斷該位置Entry對象中的key是否和get的key一致旨枯,如果不一致,就判斷下一個位置

可以發(fā)現(xiàn)混驰,set和get如果沖突嚴重的話攀隔,效率很低,因為ThreadLoalMap是Thread的一個屬性栖榨,所以即使在自己的代碼中控制了設(shè)置的元素個數(shù)昆汹,但還是不能控制其它代碼的行為。

這里需要注意的是婴栽,ThreadLoalMap的Entry是繼承WeakReference满粗,和HashMap很大的區(qū)別是驻啤,Entry中沒有next字段批糟,所以就不存在鏈表的情況了斑响。

內(nèi)存泄漏

通過之前的分析已經(jīng)知道吵血,當(dāng)使用ThreadLocal保存一個value時,會在ThreadLocalMap中的數(shù)組插入一個Entry對象戒职,按理說key-value都應(yīng)該以強引用保存在Entry對象中劲绪,但在ThreadLocalMap的實現(xiàn)中爷抓,key被保存到了WeakReference對象中鞍陨。

這就導(dǎo)致了一個問題步淹,ThreadLocal在沒有外部強引用時,發(fā)生GC時會被回收诚撵,如果創(chuàng)建ThreadLocal的線程一直持續(xù)運行贤旷,那么這個Entry對象中的value就有可能一直得不到回收,發(fā)生內(nèi)存泄露砾脑。

但是內(nèi)存泄漏是否真的會發(fā)生呢?
可以查看ThreadLocalMap的getEntry()艾杏,getEntryAfterMiss(ThreadLocal<?> key, int i, Entry e)韧衣,expungeStaleEntry(int staleSlot),代碼就不貼了购桑,首先從ThreadLocal的直接索引位置(通過ThreadLocal.threadLocalHashCode & (len-1)運算得到)獲取Entry e畅铭,如果e不為null并且key相同則返回e;如果e為null或者key不一致則向下一個位置查詢勃蜘,如果下一個位置的key和當(dāng)前需要查詢的key相等硕噩,則返回對應(yīng)的Entry,否則缭贡,如果key值為null炉擅,則擦除該位置的Entry辉懒,否則繼續(xù)向下一個位置查詢。
在這個過程中遇到的key為null的Entry都會被擦除谍失,那么Entry內(nèi)的value也就沒有強引用鏈眶俩,自然會被回收。仔細研究代碼可以發(fā)現(xiàn)快鱼,set操作也有類似的思想颠印,將key為null的這些Entry都刪除,防止內(nèi)存泄露抹竹。
當(dāng)然如果調(diào)用remove方法线罕,肯定會刪除對應(yīng)的Entry對象,最為保險窃判。


題外話:

其實shiro保存subject信息就是通過ThreadLocal來保存钞楼,然后通過SecurityUtils.getSubject()給不同層級的方法可以拿到subject,不過shiro特別之處就是用了InheritableThreadLocal兢孝。為什么用這個呢窿凤?

InheritableThreadLocal 可以讓用戶自行 new Thread 出來的線程可以獲取到 Subject,否則用戶還要額外想辦法怎么獲取到這個 Subject跨蟹。通俗的說就是雳殊,當(dāng)你啟動多線程的時候,如果用ThreadLocal將拿不到主線程的subject窗轩,這個就是解決了這個問題夯秃。配置化用InheritableThreadLocal就很好用。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末痢艺,一起剝皮案震驚了整個濱河市仓洼,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌堤舒,老刑警劉巖色建,帶你破解...
    沈念sama閱讀 212,599評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異舌缤,居然都是意外死亡箕戳,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,629評論 3 385
  • 文/潘曉璐 我一進店門国撵,熙熙樓的掌柜王于貴愁眉苦臉地迎上來陵吸,“玉大人,你說我怎么就攤上這事介牙∽吵妫” “怎么了?”我有些...
    開封第一講書人閱讀 158,084評論 0 348
  • 文/不壞的土叔 我叫張陵环础,是天一觀的道長囚似。 經(jīng)常有香客問我剩拢,道長,這世上最難降的妖魔是什么谆构? 我笑而不...
    開封第一講書人閱讀 56,708評論 1 284
  • 正文 為了忘掉前任裸扶,我火速辦了婚禮,結(jié)果婚禮上搬素,老公的妹妹穿的比我還像新娘呵晨。我一直安慰自己,他們只是感情好熬尺,可當(dāng)我...
    茶點故事閱讀 65,813評論 6 386
  • 文/花漫 我一把揭開白布摸屠。 她就那樣靜靜地躺著,像睡著了一般粱哼。 火紅的嫁衣襯著肌膚如雪季二。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 50,021評論 1 291
  • 那天揭措,我揣著相機與錄音胯舷,去河邊找鬼。 笑死绊含,一個胖子當(dāng)著我的面吹牛桑嘶,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播躬充,決...
    沈念sama閱讀 39,120評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼逃顶,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了充甚?” 一聲冷哼從身側(cè)響起以政,我...
    開封第一講書人閱讀 37,866評論 0 268
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎伴找,沒想到半個月后盈蛮,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,308評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡技矮,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,633評論 2 327
  • 正文 我和宋清朗相戀三年眉反,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片穆役。...
    茶點故事閱讀 38,768評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖梳凛,靈堂內(nèi)的尸體忽然破棺而出耿币,到底是詐尸還是另有隱情,我是刑警寧澤韧拒,帶...
    沈念sama閱讀 34,461評論 4 333
  • 正文 年R本政府宣布淹接,位于F島的核電站十性,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏塑悼。R本人自食惡果不足惜劲适,卻給世界環(huán)境...
    茶點故事閱讀 40,094評論 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望厢蒜。 院中可真熱鬧霞势,春花似錦、人聲如沸斑鸦。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,850評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽巷屿。三九已至固以,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間嘱巾,已是汗流浹背憨琳。 一陣腳步聲響...
    開封第一講書人閱讀 32,082評論 1 267
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留旬昭,地道東北人篙螟。 一個月前我還...
    沈念sama閱讀 46,571評論 2 362
  • 正文 我出身青樓,卻偏偏與公主長得像稳懒,于是被迫代替她去往敵國和親闲擦。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,666評論 2 350

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