Java研發(fā)崗面試必問,深入理解 ThreadLocal 實(shí)現(xiàn)原理與內(nèi)存泄露

前言

在面試環(huán)節(jié)中,考察"ThreadLocal"也是面試官的家常便飯,所以對(duì)它理解透徹,是非常有必要的.

有些面試官會(huì)開門見山的提問:

  • “知道ThreadLocal嗎?”
  • “講講你對(duì)ThreadLocal的理解”

當(dāng)然了,也有面試官會(huì)慢慢引導(dǎo)到這個(gè)話題上,比如提問“在多線程環(huán)境下,如何防止自己的變量被其它線程篡改”,將主動(dòng)權(quán)交給你自己,剩下的靠自己發(fā)揮。

那么ThreadLocal可以做什么,在了解它的應(yīng)用場景之前,我們先看看它的實(shí)現(xiàn)原理,只有知道了實(shí)現(xiàn)原理,才好判斷它是否符合自己的業(yè)務(wù)場景。

ThreadLocal是什么

首先,它是一個(gè)數(shù)據(jù)結(jié)構(gòu),有點(diǎn)像HashMap,可以保存"key : value"鍵值對(duì),但是一個(gè)ThreadLocal只能保存一個(gè),并且各個(gè)線程的數(shù)據(jù)互不干擾。

在線程1中初始化了一個(gè)ThreadLocal對(duì)象localName,并通過set方法,保存了一個(gè)值 占小狼 ,同時(shí)在線程1中通過 localName.get() 可以拿到之前設(shè)置的值,但是如果在線程2中,拿到的將是一個(gè)null。

這是為什么,如何實(shí)現(xiàn)?不過之前也說了,ThreadLocal保證了各個(gè)線程的數(shù)據(jù)互不干擾。

看看 set(T value) 和 get() 方法的源碼

可以發(fā)現(xiàn),每個(gè)線程中都有一個(gè) ThreadLocalMap 數(shù)據(jù)結(jié)構(gòu),當(dāng)執(zhí)行set方法時(shí),其值是保存在當(dāng)前線程的 threadLocals 變量中,當(dāng)執(zhí)行set方法中,是從當(dāng)前線程的 threadLocals變量獲取。

所以在線程1中set的值,對(duì)線程2來說是摸不到的,而且在線程2中重新set的話,也不會(huì)影響到線程1中的值,保證了線程之間不會(huì)相互干擾。

那每個(gè)線程中的 ThreadLoalMap 究竟是什么?

ThreadLoalMap

本文分析的是1.7的源碼。

從名字上看,可以猜到它也是一個(gè)類似HashMap的數(shù)據(jù)結(jié)構(gòu),但是在ThreadLocal中,并沒實(shí)現(xiàn)Map接口。

在ThreadLoalMap中,也是初始化一個(gè)大小16的Entry數(shù)組,Entry對(duì)象用來保存每一個(gè)key-value鍵值對(duì),只不過這里的key永遠(yuǎn)都是ThreadLocal對(duì)象,是不是很神奇,通過ThreadLocal對(duì)象的set方法,結(jié)果把ThreadLocal對(duì)象自己當(dāng)做key,放進(jìn)了ThreadLoalMap中。

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

Hash沖突

沒有鏈表結(jié)構(gòu),那發(fā)生hash沖突了怎么辦?

先看看ThreadLoalMap中插入一個(gè)key-value的實(shí)現(xiàn)

每個(gè)ThreadLocal對(duì)象都有一個(gè)hash值 threadLocalHashCode ,每初始化一個(gè)ThreadLocal對(duì)象,hash值就增加一個(gè)固定的大小 0x61c88647 。

在插入過程中,根據(jù)ThreadLocal對(duì)象的hash值,定位到table中的位置i,過程如下:1、如果當(dāng)前位置是空的,那么正好,就初始化一個(gè)Entry對(duì)象放在位置i上;2、不巧,位置i已經(jīng)有Entry對(duì)象了,如果這個(gè)Entry對(duì)象的key正好是即將設(shè)置的key,那么重新設(shè)置Entry中的value;3、很不巧,位置i的Entry對(duì)象,和即將設(shè)置的key沒關(guān)系,那么只能找下一個(gè)空位置;

這樣的話,在get的時(shí)候,也會(huì)根據(jù)ThreadLocal對(duì)象的hash值,定位到table中的位置,然后判斷該位置Entry對(duì)象中的key是否和get的key一致,如果不一致,就判斷下一個(gè)位置

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

內(nèi)存泄露

ThreadLocal可能導(dǎo)致內(nèi)存泄漏,為什么?先看看Entry的實(shí)現(xiàn):

通過之前的分析已經(jīng)知道,當(dāng)使用ThreadLocal保存一個(gè)value時(shí),會(huì)在ThreadLocalMap中的數(shù)組插入一個(gè)Entry對(duì)象,按理說key-value都應(yīng)該以強(qiáng)引用保存在Entry對(duì)象中,但在ThreadLocalMap的實(shí)現(xiàn)中,key被保存到了WeakReference對(duì)象中。

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

如何避免內(nèi)存泄露

既然已經(jīng)發(fā)現(xiàn)有內(nèi)存泄露的隱患,自然有應(yīng)對(duì)的策略,在調(diào)用ThreadLocal的get()、set()可能會(huì)清除ThreadLocalMap中key為null的Entry對(duì)象,這樣對(duì)應(yīng)的value就沒有GC Roots可達(dá)了,下次GC的時(shí)候就可以被回收,當(dāng)然如果調(diào)用remove方法,肯定會(huì)刪除對(duì)應(yīng)的Entry對(duì)象。

如果使用ThreadLocal的set方法之后,沒有顯示的調(diào)用remove方法,就有可能發(fā)生內(nèi)存泄露,所以養(yǎng)成良好的編程習(xí)慣十分重要,使用完ThreadLocal之后,記得調(diào)用remove方法。

————END————

  • 點(diǎn)贊;
  • ...
  • 轉(zhuǎn)發(fā)
  • ...
  • 關(guān)注!??!
  • ...
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容