ThreadLocal奪命11連問

點擊關(guān)注公眾號，Java干貨及時送達

前一段時間，有同事使用ThreadLocal踩坑了，正好引起了我的興趣。

所以近期，我抽空把ThreadLocal的源碼再研究了一下，越看越有意思，發(fā)現(xiàn)里面的東西還真不少。

我把精華濃縮了一下，匯集成了下面11個問題，看看你能頂住第幾個？

1. 為什么要用ThreadLocal?

并發(fā)編程是一項非常重要的技術(shù)，它讓我們的程序變得更加高效。

但在并發(fā)的場景中，如果有多個線程同時修改公共變量，可能會出現(xiàn)線程安全問題，即該變量最終結(jié)果可能出現(xiàn)異常。

為了解決線程安全問題，JDK出現(xiàn)了很多技術(shù)手段，比如：使用synchronized或Lock，給訪問公共資源的代碼上鎖，保證了代碼的原子性。

但在高并發(fā)的場景中，如果多個線程同時競爭一把鎖，這時會存在大量的鎖等待，可能會浪費很多時間，讓系統(tǒng)的響應(yīng)時間一下子變慢。

因此，JDK還提供了另外一種用空間換時間的新思路：ThreadLocal。

它的核心思想是：共享變量在每個線程都有一個副本，每個線程操作的都是自己的副本，對另外的線程沒有影響。

例如：

@Service
public?class?ThreadLocalService?{
????private?static?final?ThreadLocal?threadLocal?=?new?ThreadLocal<>();

????public?void?add()?{
????????threadLocal.set(1);
????????doSamething();
????????Integer?integer?=?threadLocal.get();
????}
}

2. ThreadLocal的原理是什么？

為了搞清楚ThreadLocal的底層實現(xiàn)原理，我們不得不扒一下源碼。

ThreadLocal的內(nèi)部有一個靜態(tài)的內(nèi)部類叫：ThreadLocalMap。

public?class?ThreadLocal<T>?{
?????...
?????public?T?get()?{
????????//獲取當前線程
????????Thread?t?=?Thread.currentThread();
????????//獲取當前線程的成員變量ThreadLocalMap對象
????????ThreadLocalMap?map?=?getMap(t);
????????if?(map?!=?null)?{
????????????//根據(jù)threadLocal對象從map中獲取Entry對象
????????????ThreadLocalMap.Entry?e?=?map.getEntry(this);
????????????if?(e?!=?null)?{
????????????????@SuppressWarnings("unchecked")
????????????????//獲取保存的數(shù)據(jù)
????????????????T?result?=?(T)e.value;
????????????????return?result;
????????????}
????????}
????????//初始化數(shù)據(jù)
????????return?setInitialValue();
????}
????
????private?T?setInitialValue()?{
????????//獲取要初始化的數(shù)據(jù)
????????T?value?=?initialValue();
????????//獲取當前線程
????????Thread?t?=?Thread.currentThread();
????????//獲取當前線程的成員變量ThreadLocalMap對象
????????ThreadLocalMap?map?=?getMap(t);
????????//如果map不為空
????????if?(map?!=?null)
????????????//將初始值設(shè)置到map中，key是this，即threadLocal對象，value是初始值
????????????map.set(this,?value);
????????else
???????????//如果map為空，則需要創(chuàng)建新的map對象
????????????createMap(t,?value);
????????return?value;
????}
????
????public?void?set(T?value)?{
????????//獲取當前線程
????????Thread?t?=?Thread.currentThread();
????????//獲取當前線程的成員變量ThreadLocalMap對象
????????ThreadLocalMap?map?=?getMap(t);
????????//如果map不為空
????????if?(map?!=?null)
????????????//將值設(shè)置到map中，key是this，即threadLocal對象，value是傳入的value值
????????????map.set(this,?value);
????????else
???????????//如果map為空，則需要創(chuàng)建新的map對象
????????????createMap(t,?value);
????}
????
?????static?class?ThreadLocalMap?{
????????...
?????}
?????...
}

ThreadLocal的get方法、set方法和setInitialValue方法，其實最終操作的都是ThreadLocalMap類中的數(shù)據(jù)。

其中ThreadLocalMap類的內(nèi)部如下：

static?class?ThreadLocalMap?{
????static?class?Entry?extends?WeakReference<ThreadLocal>?{
????????Object?value;

????????Entry(ThreadLocal?k,?Object?v)?{
????????????super(k);
????????????value?=?v;
????????}
???}
???...
???private?Entry[]?table;
???...
}

ThreadLocalMap里面包含一個靜態(tài)的內(nèi)部類Entry，該類繼承于WeakReference類，說明Entry是一個弱引用。

ThreadLocalMap內(nèi)部還包含了一個Entry數(shù)組，其中：Entry = ThreadLocal + value。

而ThreadLocalMap被定義成了Thread類的成員變量。

public?class?Thread?implements?Runnable?{
????...
????ThreadLocal.ThreadLocalMap?threadLocals?=?null;
}

