java ThreadLocal(應用場景及使用方式及原理)

雖然ThreadLocal與併發問題相關,但是許多程序員僅僅將它作爲一種用於“方便傳參”的工具,胖哥認爲這也許並不是ThreadLocal設計的目的,它本身是爲線程安全和某些特定場景的問題而設計的。
ThreadLocal是什麼呢!
每個ThreadLocal可以放一個線程級別的變量,但是它本身可以被多個線程共享使用,而且又可以達到線程安全的目的,且絕對線程安全。
例如:

public final static ThreadLocal<String> RESOURCE = new ThreadLocal<String>();

RESOURCE代表一個可以存放String類型的ThreadLocal對象,此時任何一個線程可以併發訪問這個變量,對它進行寫入、讀取操作,都是線程安全的。比如一個線程通過RESOURCE.set(“aaaa”);將數據寫入ThreadLocal中,在任何一個地方,都可以通過RESOURCE.get();將值獲取出來。
但是它也並不完美,有許多缺陷,就像大家依賴於它來做參數傳遞一樣,接下來我們就來分析它的一些不好的地方。
爲什麼有些時候會將ThreadLocal作爲方便傳遞參數的方式呢?

例如當許多方法相互調用時,最初的設計可能沒有想太多,有多少個參數就傳遞多少個變量,那麼整個參數傳遞的過程就是零散的。進一步思考:若A方法調用B方法傳遞了8個參數,B方法接下來調用C方法->D方法->E方法->F方法等只需要5個參數,此時在設計API時就涉及5個參數的入口,這些方法在業務發展的過程中被許多地方所複用。
某一天,我們發現F方法需要加一個參數,這個參數在A方法的入口參數中有,此時,如果要改中間方法牽涉面會很大,而且不知道修改後會不會有Bug。作爲程序員的我們可能會隨性一想,ThreadLocal反正是全局的,就放這裏吧,確實好解決。
但是此時你會發現系統中這種方式有點像在貼補丁,越貼越多,我們必須要求調用相關的代碼都使用ThreadLocal傳遞這個參數,有可能會搞得亂七八糟的。換句話說,並不是不讓用,而是我們要明確它的入口和出口是可控的。
詭異的ThreadLocal最難琢磨的是“作用域”,尤其是在代碼設計之初很亂的情況下,如果再增加許多ThreadLocal,系統就會逐漸變成神龍見首不見尾的情況。有了這樣一個省事的東西,可能許多小夥伴更加不在意設計,因爲大家都認爲這些問題都可以通過變化的手段來解決。胖哥認爲這是一種惡性循環。
對於這類業務場景,應當提前有所準備,需要粗粒度化業務模型,即使要用ThreadLocal,也不是加一個參數就加一個ThreadLocal變量。例如,我們可以設計幾種對象來封裝入口參數,在接口設計時入口參數都以對象爲基礎。
也許一個類無法表達所有的參數意思,而且那樣容易導致強耦合。
通常我們按照業務模型分解爲幾大類型對象作爲它們的參數包裝,並且將按照對象屬性共享情況進行抽象,在繼承關係的每一個層次各自擴展相應的參數,或者說加參數就在對象中加,共享參數就在父類中定義,這樣的參數就逐步規範化了。

我們回到正題,探討一下ThreadLocal到底是用來做什麼的?爲此我們探討下文中的幾個話題。

(1)應用場景及使用方式
爲了說明ThreadLocal的應用場景,我們來看一個框架的例子。Spring的事務管理器通過AOP切入業務代碼,在進入業務代碼前,會根據對應的事務管理器提取出相應的事務對象,假如事務管理器是DataSourceTransactionManager,就會從DataSource中獲取一個連接對象,通過一定的包裝後將其保存在ThreadLocal中。並且Spring也將DataSource進行了包裝,重寫了其中的getConnection()方法,或者說該方法的返回將由Spring來控制,這樣Spring就能讓線程內多次獲取到的Connection對象是同一個。

