刮痧ThreadLocal

多線程與一致性

為了提高我們程序的性能,很多時候我們都會使用多線程以解決各種場景,但隨之而來的是多線程帶來的數據一致性問題該如何解決。

如何解決一致性問題?
  • 排隊:如果多個線程操作‘同一份數據’,那就排個隊吧,一個一個來,這樣后面一個線程總能得到最新的修改值,例如操作系統中的鎖,管程,屏障等都是這種排隊機制。<ins style="box-sizing: border-box;">缺點是:性能低。</ins>
  • 投票:投票的機制就是多個人同時決策一件事,這個就涉及到了算法,往往會產生很多其他問題,比如欺詐
  • 避免:直觀意思就是避免多個線程之間產生一致性問題,那該如何去做呢?例如git,ThreadLocal正是采用的這種避免的方式來完成多線程的執行
ThreadLocal定義:

定義:ThreadLocal提供了線程局部變量,一個線程局部變量在多個線程中分別由獨立的值(副本)。

提問:既然是每個線程獨有的,為什么不直接在調用線程的時候,在相應的線程方法里聲明這個局部變量呢?

:同一個線程可能會調用到很多不同的類和方法,這樣就要在不同的地方用到這個變量,自己去實現,代價太大,用ThreadLocal更加方便,且線程安全。

線程模型
線程模型

對應每個線程來說都有自己的獨占數據,這些數據事進行來分配的,每個線程都有一個ThreadLocalMap對象,它本身是一個hash表,里面會放一些線程的局部變量,而ThreadLocal的核心也是這個ThreadLocalMap。

4種核心應用場景
1.資源持有:

例如有三個不同的類,在一次web請求中調用這三個類,但是用戶是一個,那么用戶數據就可以保存在一個線程里。 如圖:


資源持有
2.線程一致:

例如JDBC事務,我們每次對數據庫操作都會走getConnection,jdbc保證只要你是同一個線程過來的請求,不管是哪一個part,都返回的是同一個連接,就是使用ThreadLocal來做的,達到維護一致性的目的。Mybatis使用SqlSessionManager保證了我們同一個線程取出來的連接總是同一個。它是如何做到的呢?其實很簡單,就是內部使用了一個ThreadLocal。


線程 一致
3.線程安全:

如果一個線程的調用鏈路比較長,中間出現異常,那我們可以把出錯信息放在ThreadLocal里,然后在后續的鏈路中使用這個值,可以達到多線程在處理這個場景的時候保證線程安全。


線程安全
4.并發計算:

例如一個大的任務,拆分成多個小任務,分別計算,最后再進行結果匯總,那么我們可以把每個線程的計算結果放進ThreadLocal中,最后進行匯總計算。 實現案例:比如需要統計一段時間內某個接口的調用量


并發計算
線程不安全實現:
@RestController
@RequestMapping("orders")
public class OrderController {

    private Integer count = 0;

    @GetMapping("/visit")
    public Integer visit() throws InterruptedException {
        count++;
        Thread.sleep(100);
        return 0;
    }

    @GetMapping("/stat")
    public Integer stat() {
        return count;
    }
}

count++操作,首先我們是從內存里面讀取原來的值,放在了線程本地內存里。然后進行 +1 操作,再寫回到內存里。這個時候如果多個線程操作的話,有可能線程A這邊還沒來得及寫,線程B那邊讀取的是原來的值。這樣子的話就會造成數據不一致的問題。結果就會比預期的小。 結果明顯是count的值與我們所期望的值不一致

如何解決?

當然方法很多,比如加鎖,但今天我們要用ThreadLocal實現

@RestController
@RequestMapping("orders")
public class OrderController {

    private static final ThreadLocal<Integer> TL = ThreadLocal.withInitial(() -> 0);

    @GetMapping("/visit")
    public Integer visit() throws InterruptedException {
        Thread.sleep(100);
        TL.set(TL.get() + 1);
        return 0;
    }

    @GetMapping("/stat")
    public Integer stat() {
        return TL.get();
    }
}

這樣即可達到我們的計數目的。