下面用一張圖從宏觀上，認識一下ThreadLocal的整體結(jié)構(gòu)：從上圖中看出，在每個Thread類中，都有一個ThreadLocalMap的成員變量，該變量包含了一個Entry數(shù)組，該數(shù)組真正保存了ThreadLocal類set的數(shù)據(jù)。

Entry是由threadLocal和value組成，其中threadLocal對象是弱引用，在GC的時候，會被自動回收。而value就是ThreadLocal類set的數(shù)據(jù)。

下面用一張圖總結(jié)一下引用關(guān)系：上圖中除了Entry的key對ThreadLocal對象是弱引用，其他的引用都是強引用。

需要特別說明的是，上圖中ThreadLocal對象我畫到了堆上，其實在實際的業(yè)務(wù)場景中不一定在堆上。因為如果ThreadLocal被定義成了static的，ThreadLocal的對象是類共用的，可能出現(xiàn)在方法區(qū)。

3. 為什么用ThreadLocal做key？

不知道你有沒有思考過這樣一個問題：ThreadLocalMap為什么要用ThreadLocal做key，而不是用Thread做key？

如果在你的應(yīng)用中，一個線程中只使用了一個ThreadLocal對象，那么使用Thread做key也未嘗不可。

@Service
public?class?ThreadLocalService?{
????private?static?final?ThreadLocal?threadLocal?=?new?ThreadLocal<>();
}????

但實際情況中，你的應(yīng)用，一個線程中很有可能不只使用了一個ThreadLocal對象。這時使用Thread做key不就出有問題？

@Service
public?class?ThreadLocalService?{
????private?static?final?ThreadLocal?threadLocal1?=?new?ThreadLocal<>();
????private?static?final?ThreadLocal?threadLocal2?=?new?ThreadLocal<>();
????private?static?final?ThreadLocal?threadLocal3?=?new?ThreadLocal<>();
}????

假如使用Thread做key時，你的代碼中定義了3個ThreadLocal對象，那么，通過Thread對象，它怎么知道要獲取哪個ThreadLocal對象呢？

如下圖所示：

因此，不能使用Thread做key，而應(yīng)該改成用ThreadLocal對象做key，這樣才能通過具體ThreadLocal對象的get方法，輕松獲取到你想要的ThreadLocal對象。

如下圖所示：

4. Entry的key為什么設(shè)計成弱引用？

前面說過，Entry的key，傳入的是ThreadLocal對象，使用了WeakReference對象，即被設(shè)計成了弱引用。

那么，為什么要這樣設(shè)計呢？

假如key對ThreadLocal對象的弱引用，改為強引用。我們都知道ThreadLocal變量對ThreadLocal對象是有強引用存在的。

即使ThreadLocal變量生命周期完了，設(shè)置成null了，但由于key對ThreadLocal還是強引用。

此時，如果執(zhí)行該代碼的線程使用了線程池，一直長期存在，不會被銷毀。

就會存在這樣的強引用鏈：Thread變量 -> Thread對象 -> ThreadLocalMap -> Entry -> key -> ThreadLocal對象。

那么，ThreadLocal對象和ThreadLocalMap都將不會被GC回收，于是產(chǎn)生了內(nèi)存泄露問題。

為了解決這個問題，JDK的開發(fā)者們把Entry的key設(shè)計成了弱引用。

弱引用的對象，在GC做垃圾清理的時候，就會被自動回收了。

如果key是弱引用，當ThreadLocal變量指向null之后，在GC做垃圾清理的時候，key會被自動回收，其值也被設(shè)置成null。

如下圖所示：接下來，最關(guān)鍵的地方來了。

由于當前的ThreadLocal變量已經(jīng)被指向null了，但如果直接調(diào)用它的get、set或remove方法，很顯然會出現(xiàn)空指針異常。因為它的生命已經(jīng)結(jié)束了，再調(diào)用它的方法也沒啥意義。

此時，如果系統(tǒng)中還定義了另外一個ThreadLocal變量b，調(diào)用了它的get、set或remove，三個方法中的任何一個方法，都會自動觸發(fā)清理機制，將key為null的value值清空。

如果key和value都是null，那么Entry對象會被GC回收。如果所有的Entry對象都被回收了，ThreadLocalMap也會被回收了。

這樣就能最大程度的解決內(nèi)存泄露問題。

需要特別注意的地方是：

1. key為null的條件是，ThreadLocal變量指向null，并且key是弱引用。如果ThreadLocal變量沒有斷開對ThreadLocal的強引用，即ThreadLocal變量沒有指向null，GC就貿(mào)然的把弱引用的key回收了，不就會影響正常用戶的使用？
2. 如果當前ThreadLocal變量指向null了，并且key也為null了，但如果沒有其他ThreadLocal變量觸發(fā)get、set或remove方法，也會造成內(nèi)存泄露。

下面看看弱引用的例子：

public?static?void?main(String[]?args)?{
????WeakReference