知道ThreadLocal嗎？一起聊聊到底有啥用

摘要：ThreadLocal是 java 提供的一個方便對象在本線程內不同方法中傳遞和獲取的類。用它定義的變量，僅在本線程中可見和維護，不受其他線程的影響，與其他線程相互隔離。

本文分享自華為雲社區《ThreadLocal：線程專屬的變量》，作者： zuozewei 。

ThreadLocal 簡介

ThreadLocal是 java 提供的一個方便對象在本線程內不同方法中傳遞和獲取的類。用它定義的變量，僅在本線程中可見和維護，不受其他線程的影響，與其他線程相互隔離。

那 ThreadLocal 到底解決了什么問題，又適用於什么樣的場景？

This class provides thread-local variables. These variables differ from their normal counterparts in that each thread that accesses one (via its get or set method) has its own, independently initialized copy of the variable. ThreadLocal instances are typically private static fields in classes that wish to associate state with a thread (e.g., a user ID or Transaction ID).
Each thread holds an implicit reference to its copy of a thread-local variable as long as the thread is alive and the ThreadLocal instance is accessible; after a thread goes away, all of its copies of thread-local instances are subject to garbage collection (unless other references to these copies exist).

核心意思是

ThreadLocal 提供了線程本地的實例。它與普通變量的區別在於，每個使用該變量的線程都會初始化一個完全獨立的實例副本。ThreadLocal 變量通常被private static修飾。當一個線程結束時，它所使用的所有 ThreadLocal 相對的實例副本都可被回收。

總的來說，ThreadLocal 適用於每個線程需要自己獨立的實例且該實例需要在多個方法中被使用，也即變量在線程間隔離而在方法或類間共享的場景。后文會通過實例詳細闡述該觀點。另外，該場景下，並非必須使用 ThreadLocal ，其它方式完全可以實現同樣的效果，只是 ThreadLocal 使得實現更簡潔。

ThreadLocal 使用

ThreadLocal 通過 set 方法可以給變量賦值，通過 get 方法獲取變量的值。當然，也可以在定義變量時通過 ThreadLocal.withInitial 方法給變量賦初始值，或者定義一個繼承 ThreadLocal 的類，然后重寫 initialValue 方法。

下面通過如下代碼說明 ThreadLocal 的使用方式：

public class TestThreadLocal
{
    private static ThreadLocal<StringBuilder> builder = ThreadLocal.withInitial(StringBuilder::new);

    public static void main(String[] args)
    {
        for (int i = 0; i < 5; i++)
        {
            new Thread(() -> {
                String threadName = Thread.currentThread().getName();
                for (int j = 0; j < 3; j++)
                {
                    append(j);
                    System.out.printf("%s append %d, now builder value is %s, ThreadLocal instance hashcode is %d, ThreadLocal instance mapping value hashcode is %d\n", threadName, j, builder.get().toString(), builder.hashCode(), builder.get().hashCode());
                }

                change();
                System.out.printf("%s set new stringbuilder, now builder value is %s, ThreadLocal instance hashcode is %d, ThreadLocal instance mapping value hashcode is %d\n", threadName, builder.get().toString(), builder.hashCode(), builder.get().hashCode());
            }, "thread-" + i).start();
        }
    }

    private static void append(int num) {
        builder.get().append(num);
    }

    private static void change() {
        StringBuilder newStringBuilder = new StringBuilder("HelloWorld");
        builder.set(newStringBuilder);
    }
}

在例子中，定義了一個 builder 的 ThreadLocal 對象，然后啟動 5 個線程，分別對 builder 對象進行訪問和修改操作，這兩個操作放在兩個不同的函數 append、change 中進行，兩個函數訪問 builder 對象也是直接獲取，而不是放入函數的入參中傳遞進來。

代碼輸出如下:

thread-0 append 0, now builder value is 0, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 566157654
thread-0 append 1, now builder value is 01, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 566157654
thread-4 append 0, now builder value is 0, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 654647086
thread-3 append 0, now builder value is 0, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1803363945
thread-2 append 0, now builder value is 0, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1535812498
thread-1 append 0, now builder value is 0, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 2075237830
thread-2 append 1, now builder value is 01, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1535812498
thread-3 append 1, now builder value is 01, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1803363945
thread-4 append 1, now builder value is 01, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 654647086
thread-0 append 2, now builder value is 012, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 566157654
thread-0 set new stringbuilder, now builder value is HelloWorld, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1773033190
thread-4 append 2, now builder value is 012, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 654647086
thread-4 set new stringbuilder, now builder value is HelloWorld, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 700642750
thread-3 append 2, now builder value is 012, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1803363945
thread-3 set new stringbuilder, now builder value is HelloWorld, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1706743158
thread-2 append 2, now builder value is 012, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1535812498
thread-2 set new stringbuilder, now builder value is HelloWorld, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1431127699
thread-1 append 1, now builder value is 01, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 2075237830
thread-1 append 2, now builder value is 012, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 2075237830
thread-1 set new stringbuilder, now builder value is HelloWorld, ThreadLocal instance hashcode is 796465865, ThreadLocal instance mapping value hashcode is 1970695360