還有很多方法可以實現,比如我們經常用的原子類Automatic或者synchronized等,他們的實現思想不同,加鎖和原子類使用的是【排隊】思想,而ThreadLocal使用的是【避免】思想,效率更高。

刮痧ThreadLocal源碼

API: ThreadLocal已經支持泛型,該類的類名已經變為ThreadLocal<T>。API方法也相應進行了調整,新版本的API方法分別是void set(T value)、T get()以及T initialValue()。

1.set:
public void set(T value) {
    //(1)獲取當前線程(調用者線程)
    Thread t = Thread.currentThread();
    //(2)以當前線程作為key值,去查找對應的線程變量,找到對應的map
    ThreadLocalMap map = getMap(t);
    //(3)如果map不為null,就直接添加本地變量,key為當前線程,值為添加的本地變量值
    if (map != null)
        map.set(this, value);
    //(4)如果map為null,說明首次添加,需要首先創建出對應的map
    else
        createMap(t, value);
}

ThreadLocalMap getMap(Thread t) {
    return t.threadLocals; //獲取線程自己的變量threadLocals,并綁定到當前調用線程的成員變量threadLocals上
}

void createMap(Thread t, T firstValue) {
     t.threadLocals = new ThreadLocalMap(this, firstValue);
 }
 createMap方法不僅創建了threadLocals,同時也將要添加的本地變量值添加到了threadLocals中。

先拿到當前的線程,然后通過它去拿到一個Map,如果這個Map存在,就把value塞進去,否則就創建一個新的。

2.get:
public T get() {
    //(1)獲取當前線程
    Thread t = Thread.currentThread();
    //(2)獲取當前線程的threadLocals變量
    ThreadLocalMap map = getMap(t);
    //(3)如果threadLocals變量不為null,就可以在map中查找到本地變量的值
    if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    //(4)執行到此處,threadLocals為null,調用該更改初始化當前線程的threadLocals變量
    return setInitialValue();
}

private T setInitialValue() {
    //protected T initialValue() {return null;}
    T value = initialValue();
    //獲取當前線程
    Thread t = Thread.currentThread();
    //以當前線程作為key值,去查找對應的線程變量,找到對應的map
    ThreadLocalMap map = getMap(t);
    //如果map不為null,就直接添加本地變量,key為當前線程,值為添加的本地變量值
    if (map != null)
        map.set(this, value);
    //如果map為null,說明首次添加,需要首先創建出對應的map
    else
        createMap(t, value);
    return value;
}

先通過getMap方法拿到當前線程對應的Map,然后從里面取出value。如果沒有value,就調用ThreadLocal提供的初始化方法,初始化一個值。

3.remove:
public void remove() {
    ThreadLocalMap m = getMap(Thread.currentThread());
    if (m != null) {
        m.remove(this);
    }
}

private void remove(ThreadLocal<?> key) {
    Entry[] tab = table;
    int len = tab.length;
    int i = key.threadLocalHashCode & (len-1);
    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        if (e.get() == key) {
            e.clear();
            expungeStaleEntry(i);
            return;
        }
    }
}

我們在開發一個多線程的程序時,往往會使用線程池。而線程池的功能就是線程的復用。那如果線程池和ThreadLocal在一起就可能會造成問題,所以使用完ThreadLocal,顯式調用一下remove方法。

ThreadLocal不支持繼承性,ThreadLocal類是不能提供子線程訪問父線程的本地變量的,而InheritableThreadLocal類則可以做到這個功能

public class InheritableThreadLocal<T> extends ThreadLocal<T> {

    protected T childValue(T parentValue) {
        return parentValue;
    }

    ThreadLocalMap getMap(Thread t) {
       return t.inheritableThreadLocals;
    }

    void createMap(Thread t, T firstValue) {
        t.inheritableThreadLocals = new ThreadLocalMap(this, firstValue);
    }
}


關注微信公眾號:晏子噠噠

在刮痧技術路上,我們一同成長

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,345評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,494評論 3 416
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,283評論 0 374
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,953評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,714評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,186評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,255評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,410評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,940評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,776評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,976評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,518評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,210評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,642評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,878評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,654評論 3 391
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,958評論 2 373