爲什麼要放在ThreadLocal裏面呢?
因爲Spring在AOP後並不能嚮應用程序傳遞參數,應用程序的每個業務代碼是事先定義好的,Spring並不會要求在業務代碼的入口參數中必須編寫Connection的入口參數。此時Spring選擇了ThreadLocal,通過它保證連接對象始終在線程內部,任何時候都能拿到,此時Spring非常清楚什麼時候回收這個連接,也就是非常清楚什麼時候從ThreadLocal中刪除這個元素(在9.2節中會詳細講解)。

從Spring事務管理器的設計上可以看出,Spring利用ThreadLocal得到了一個很完美的設計思路,同時它在設計時也十分清楚ThreadLocal中元素應該在什麼時候刪除。由此,我們簡單地認爲ThreadLocal儘量使用在一個全局的設計上,而不是一種打補丁的間接方法。

瞭解了基本應用場景後,接下來看一個例子。定義一個類用於存放靜態的ThreadLocal對象,通過多個線程並行地對ThreadLocal對象進行set、get操作,並將值進行打印,來看看每個線程自己設置進去的值和取出來的值是否是一樣的。代碼如下:
代碼清單5-8 簡單的ThreadLocal例子

public class ThreadLocalTest {


static class ResourceClass {

	public final static ThreadLocal<String> RESOURCE_1 =
								   new ThreadLocal<String>();

	public final static ThreadLocal<String> RESOURCE_2 =
								   new ThreadLocal<String>();

}

static class A {

	public void setOne(String value) {
		ResourceClass.RESOURCE_1.set(value);
	}

	public void setTwo(String value) {
		ResourceClass.RESOURCE_2.set(value);
	}
}

static class B {
	public void display() {
		System.out.println(ResourceClass.RESOURCE_1.get()
					+ ":" + ResourceClass.RESOURCE_2.get());
	}
}

public static void main(String []args) {
	final A a = new A();
	final B b = new B();
	for(int i = 0 ; i < 15 ; i ++) {
		final String resouce1 = "線程-" + I;
		final String resouce2 = " value = (" + i + ")";
		new Thread() {
			public void run() {
			try {
				a.setOne(resouce1);
				a.setTwo(resouce2);
				b.display();
			}finally {
				ResourceClass.RESOURCE_1.remove();
				ResourceClass.RESOURCE_2.remove();
			}
		}
	}.start();
	}
}
}

關於這段代碼,我們先說幾點。
◎ 定義了兩個ThreadLocal變量,最終的目的就是要看最後兩個值是否能對應上,這樣纔有機會證明ThreadLocal所保存的數據可能是線程私有的。
◎ 使用兩個內部類只是爲了使測試簡單,方便大家直觀理解,大家也可以將這個例子的代碼拆分到多個類中,得到的結果是相同的。
◎ 測試代碼更像是爲了方便傳遞參數,因爲它確實傳遞參數很方便,但這僅僅是爲了測試。
◎ 在finally裏面有remove()操作,是爲了清空數據而使用的。爲何要清空數據,在後文中會繼續介紹細節。
測試結果如下:
線程-6: value = (6)
線程-9: value = (9)
線程-0: value = (0)
線程-10: value = (10)
線程-12: value = (12)
線程-14: value = (14)
線程-11: value = (11)
線程-3: value = (3)
線程-5: value = (5)
線程-13: value = (13)
線程-2: value = (2)
線程-4: value = (4)
線程-8: value = (8)
線程-7: value = (7)
線程-1: value = (1)
大家可以看到輸出的線程順序並非最初定義線程的順序,理論上可以說明多線程應當是併發執行的,但是依然可以保持每個線程裏面的值是對應的,說明這些值已經達到了線程私有的目的。
不是說共享變量無法做到線程私有嗎?它又是如何做到線程私有的呢?這就需要我們知道一點點原理上的東西,否則用起來也沒那麼放心,請看下面的介紹。