• 從輸出中 1~6 行可以看出，不同線程訪問的是同一個 builder 對象（不同線程輸出的 ThreadLocal instance hashcode 值相同），但是每個線程獲得的 builder 對象存儲的實例 StringBuilder 不同（不同線程輸出的 ThreadLocal instance mapping value hashcode 值不相同）。
• 從輸出中1~2、9~10 行可以看出，同一個線程中修改 builder 對象存儲的實例的值時，並不會影響到其他線程的 builder 對象存儲的實例（thread-4 線程改變存儲的 StringBuilder 的值並不會引起 thread-0 線程的 ThreadLocal instance mapping value hashcode 值發生改變）
• 從輸出中 9~13 行可以看出，一個線程對 ThreadLocal 對象存儲的值發生改變時，並不會影響其他的線程（thread-0 線程調用 set 方法改變本線程 ThreadLocal 存儲的對象值，本線程的 ThreadLocal instance mapping value hashcode 發生改變，但是 thread-4 的 ThreadLocal instance mapping value hashcode 並沒有因此改變）。

ThreadLocal 原理

ThreadLocal 能在每個線程間進行隔離，其主要是靠在每個 Thread 對象中維護一個 ThreadLocalMap 來實現的。因為是線程中的對象，所以對其他線程不可見，從而達到隔離的目的。那為什么是一個 Map 結構呢。主要是因為一個線程中可能有多個 ThreadLocal 對象，這就需要一個集合來進行存儲區分，而用 Map 可以更快地查找到相關的對象。
ThreadLocalMap 是 ThreadLocal 對象的一個靜態內部類，內部維護一個 Entry 數組，實現類似 Map 的 get 和 put 等操作，為簡單起見，可以將其看做是一個 Map，其中 key 是 ThreadLocal 實例，value 是 ThreadLocal 實例對象存儲的值。

ThreadLocal 適用場景

如上文所述，ThreadLocal 適用於如下場景：

• 每個線程需要有自己單獨的實例，如實現每個線程單例類或每個線程上下文信息（例如事務ID）。
• ThreadLocal 適用於變量在線程間隔離且在方法間共享的場景，提供了另一種擴展 Thread 的方法。如果要保留信息或將信息從一個方法調用傳遞到另一個方法，則可以使用 ThreadLocal 進行傳遞。
• 由於不需要修改任何方法，因此可以提供極大的靈活性。

案例一

這里一個處理 flag 的類，通過 ThreadLocal 使用，可以保證每個請求都擁有唯一的一個追蹤標記。

public class TestFlagHolder {

  private final static ThreadLocal<String> TEST_FLAG = new ThreadLocal<>();

  public static void set(String value) {
    TEST_FLAG.set(value);
  }

  public static String get() {
    return TEST_FLAG.get();
  }

  public static String get4log() {
    if (TEST_FLAG.get() == null) {
      return "-";
    }
    return TEST_FLAG.get();
  }

  public static void remove() {
    TEST_FLAG.remove();
  }

}

案例二

在同一線程中 trace 信息的傳遞：

ThreadLocal<String> traceContext = new ThreadLocal<>();

String traceId = Tracer.startServer();
traceContext.set(traceId) //生成trace信息 傳入threadlocal
...
Tracer.startClient(traceContext.get()); //從threadlocal獲取trace信息
Tracer.endClient();
...
Tracer.endServer();

案例三

給同一個請求的每一行日志增加一個相同的標記。這樣，只要拿到這個標記就可以查詢到這個請求鏈路上所有步驟的耗時了，我們把這個標記叫做 requestId，我們可以在程序的入口處生成一個 requestId，然后把它放在線程的上下文中，這樣就可以在需要時隨時從線程上下文中獲取到 requestId 了。

簡單的代碼實現就像下面這樣：

String requestId = UUID.randomUUID().toString();
ThreadLocal<String> tl = new ThreadLocal<String>(){
    @Override
    protected String initialValue() {
        return requestId;
    }
}; //requestId存儲在線程上下文中
long start = System.currentTimeMillis();
processA();
Logs.info("rid : " + tl.get() + ", process A cost " + (System.currentTimeMillis() - start)); // 日志中增加requestId
start = System.currentTimeMillis();
processB();
Logs.info("rid : " + tl.get() + ", process B cost " + (System.currentTimeMillis() - start));
start = System.currentTimeMillis();
processC();
Logs.info("rid : " + tl.get() + ", process C cost " + (System.currentTimeMillis() - start));

有了 requestId，你就可以清晰地了解一個調用鏈路上的耗時分布情況了。

小結

無論是單體系統還是微服務化架構，無論是鏈路打標還是服務追蹤，你都需要在系統中增加 TraceId，這樣可以將你的鏈路串起來，給你呈現一個完整的問題場景。如果 TraceId 可以在客戶端上生成，在請求業務接口的時候傳遞給服務端，那么就可以把客戶端的日志體系也整合進來，對於問題的排查幫助更大。

同時，在全鏈路壓測框架中，Trace 信息的傳遞功能是基於 ThreadLocal 的。但實際業務中可能會使用異步調用，這樣就會丟失 Trace 信息，破壞了鏈路的完整性。所以在實際的項目建議大家不要輕意使用 ThreadLocal。

參考資料：

• [1]：《高並發系統設計40問》
• [2]：https://juejin.cn/post/6844904016288317448#heading-6

 

點擊關注，第一時間了解華為雲新鮮技術~