你問這誰會啊?ThreadLocal 父子線程之間該如何傳遞數據?
忘記之前是哪個公司面試的時候問到的,并不是一個常見的問題,我當時也沒回答正確,就按照線程通信那一套比如什么 synchronized、Locks、volatile 啥的 XJB 說的,面試完找了些資料今天整理了下分享給大家~
ThreadLocal 的具體原理這篇文章就不解釋了,能干啥大伙兒都倒背如流,其實就兩點:
- 鏈路透傳(通俗來說就是方便做參數傳遞,不用在調用方法時攜帶一堆請求參數)
- 線程隔離
每個線程都有自己的一個 ThreadLocalMap,ThreadLocal 持有的數據就是存在這個 Map 里的(Thread.ThreadLocalMap threadLocals),所以能夠實現線程隔離,畢竟每個線程的 ThreadLocalMap 都是不一樣的
那如果子線程想要拿到父線程的中的 ThreadLocal 值怎么辦呢?
比如會有以下的這種代碼的實現。在子線程中調用 get 時,我們拿到的 Thread 對象是當前子線程對象,對吧,每個線程都有自己獨立的 ThreadLocal,那么當前子線程的 ThreadLocalMap 是 null 的(而父線程,也就是 main 線程中的 ThreadLocalMap 是有數據的),所以我們得到的 value 也是 null
結果輸出如下:
要如何解決這個問題呢?
我們先來從 Thread 類中找找思路:
你會發現,在 ThreadLocalMap threadLocals 的下方,還有一個 ThreadLocalMap 變量 inherittableThreadLocals,inherit 翻譯為繼承
先看下這個變量的注釋:InheritableThreadLocal values pertaining to this thread. This map is maintained by the InheritableThreadLocal class.
oho,這里出現了一個渣渣輝都從未體驗過的傳新類:InheritableThreadLocal
翻譯一下注釋,大概就是,如果你使用 InheritableThreadLocal,那么保存的數據都已經不在原來的 ThreadLocal.ThreadLocalMap threadLocals 里面了,而是在一個新的 ThreadLocal.ThreadLocalMap inheritableThreadLocals 變量中了。
所以,如果想讓上面那段代碼中,子線程能夠拿到父線程的 ThreadLocal 值,只需要把 ThreadLocal 聲明改為 InheritableThreadLocal 就可以了
下面我們具體來看下 InheritableThreadLocal 是怎么做到父子線程傳值的。
首先看下 new Thread 的時候線程都做了些什么 Thread#init()
核心其實就是上面幾句代碼,如果你設置了 inheritableThreadLocals 變量,那么 Thread 就會把父線程 ThreadLocal threadLocals 中的所有數據都 copy 到子線程的 InheritableThreadLocal inheritableThreadLocals 中。
而且,copy 調用的 createInheritedMap 方法其實是一個淺拷貝函數,key 和 value 都是原來的引用地址,這里所謂的 copy 其實就是把一個 Map 中的數據復制到另一個 Map 中:
至此,大致的解釋了 InheritableThreadLocal 為什么能解決父子線程傳遞 Threadlcoal 值的問題了,總結下:
在創建InheritableThreadLocal 對象的時候賦值給線程的 t.inheritableThreadLocals 變量。
在創建新線程的時候會 check 父線程中 t.inheritableThreadLocals 變量是否為 null,如果不為 null 則 copy 一份數據到子線程的 t.inheritableThreadLocals 成員變量中去。
InheritableThreadLocal 重寫了 getMap(Thread) 方法,所以 get 的時候,就會從 t.inheritableThreadLocals 中拿到 ThreadLocalMap 對象,從而實現了可以拿到父線程 ThreadLocal 中的值。