(2)ThreadLocal內在原理
從前面的操作可以發現,ThreadLocal最常見的操作就是set、get、remove三個動作,下面來看看這三個動作到底做了什麼事情。首先看set操作,源碼片段如圖5-5所示。

圖5-5 ThreadLcoal.set源碼片段
圖5-5中的第一條代碼取出了當前線程t,然後調用getMap(t)方法時傳入了當前線程,換句話說,該方法返回的ThreadLocalMap和當前線程有點關係,我們先記錄下來。進一步判定如果這個map不爲空,那麼設置到Map中的Key就是this,值就是外部傳入的參數。這個this是什麼呢?就是定義的ThreadLocal對象。
代碼中有兩條路徑需要追蹤,分別是getMap(Thread)和createMap(Thread , T)。首先來看看getMap(t)操作,如圖5-6所示。
圖5-6 getMap(Thread)操作

在這裏,我們看到ThreadLocalMap其實就是線程裏面的一個屬性,它在Thread類中的定義是:
ThreadLocal.ThreadLocalMap threadLocals = null;
這種方法很容易讓人混淆,因爲這個ThreadLocalMap是ThreadLocal裏面的內部類,放在了Thread類裏面作爲一個屬性而存在,ThreadLocal本身成爲這個Map裏面存放的Key,用戶輸入的值是Value。太亂了,理不清楚了,畫個圖來看看(見圖5-7)。
簡單來講,就是這個Map對象在Thread裏面作爲私有的變量而存在,所以是線程安全的。ThreadLocal通過Thread.currentThread()獲取當前的線程就能得到這個Map對象,同時將自身作爲Key發起寫入和讀取,由於將自身作爲Key,所以一個ThreadLocal對象就能存放一個線程中對應的Java對象,通過get也自然能找到這個對象。

圖5-7 Thread與ThreadLocal的僞代碼關聯關係

如果還沒有理解,則可以將思維放寬一點。當定義變量String a時,這個“a”其實只是一個名稱(在第3章中已經說到了常量池),虛擬機需要通過符號表來找到相應的信息,而這種方式正好就像一種K-V結構,底層的處理方式也確實很接近這樣,這裏的處理方式是顯式地使用Map來存放數據,這也是一種實現手段的變通。
現在有了思路,繼續回到上面的話題,爲了驗證前面的推斷和理解,來看看createMap方法的細節,如圖5-8所示。

圖5-8 createMap操作
這段代碼是執行一個創建新的Map的操作,並且將第一個值作爲這個Map的初始化值,由於這個Map是線程私有的,不可能有另一個線程同時也在對它做put操作,因此這裏的賦值和初始化是絕對線程安全的,也同時保證了每一個外部寫入的值都將寫入到Map對象中。
最後來看看get()、remove()代碼,或許看到這裏就可以認定我們的理論是正確的,如圖5-9所示。

圖5-9 get()/remove()方法的代碼片段
給我們的感覺是,這樣實現是一種技巧,而不是一種技術。
其實是技巧還是技術完全是從某種角度來看的,或者說是從某種抽象層次來看的,如果這段代碼在C++中實現,難道就叫技術,不是技巧了嗎?當然不是!胖哥認爲技術依然是建立在思想和方法基礎上的,只是看實現的抽象層次在什麼級別。就像在本書中多個地方探討的一些基礎原理一樣,我們探討了它的思想,其實它的實現也是基於某種技巧和手段的,只是對程序封裝後就變成了某種語法和API,因此胖哥認爲,一旦學會使用技巧思考問題,就學會了通過技巧去看待技術本身。我們應當通過這種設計,學會一種變通和發散的思維,學會理解各種各樣的場景,這樣便可以積累許多真正的財富,這些財富不是通過某些工具的使用或測試就可以獲得的。
ThreadLocal的這種設計很完美嗎?
不是很完美,它依然有許多坑,在這裏對它容易誤導程序員當成傳參工具就不再多提了,下面我們來看看它的使用不當會導致什麼技術上的問題。

