一、什麼是 ThreadLocal

ThreadLocal 提供了執行緒的區域性變數，每個執行緒都可以通過 set() 和 get() 來對這個區域性變數進行操作，但不會和其他執行緒的區域性變數衝突，實現了執行緒間的據隔離。

簡單講：一個獲取使用者的請求執行緒 A，如果向 ThreadLocal 填充變數 AValue（只能被執行緒 A 操作），該變數對其他獲取使用者的請求執行緒 B、C...是隔離的.

最簡單的使用方式：

類似一次 HTTP 請求執行緒中，利用 ThreadLocal 儲存 Cookie 物件，進行狀態管理。set Cookie：

private ThreadLocal httpThreadLocal = new ThreadLocal();

httpThreadLocal.set(“Cookie: sid=13420771402233”)
 

上面儲存格式是 String ，實際場景儲存的是具體的物件。在這次 HTTP 請求過程中，任何時候都可以獲取 Cookie 。獲取方式很簡單 get Cookie：

String cookieValue = (String) httpThreadLocal.get();

Thread 與 ThreadLocal 物件引用關係圖

二、你熟悉的場景

2.1 資料庫連線池

比如一次請求執行緒進來，業務 Dao 需要更新 user 表和 user-detail 表。如果是 new 出兩個資料庫 Connection ，分別不同的 Connection 操作 user 表和 user-detail 表，就無法保證事務。那麼資料庫連線池是如何保證的？

答案是：利用 ThreadLocal 儲存唯一 Connection 物件。每次請求執行緒，pool.getConnection 獲取連線的時候都會這樣操作：

• 會從 ThreadLocal 獲取 Connection 物件。如果有，則保證了後面多個數據庫操作共用同一個 Connection ，從而保證了事務。
• 如果沒有，往 ThreadLocal 新增Connection 物件，並返回到執行緒

錯誤的做法

public class XXXService {

    private Connection conn;
}

因為 conn 是執行緒不安全的。這樣會導致多個請求公用一個連線。請求量很大的情況下，延遲各種。你懂。

因此，使用 ThreadLocal 保證每個請求執行緒的 Connection 是唯一的。即每個執行緒有自己的連線。

繼續講到 Spring 框架，在事務開始時，會給當前執行緒一個Jdbc Connection,在整個事務過程，都是使用該執行緒繫結的connection來執行資料庫操作，實現了事務的隔離性。Spring框架裡面就是用的ThreadLocal來實現這種隔離

2.2 HTTP Cookie

比如你訪問百度、我訪問百度，會有不同 Cookie 。而且你不能訪問我的 Cookie，我也不能。顧名思義，使用 ThreadLocal 保證每個 HTTP 請求執行緒的 Cookie 是唯一的。

Cookie 這樣才能做 Session 等狀態管理。

三、實戰場景

總結一下就是：ThreadLocal 可以讓同一個執行緒中上下文之間資料共享

在上面章節 二、你熟悉的場景 其實介紹了很多現有場景。那麼我這邊具體的實戰場景是什麼？

簡單的例子：

適用滿足這兩個條件的場景：1.每個執行緒獨有的一些資訊，2.這些資訊又會在多個方法或類中用到。

1. 一個請求執行緒，裡面有兩個非同步小執行緒，各有一個方法。分別處理 A 或 B 業務
2. 一種方法是傳遞不可變的入參
3. 另一種就是 ThreadLocal，放在 ThreadLocal 的入參，會被各個方法共享。而且多個請求執行緒互不影響

複雜的例子：

一次發貨操作：會根據入參，進行元件化、流程編排話。那麼入參會被各個地方用到，而且有些流程元件是非同步的（類似 new thread 操作的）。這時候可以定一個 XXContext 上下文：

public class XXContext {
    
    private static ThreadLocal<Map<Class<?>, Object>> context = new InheritableThreadLocal<>();
    
    /**
     * 把引數設定到上下文的Map中
     */
    public static void put(Object obj) {
        Map<Class<?>, Object> map = context.get();
        if (map == null) {
            map = new HashMap<>();
            context.set(map);
        }
        if (obj instanceof Enum) {
            map.put(obj.getClass().getSuperclass(), obj);
        } else {
            map.put(obj.getClass(), obj);
        }
    }
    
    /**
     * 從上下文中，根據類名取出引數
     */
    @SuppressWarnings("unchecked")
    public static <T> T get(Class<T> c) {
        Map<Class<?>, Object> map = context.get();
        if (map == null) {
            return null;
        }
        return (T) map.get(c);
    }
    
    /**
     * 清空ThreadLocal的資料
     */
    public static void clean() {
        context.remove();
    }
}

程式碼解析：

• 都是 static 操作，類似 DateUtil 玩法
• 記得每次請求執行緒後清理。可以 AOP 去清理，加個註解就行。因為同一個請求執行緒可能被業務方公用。

（完）

相關推薦