ThreadLocal 的使用场景

news2025/1/10 21:01:02

在现代电商平台中，ThreadLocal 常用于以下场景，特别是与线程隔离相关的业务中，以提高性能和简化上下文传递。

1. 用户上下文信息管理
场景：在用户发起的每次请求中，需要携带用户 ID、角色、权限等信息，而这些信息需要在整个请求线程中传递。
应用：
将用户上下文信息（如 userId、sessionId 等）存储到 ThreadLocal 中，便于在业务处理的不同阶段获取，而无需显式传递。
如在用户权限校验、订单处理、日志记录中频繁使用。

public class UserContext {
    private static ThreadLocal<String> userId = new ThreadLocal<>();
    
    public static void setUserId(String id) {
        userId.set(id);
    }
    
    public static String getUserId() {
        return userId.get();
    }
    
    public static void clear() {
        userId.remove();
    }
}

2. 分布式链路追踪
场景：电商平台中调用链复杂，每个请求需要唯一的 TraceID 来追踪整个分布式系统的调用链。
应用：
将 TraceID 存储在 ThreadLocal 中，方便在每个服务组件中传递和使用。
例如，在日志框架中打印 TraceID，帮助快速定位问题。

public class TraceContext {
    private static ThreadLocal<String> traceId = new ThreadLocal<>();
    
    public static void setTraceId(String id) {
        traceId.set(id);
    }
    
    public static String getTraceId() {
        return traceId.get();
    }
    
    public static void clear() {
        traceId.remove();
    }
}

3. 数据库连接管理
场景：在电商业务中，某些操作需要特定的数据库隔离级别，或者需要在一个线程内保持同一个数据库连接。
应用：
利用 ThreadLocal 缓存数据库连接，避免频繁获取连接。
适用于手动管理事务时，在一个线程内保证同一连接被复用。

public class ConnectionManager {
    private static ThreadLocal<Connection> connectionThreadLocal = new ThreadLocal<>();
    
    public static Connection getConnection() {
        Connection conn = connectionThreadLocal.get();
        if (conn == null) {
            conn = createNewConnection(); // 创建新连接
            connectionThreadLocal.set(conn);
        }
        return conn;
    }
    
    public static void closeConnection() {
        Connection conn = connectionThreadLocal.get();
        if (conn != null) {
            conn.close();
            connectionThreadLocal.remove();
        }
    }
}

4. 防重复提交
场景：电商平台中，防止用户重复提交订单或操作（例如秒杀、抢购）。
应用：
使用 ThreadLocal 存储请求的唯一标识符（如 RequestId 或 Token），以便在一次线程执行中识别请求是否已处理。

public class RequestContext {
    private static ThreadLocal<String> requestId = new ThreadLocal<>();
    
    public static void setRequestId(String id) {
        requestId.set(id);
    }
    
    public static String getRequestId() {
        return requestId.get();
    }
    
    public static void clear() {
        requestId.remove();
    }
}

5. 多数据源动态切换
场景：在订单查询、数据报表等场景下，不同业务模块可能需要访问不同的数据源。
应用：
利用 ThreadLocal 存储当前线程需要使用的数据源标识符，在数据源路由时动态切换。

public class DataSourceContext {
    private static ThreadLocal<String> dataSourceKey = new ThreadLocal<>();
    
    public static void setDataSourceKey(String key) {
        dataSourceKey.set(key);
    }
    
    public static String getDataSourceKey() {
        return dataSourceKey.get();
    }
    
    public static void clear() {
        dataSourceKey.remove();
    }
}

6. 日志输出优化
场景：需要在同一个线程的所有日志中打印统一的信息（如用户 ID、请求 ID 等）。
应用：
将公共信息（如用户 ID、TraceID 等）存入 ThreadLocal，由日志框架在打印日志时自动获取。

public class LogContext {
    private static ThreadLocal<String> logInfo = new ThreadLocal<>();
    
    public static void setLogInfo(String info) {
        logInfo.set(info);
    }
    
    public static String getLogInfo() {
        return logInfo.get();
    }
    
    public static void clear() {
        logInfo.remove();
    }
}