ThreadStatic 的用法 

1. 普通的 static 變量

相信很多朋友在代碼中都使用過(guò) static 變量，它的好處多多，比如說(shuō)我經(jīng)常會(huì)用 static 去做一個(gè)進(jìn)程級(jí)緩存，從而提高程序的性能，當(dāng)然你也可以作為一個(gè)非常好的一級(jí)緩存，如下代碼：

    public class Test
    {
        public static Dictionary<int, string> cachedDict = new Dictionary<int, string>();
    }

剛才我也說(shuō)到了，這是一個(gè)進(jìn)程級(jí)的緩存，多個(gè)線程都看得見(jiàn)，所以在多線程的環(huán)境下，你需要特別注意同步的問(wèn)題。要么使用鎖，要么使用 ConcurrentDictionary，我覺(jué)得這也是一個(gè)思維定式，很多時(shí)候思維總是在現(xiàn)有基礎(chǔ)上去修補(bǔ)，去亡羊補(bǔ)牢，而沒(méi)有跳出這個(gè)思維從根基上去處理，說(shuō)這么多是什么意思呢？我舉一個(gè)例子：

在市面上常見(jiàn)的鏈?zhǔn)礁櫩蚣苤?，比如說(shuō)：Zikpin，SkyWalking，會(huì)使用一些集合去存儲(chǔ)跟蹤當(dāng)前線程的一些鏈路信息，比如說(shuō) A -> B -> C -> D -> B -> A，常規(guī)的思維就像上面說(shuō)的一樣，定義一個(gè)全局 cachedDict，然后使用各種同步機(jī)制，其實(shí)你也可以降低 cachedDict 的訪問(wèn)作用域，將 全局訪問(wèn) 改成 Thread級(jí)訪問(wèn)，這難道不是更好的解決思路嗎？

2. 用 ThreadStatic 標(biāo)記 static 變量

要想做到 Thread級(jí)作用域，實(shí)現(xiàn)起來(lái)非常簡(jiǎn)單，在 cachedDict 上打一個(gè) ThreadStatic 特性即可，修改代碼如下：

    public class Test
    {
        [ThreadStatic]
        public static Dictionary<int, string> cachedDict = new Dictionary<int, string>();
    }
 

接下來(lái)可以開(kāi)多個(gè)線程給 cachedDict 灌數(shù)據(jù)，看看 dict 是不是 Thread級(jí)作用域，實(shí)現(xiàn)代碼如下：

    class Program
    {
        static void Main(string[] args)
        {
            var task1 = Task.Run(() =>
            {
                if (Test.cachedDict == null) Test.cachedDict = new Dictionary<int, string>();
                Test.cachedDict.Add(1, "mary");
                Test.cachedDict.Add(2, "john");

                Console.WriteLine($"thread={Thread.CurrentThread.ManagedThreadId} 的 dict 有記錄: {Test.cachedDict.Count}");
            });

            var task2 = Task.Run(() =>
            {
                if (Test.cachedDict == null) Test.cachedDict = new Dictionary<int, string>();
                Test.cachedDict.Add(3, "python");
                Test.cachedDict.Add(4, "jaskson");
                Test.cachedDict.Add(5, "elen");

                Console.WriteLine($"thread={Thread.CurrentThread.ManagedThreadId} 的 dict 有記錄: {Test.cachedDict.Count}");
            });

            Console.ReadLine();
        }
    }

    public class Test
    {
        [ThreadStatic]
        public static Dictionary<int, string> cachedDict = new Dictionary<int, string>();
    }
 

 

從結(jié)果來(lái)看，確實(shí)是一個(gè) Thread 級(jí)，而且還避免了線程間同步開(kāi)銷，哈哈????，這么神奇的東西，難怪有讀者想看看底層到底是怎么實(shí)現(xiàn)的。

用 Windbg 挖 ThreadStatic 

1. 對(duì) TEB 和 TLS 的認(rèn)識(shí)

• TEB (Thread Environment Block)

每一個(gè)線程都有一份屬于自己專屬的私有數(shù)據(jù)，這些數(shù)據(jù)就放在 Thread 的 TEB 中，如果你想看的話，可以在 windbg 中打印出來(lái)。