(3)ThreadLocal的坑
通過上面的分析,我們可以認識到ThreadLocal其實是與線程綁定的一個變量,如此就會出現一個問題:如果沒有將ThreadLocal內的變量刪除(remove)或替換,它的生命週期將會與線程共存。因此,ThreadLocal的一個很大的“坑”就是當使用不當時,導致使用者不知道它的作用域範圍。

大家可能認爲線程結束後ThreadLocal應該就回收了,如果線程真的註銷了確實是這樣的,但是事實有可能並非如此,例如在線程池中對線程管理都是採用線程複用的方法(Web容器通常也會採用線程池),在線程池中線程很難結束甚至於永遠不會結束,這將意味着線程持續的時間將不可預測,甚至與JVM的生命週期一致。那麼相應的ThreadLocal變量的生命週期也將不可預測。

也許系統中定義少量幾個ThreadLocal變量也無所謂,因爲每次set數據時是用ThreadLocal本身作爲Key的,相同的Key肯定會替換原來的數據,原來的數據就可以被釋放了,理論上不會導致什麼問題。但世事無絕對,如果ThreadLocal中直接或間接包裝了集合類或複雜對象,每次在同一個ThreadLocal中取出對象後,再對內容做操作,那麼內部的集合類和複雜對象所佔用的空間可能會開始膨脹。

拋開代碼本身的問題,舉一個極端的例子。如果不想定義太多的ThreadLocal變量,就用一個HashMap來存放,這貌似沒什麼問題。由於ThreadLocal在程序的任何一個地方都可以用得到,在某些設計不當的代碼中很難知道這個HashMap寫入的源頭,在代碼中爲了保險起見,通常會先檢查這個HashMap是否存在,若不存在,則創建一個HashMap寫進去;若存在,通常也不會替換掉,因爲代碼編寫者通常會“害怕”因爲這種替換會丟掉一些來自“其他地方寫入HashMap的數據”,從而導致許多不可預見的問題。

在這樣的情況下,HashMap第一次放入ThreadLocal中也許就一直不會被釋放,而這個HashMap中可能開始存放許多Key-Value信息,如果業務上存放的Key值在不斷變化(例如,將業務的ID作爲Key),那麼這個HashMap就開始不斷變長,並且很可能在每個線程中都有一個這樣的HashMap,逐漸地形成了間接的內存泄漏。曾經有很多人吃過這個虧,而且吃虧的時候發現這樣的代碼可能不是在自己的業務系統中,而是出現在某些二方包、三方包中(開源並不保證沒有問題)。

要處理這種問題很複雜,不過首先要保證自己編寫的代碼是沒問題的,要保證沒問題不是說我們不去用ThreadLocal,甚至不去學習它,因爲它肯定有其應用價值。在使用時要明白ThreadLocal最難以捉摸的是“不知道哪裏是源頭”(通常是代碼設計不當導致的),只有知道了源頭才能控制結束的部分,或者說我們從設計的角度要讓ThreadLocal的set、remove有始有終,通常在外部調用的代碼中使用finally來remove數據,只要我們仔細思考和抽象是可以達到這個目的的。有些是二方包、三方包的問題,對於這些問題我們需要學會的是找到問題的根源後解決,關於二方包、三方包的運行跟蹤,可參看第3.7.9節介紹的BTrace工具。

補充:在任何異步程序中(包括異步I/O、非阻塞I/O),ThreadLocal的參數傳遞是不靠譜的,因爲線程將請求發送後,就不再等待遠程返回結果繼續向下執行了,真正的返回結果得到後,處理的線程可能是另一個。

############################個人總結 ####################################

Thread.java源碼中:

