InheritableThreadLocal異步傳遞數(shù)據(jù)實現(xiàn)原理
由于上次主要分析如何解決異步獲取不到Session問題,所以沒有展開分析留下的那個思考題:使用InheritableThreadLocal傳遞Session,為什么說使用線程池不一定能獲取到Session,而不是一定獲取不到?
在Java中,一個Java線程就是一個操作系統(tǒng)線程,創(chuàng)建一個線程需要通過new Thread創(chuàng)建,由JVM為Thread綁定操作系統(tǒng)線程,即便是使用線程池,也需要通過new Thread創(chuàng)建線程。
Thread類有兩個ThreadLocal字段:
- public class Thread implements Runnable {
- ThreadLocal.ThreadLocalMap threadLocals = null;
- ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;
- }
InheritableThreadLocal是ThreadLocal的子類,本質(zhì)上就是一個ThreadLocal。
在Thread類中,threadLocals與inheritableThreadLocals都是線程對象私有的,只能通過當(dāng)前線程對象寫入和獲取數(shù)據(jù),只是Thread會將寫入inheritableThreadLocals的數(shù)據(jù)傳遞給子線程的inheritableThreadLocals。
當(dāng)我們往ThreadLocal或者InheritableThreadLocal寫入數(shù)據(jù)時,寫入過程為:
- 1、ThreadLocal或者InheritableThreadLocal先調(diào)用Thread#currentThread靜態(tài)方法獲取當(dāng)前線程的Thread對象;
- 2、獲取Thread對象的threadLocals或者inheritableThreadLocals;
- 3、將ThreadLocal或者InheritableThreadLocal對象作為key,將數(shù)據(jù)寫入到當(dāng)前Thread對象的threadLocals或者inheritableThreadLocals字段中。
因此,Thread的threadLocals與inheritableThreadLocals的key是ThreadLocal或者InheritableThreadLocal實例,value是寫入的數(shù)據(jù)。
關(guān)于threadLocals我在前面一篇《反向理解ThreadLocal,或許這樣更容易理解》已經(jīng)詳細介紹過了,本篇重點分析inheritableThreadLocals是如何傳遞給子線程的。
默認情況下,當(dāng)我們使用new Thread()創(chuàng)建一個線程時,在Thread的構(gòu)造方法中會通過Thread#currentThread獲取當(dāng)前線程,將當(dāng)前線程作為新創(chuàng)建線程的父線程,所以就有了父子線程關(guān)系。
無論使用哪個重載的構(gòu)造方法創(chuàng)建Thread,都會在構(gòu)造方法中調(diào)用init方法完成初始化為Thread字段賦值,而init方法中有這樣一段代碼:
- private void init(ThreadGroup g, Runnable target, String name,
- long stackSize, AccessControlContext acc,
- boolean inheritThreadLocals) {
- ......
- if (inheritThreadLocals && parent.inheritableThreadLocals != null)
- this.inheritableThreadLocals =
- ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
- ......
- }
在init方法中,由于inheritThreadLocals參數(shù)默認為true,所以只要父線程的inheritableThreadLocals字段不為空,就copy一份父線程的inheritableThreadLocals給當(dāng)前創(chuàng)建的線程對象,這就實現(xiàn)了將父線程的inheritableThreadLocals存儲的數(shù)據(jù)傳遞給子線程。
使用InheritableThreadLocal我們不得不考慮的問題:內(nèi)存泄漏。
ThreadLocal.ThreadLocalMap使用數(shù)組存儲元素,與HashMap不同,它通過開放定址法解決hash沖突,不存在鏈表,通過動態(tài)擴容數(shù)組可無限存儲元素,數(shù)組元素的類型為Entry。
當(dāng)我們往ThreadLocal.ThreadLocalMap寫入一個key-value時,ThreadLocalMap把key和value包裝成一個Entry,并通過key的hashcode值計算索引值,將Entry放到數(shù)組中。
ThreadLocal.ThreadLocalMap.Entry類的源碼如下:
- static class Entry extends WeakReference<ThreadLocal<?>> {
- Object value;
- Entry(ThreadLocal<?> k, Object v) {
- super(k);
- value = v;
- }
- }
雖然key為弱引用的ThreadLocal,當(dāng)ThreadLocal釋放時,Entry的key變?yōu)閚ull,但由于value還在,如果Thread不釋放,那么Entry也就不會被垃圾收集器回收。
但如果線程是臨時創(chuàng)建的,在方法中創(chuàng)建且沒有被其它地方引用,當(dāng)線程執(zhí)行完成時就會被JVM銷毀,在線程實際退出之前由JVM調(diào)用線程的exit方法給線程對象完成清理。exit方法部分源碼如下。
- private void exit() {
- ......
- threadLocals = null;
- inheritableThreadLocals = null;
- ......
- }
因此,只要Thread對象的exit方法被調(diào)用,就不會存在內(nèi)存泄漏問題。只要線程用完就銷毀,那么使用InheritableThreadLocal,在子線程中不需要調(diào)用InheritableThreadLocal的remove方法也不會存在內(nèi)存泄漏的可能。
比如我們在項目中使用InheritableThreadLocal實現(xiàn)將Session傳遞給子線程:
- @GetMapping("/test")
- public SsoUser test() {
- // 獲取登錄用戶
- SsoUser ssoUser = SsoUserManager.curLoggedUser();
- System.out.println(ssoUser.getUserCode());
- // 支持子線程傳遞
- new Thread(() -> {
- try {
- Thread.sleep(100);
- SsoUser ssoUser2 = SsoUserManager.curLoggedUser();
- System.out.println(ssoUser2.getUserCode());
- } catch (InterruptedException e) {
- }
- }).start();
- return ssoUser;
- }
在此案例中,由于子線程只是臨時創(chuàng)建的,所以我們不需要在子線程中調(diào)用InheritableThreadLocal的remove方法,只需要在父線程調(diào)用一次remove方法,因為tomcat的work線程是不會在一次請求結(jié)束后就銷毀的。
現(xiàn)在我們已經(jīng)知道了InheritableThreadLocal是如何實現(xiàn)將數(shù)據(jù)傳遞給子線程的,思考題的答案也就有了一半:由于InheritableThreadLocal只能將線程上下文傳遞給當(dāng)前線程創(chuàng)建的子線程,所以只有線程池中的線程是由當(dāng)前線程創(chuàng)建的才能夠傳遞。
但要知道另一半答案我們還需要從線程池中尋找。
使用不同參數(shù)構(gòu)建的線程池不同,常見的有單線程的線程池、只有固定數(shù)量核心線程的線程池、有固定數(shù)量核心線程和非核心線程的線程池、只有非核心線程的線程池。
線程池的幾個構(gòu)造參數(shù)說明如下:
- corePoolSize:核心線程數(shù),不會被釋放的線程數(shù)量(設(shè)置allowCoreThreadTimeOut為ture時例外);
- maximumPoolSize:線程池的最大線程數(shù),等于核心線程與非核心線程的數(shù)量總和;
- keepAliveTime:非核心線程最大空閑等待時間,在指定空閑時間后如果還沒有任務(wù)則釋放該線程;
- workQueue:任務(wù)隊列,當(dāng)核心線程數(shù)用完時,任務(wù)被放入隊列。
一、線程池是臨時線程池
如果線程池是在當(dāng)前線程創(chuàng)建的,且任務(wù)都是由當(dāng)前線程提交的,線程池用完就消毀了,那么不管是哪種線程池,池中的線程都是由當(dāng)前線程所創(chuàng)建,在這種場景下,InheritableThreadLocal能夠?qū)ontext傳給給線程池中的任一線程。
二、線程池是全局線程池
如果線程池是全局線程池:
- 沒有核心線程且非核心線程的keepAliveTime等于0:線程都是用到才創(chuàng)建,且由于keepAliveTime等于0,線程用完可能就釋放了,在這種場景下,相當(dāng)于是由當(dāng)前線程創(chuàng)建子線程執(zhí)行任務(wù),因此能夠?qū)崿F(xiàn)透傳;
- 沒有非核心線程:前(核心線程數(shù))個任務(wù)的提交都會創(chuàng)建線程,也都是由當(dāng)前線程創(chuàng)建,所以只有這幾個任務(wù)的執(zhí)行是能夠正常獲取父線程寫入InheritableThreadLocal的數(shù)據(jù)的,后面提交的任務(wù)就不知道會被哪個核心線程拉取執(zhí)行了;
- 其它:....
因此,如果線程池是全局線程池,那么無論是哪個情況,都不建議使
本文轉(zhuǎn)載自微信公眾號「Java藝術(shù)」,可以通過以下二維碼關(guān)注。轉(zhuǎn)載本文請聯(lián)系Java藝術(shù)公眾號。