0:000> !teb
TEB at 0000001e1cdd3000
    ExceptionList:        0000000000000000
    StackBase:            0000001e1cf80000
    StackLimit:           0000001e1cf6e000
    SubSystemTib:         0000000000000000
    FiberData:            0000000000001e00
    ArbitraryUserPointer: 0000000000000000
    Self:                 0000001e1cdd3000
    EnvironmentPointer:   0000000000000000
    ClientId:             0000000000005980 . 0000000000005aa8
    RpcHandle:            0000000000000000
    Tls Storage:          000001b599d06db0
    PEB Address:          0000001e1cdd2000
    LastErrorValue:       0
    LastStatusValue:      c0000139
    Count Owned Locks:    0
    HardErrorMode:        0
 

從 teb 的結(jié)構(gòu)中可以看出，既有 線程本地存儲(chǔ)(TLS)，也有異常相關(guān)信息的存儲(chǔ) (ExceptionList) 等等相關(guān)信息。

• TLS (Thread Local Storage)

進(jìn)程會(huì)在啟動(dòng)后給 TLS 分配總共 1088 個(gè)槽位，每個(gè)線程都會(huì)分配一個(gè)專屬的 tlsindex 索引，并且擁有一組 slots 槽位，可以用 windbg 去驗(yàn)證一下。

0:000> !tls
Usage:
tls <slot> [teb]
  slot:  -1 to dump all allocated slots
         {0-0n1088} to dump specific slot
  teb:   <empty> for current thread
         0 for all threads in this process
         <teb address> (not threadid) to dump for specific thread.
0:000> !tls -1
TLS slots on thread: 5980.5aa8
0x0000 : 0000000000000000
0x0001 : 0000000000000000
0x0002 : 0000000000000000
0x0003 : 0000000000000000
0x0004 : 0000000000000000
...
0x0019 : 0000000000000000
0x0040 : 0000000000000000

0:000> !t                                                                                                        Lock  
 DBG   ID OSID ThreadOBJ           State GC Mode     GC Alloc Context                  Domain           Count Apt Exception
   0    1 5aa8 000001B599CEED90    2a020 Preemptive  000001B59B9042F8:000001B59B905358 000001b599cdb130 1     MTA 
   5    2  90c 000001B599CF4930    2b220 Preemptive  0000000000000000:0000000000000000 000001b599cdb130 0     MTA (Finalizer) 
   7    3   74 000001B59B7272A0  102a220 Preemptive  0000000000000000:0000000000000000 000001b599cdb130 0     MTA (Threadpool Worker) 
   9    4 2058 000001B59B7BAFF0  1029220 Preemptive  0000000000000000:0000000000000000 000001b599cdb130 0     MTA (Threadpool Worker) 

從上面的 {0-0n1088} to dump specific slot 中可以看出，進(jìn)程中總會(huì)有 1088 個(gè)槽位，而且當(dāng)前主線程 5aa8 擁有 27 個(gè) slot 槽位。

好了，基本概念介紹完了，接下來(lái)準(zhǔn)備分析一下匯編代碼了。 

2. 從匯編代碼中尋找答案

為了更好的用 windbg 去挖，我就定義一個(gè)簡(jiǎn)單的 ThreadStatic int 變量，代碼如下：

    class Program
    {
        [ThreadStatic]
        public static int i = 0;

        static void Main(string[] args)
        {
            i = 10;   // 12 line

            var num = i;

            Console.ReadLine();
        }
    }
 

接下來(lái)用 !U 反匯編一下 Main 函數(shù)的代碼，著重看一下第 12 行代碼的 i = 10;。