ThreadLocal.ThreadLocalMap threadLocals = null;
即:每個Thread對象都有一個ThreadLocal.ThreadLocalMap成員變量,ThreadLocal.ThreadLocalMap是一個ThreadLocal類的靜態內部類(如下所示),所以Thread類可以進行引用.
static class ThreadLocalMap {
所以每個線程都會有一個ThreadLocal.ThreadLocalMap對象的引用
當在ThreadLocal中進行設值的時候:

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;
}

首先獲取當前線程的引用,然後獲取當前線程的ThreadLocal.ThreadLocalMap對象(t.threadLocals變量就是ThreadLocal.ThreadLocalMap的變量),如果該對象爲空就創建一個,如下所示:
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
這個this變量就是ThreadLocal的引用,對於同一個ThreadLocal對象每個線程都是相同的,但是每個線程各自有一個ThreadLocal.ThreadLocalMap對象保存着各自ThreadLocal引用爲key的值,所以互不影響,而且:如果你新建一個ThreadLocal的對象,這個對象還是保存在每個線程同一個ThreadLocal.ThreadLocalMap對象之中,因爲一個線程只有一個ThreadLocal.ThreadLocalMap對象,這個對象是在第一個ThreadLocal第一次設值的時候進行創建,如上所述的createMap方法.
ThreadLocalMap(ThreadLocal firstKey, Object firstValue) {
table = new Entry[INITIAL_CAPACITY];
int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
table[i] = new Entry(firstKey, firstValue);
size = 1;
setThreshold(INITIAL_CAPACITY);
}

總結:

深入研究java.lang.ThreadLocal類:http://blog.csdn.net/xiaohulunb/article/details/19603611

API說明:

ThreadLocal(),T get(),protected T initialValue(),void remove(),void set(T value)

典型實例:

1.Hiberante的Session 工具類HibernateUtil

2.通過不同的線程對象設置Bean屬性,保證各個線程Bean對象的獨立性。

ThreadLocal使用的一般步驟:

1、在多線程的類(如ThreadDemo類)中,創建一個ThreadLocal對象threadXxx,用來保存線程間需要隔離處理的對象xxx。
2、在ThreadDemo類中,創建一個獲取要隔離訪問的數據的方法getXxx(),在方法中判斷,若ThreadLocal對象爲null時候,應該new()一個隔離訪問類型的對象,並強制轉換爲要應用的類型。
3、在ThreadDemo類的run()方法中,通過getXxx()方法獲取要操作的數據,這樣可以保證每個線程對應一個數據對象,在任何時刻都操作的是這個對象。
與Synchonized的對比:

ThreadLocal和Synchonized都用於解決多線程併發訪問。但是ThreadLocal與synchronized有本質的區別。synchronized是利用鎖的機制,使變量或代碼塊在某一時該只能被一個線程訪問。而ThreadLocal爲每一個線程都提供了變量的副本,使得每個線程在某一時間訪問到的並不是同一個對象,這樣就隔離了多個線程對數據的數據共享。而Synchronized卻正好相反,它用於在多個線程間通信時能夠獲得數據共享。

Synchronized用於線程間的數據共享,而ThreadLocal則用於線程間的數據隔離。

一句話理解ThreadLocal:向ThreadLocal裏面存東西就是向它裏面的Map存東西的,然後ThreadLocal把這個Map掛到當前的線程底下,這樣Map就只屬於這個線程了。

使用ThreadLocal改進你的層次的劃分(spring事務的實現):http://blog.csdn.net/zhouyong0/article/details/7761835

源碼剖析之ThreadLocal:http://wangxinchun.iteye.com/blog/1884228

Java中的ThreadLocal源碼解析(上):http://maosidiaoxian.iteye.com/blog/1939142

ThreadLocal與synchronized:http://blog.csdn.net/yangairong1984/article/details/2294572

Java線程:深入ThreadLocal:http://lavasoft.blog.51cto.com/62575/258459(一個ThreadLocal的模擬實現)

Java多線程(六)、ThreadLocal類:http://blog.csdn.net/lonelyroamer/article/details/7998137

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章