0:000> !U /d 00007ffbe0ae0ffb
E:\net5\ConsoleApp5\ConsoleApp5\Program.cs @ 12:
00007ffb`e0ae0fd6 48b9b0fbb7e0fb7f0000 mov rcx,7FFBE0B7FBB0h
00007ffb`e0ae0fe0 ba01000000      mov     edx,1
00007ffb`e0ae0fe5 e89657a95f      call    coreclr!JIT_GetSharedNonGCThreadStaticBase (00007ffc`40576780)
00007ffb`e0ae0fea c7401c0a000000  mov     dword ptr [rax+1Ch],0Ah
 

從匯編指令上來(lái)看，最后的 10 賦給了 rax+1Ch 的低32位，那 rax 的地址從哪里來(lái)的呢？可以看出核心邏輯在 JIT_GetSharedNonGCThreadStaticBase 方法內(nèi)，接下來(lái)就得研究一下這個(gè)方法都干嘛了。 

3. 調(diào)試核心函數(shù) JIT_GetSharedNonGCThreadStaticBase

接下來(lái)在第 12 處設(shè)置一個(gè)斷點(diǎn) !bpmd Program.cs:12 處，方法的簡(jiǎn)化匯編代碼如下：

    coreclr!JIT_GetSharedNonGCThreadStaticBase:
00007ffc`2c38679a 448b0dd7894300         mov     r9d, dword ptr [coreclr!_tls_index (00007ffc`2c7bf178)]
00007ffc`2c3867a1 654c8b042558000000     mov     r8, qword ptr gs:[58h]
00007ffc`2c3867aa b908000000             mov     ecx, 8
00007ffc`2c3867af 4f8b04c8               mov     r8, qword ptr [r8+r9*8]
00007ffc`2c3867b3 4e8b0401               mov     r8, qword ptr [rcx+r8]
00007ffc`2c3867b7 493b8060040000         cmp     rax, qword ptr [r8+460h]
00007ffc`2c3867be 732b                   jae     coreclr!JIT_GetSharedNonGCThreadStaticBase+0x6b (00007ffc`2c3867eb)
00007ffc`2c3867c0 4d8b8058040000         mov     r8, qword ptr [r8+458h]
00007ffc`2c3867c7 498b04c0               mov     rax, qword ptr [r8+rax*8]
00007ffc`2c3867cb 4885c0                 test    rax, rax
00007ffc`2c3867ce 741b                   je      coreclr!JIT_GetSharedNonGCThreadStaticBase+0x6b (00007ffc`2c3867eb)
00007ffc`2c3867d0 8bca                   mov     ecx, edx
00007ffc`2c3867d2 f644011801             test    byte ptr [rcx+rax+18h], 1
00007ffc`2c3867d7 7412                   je      coreclr!JIT_GetSharedNonGCThreadStaticBase+0x6b (00007ffc`2c3867eb)
00007ffc`2c3867d9 488b4c2420             mov     rcx, qword ptr [rsp+20h]
00007ffc`2c3867de 4833cc                 xor     rcx, rsp
00007ffc`2c3867e1 e89a170600             call    coreclr!__security_check_cookie (00007ffc`2c3e7f80)
00007ffc`2c3867e6 4883c438               add     rsp, 38h
00007ffc`2c3867ea c3                     ret  
 

接下來(lái)我仔細(xì)分析下這里的 mov 操作。 

1) dword ptr [coreclr!_tls_index (00007ffc`2c7bf178)]

這個(gè)很簡(jiǎn)單，獲取該線程專屬的 tls_index 索引 

2) qword ptr gs:[58h]

這里的 gs:[58h] 是什么意思呢？應(yīng)該有朋友知道，gs寄存器 是專門用于存放當(dāng)前線程的 teb 地址，后面的 58 表示在 teb 地址上的偏移量，那問(wèn)題來(lái)了，這個(gè)地址到底指向誰(shuí)了呢？其實(shí)你可以把 teb 的數(shù)據(jù)結(jié)構(gòu)給打印出來(lái)就明白了。

0:000> dt teb
coreclr!TEB
   +0x000 NtTib            : _NT_TIB
   +0x038 EnvironmentPointer : Ptr64 Void
   +0x040 ClientId         : _CLIENT_ID
   +0x050 ActiveRpcHandle  : Ptr64 Void
   +0x058 ThreadLocalStoragePointer : Ptr64 Void
   +0x060 ProcessEnvironmentBlock : Ptr64 _PEB
   ...
 

上面這句 +0x058 ThreadLocalStoragePointer : Ptr64 Void 可以看出，其實(shí)就是指向 ThreadLocalStoragePointer 。 

3) qword ptr [r8+r9*8]

有了前兩步的基礎(chǔ)，這句匯編就很簡(jiǎn)單了，它做了一個(gè)索引操作: ThreadLocalStoragePointer[tls_index] ，對(duì)不對(duì)，從而獲取屬于該線程的 tls 內(nèi)容，這個(gè) ThreadStatic 的變量就會(huì)存放在這個(gè)數(shù)組的某一個(gè)內(nèi)存塊中。