【JUC并发编程】14 TransmittableThreadLocal解决线程池中ThreadLocal无法在父子线程之间传递的问题

末蓝、 2023-09-29 12:57 37阅读 0赞

## 一、前言 ##

在上一篇博文（[https://blog.csdn.net/Saintmm/article/details/124461915][https_blog.csdn.net_Saintmm_article_details_124461915]）中介绍了InheritableThreadLocal如何实现父子线程之间传递ThreadLocal信息；最后提到InheritableThreadLocal的一个问题：在使用线程池时InheritableThreadLocal会完全失效。

## 二、TransmittableThreadLocal ##

因为线程池中的线程是会被复用的，所以在使用线程池进行多线程编程时，操作线程本地变量时需要特别注意；下面我们使用TransmittableThreadLocal 解决线程本地变量在线程池之间的传递问题。

### 1、问题复现 ###

**1> 复现案例：**  
创建一个核心线程数为2，最大线程数为2的线程池，在主线程中每次对线程本地变量赋值之后，提交一个线程任务到线程池中运行，如下：

public class InheritableThreadLocalTest1 {
        
    
        public final static InheritableThreadLocal<String> inheritableThreadLocal = new InheritableThreadLocal<>();
        public static final int CORE_THREAD_SIZE = 2;
    
        public static ThreadPoolExecutor executorService = new ThreadPoolExecutor(CORE_THREAD_SIZE,
                CORE_THREAD_SIZE,
                10,
                TimeUnit.SECONDS,
                new LinkedBlockingQueue<>());
    
        public static void main(String[] args) throws InterruptedException {
        
    
            inheritableThreadLocal.set("main thread-1");
            executorService.submit(() -> {
        
                System.out.println("1 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get());
            });
    
            inheritableThreadLocal.set("main thread-2");
            executorService.submit(() -> {
        
                System.out.println("2 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get());
            });
    
            inheritableThreadLocal.set("main thread-3");
            executorService.submit(() -> {
        
                System.out.println("3 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get());
            });
    
            inheritableThreadLocal.set("main thread-4");
            executorService.submit(() -> {
        
                System.out.println("4 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get());
            });
    
        }
    }

**2> 运行结果如下：**  
![在这里插入图片描述][51f17461b3064d45a34e064672500e45.png]

**3> 结果分析：从运行结果结合线程池的运行机制可以看出：**

> 1.  第一任务提交到线程池时，线程池中没有核心线程，创建一个线程，此时可以获取到父线程（main线程）中最新的InheritableThreadLocal值；
> 2.  第二个任务提交到线程池时，线程池中核心线程数为1，1 < 2，所以再创建一个线程，此时也可以获取到父线程（main线程）中最新的InheritableThreadLocal值；
> 3.  第三个任务提交到线程池时，线程池中核心线程数为2，线程进入到阻塞队列，等待空闲线程来调度任务。假如此时coreThread1刚执行完任务1，来执行任务3，则此时的InheritableThreadLocal是历史版本的（main thread-1）。
> 4.  第四个任务提交到线程池时，线程池中核心线程数为2，线程进入到阻塞队列，等待空闲线程来调度任务。假如此时coreThread2刚执行完任务2，来执行任务4，则此时的InheritableThreadLocal是历史版本的（main thread-2）。

因此，我们可以确定，只有当核心线程数不足时才会正确的传递的`InheritableThreadLocal`，而如果线程池中的线程在循环使用了，则不会加载最新的`InheritableThreadLocal`，而是使用历史上的某个版本数据。

线程池部分内容请见博文：[源码分析ThreadPoolExecutor线程池实现原理][ThreadPoolExecutor]。

### 2、解决方案 ###

引入阿里提供的**TransmittableThreadLocal**，github地址为：[https://github.com/alibaba/transmittable-thread-local][https_github.com_alibaba_transmittable-thread-local]。

#### 1）引入maven依赖 ####

<dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>transmittable-thread-local</artifactId>
        <version>2.12.6</version>
    </dependency>

#### 2）代码中使用 ####

**1> 修改线程本地变量类型为TransmittableThreadLocal**

public final static TransmittableThreadLocal<String> inheritableThreadLocal = new TransmittableThreadLocal<>();

其次，官方给定了三种方式保证`TransmittableThreadLocal`的正确传递：

> 2.1> 修饰Runnable和Callable  
> 2.2> TtlExecutors 修饰线程池  
> 2.3> 使用Java Agent来修饰JDK线程池实现类

##### 2.1> 修饰Runnable和Callable： #####

public class TransmittableThreadLocalTest {
        
    
        public final static TransmittableThreadLocal<String> inheritableThreadLocal = new TransmittableThreadLocal<>();
        public static final int CORE_THREAD_SIZE = 2;
    
        public static ThreadPoolExecutor executorService = new ThreadPoolExecutor(CORE_THREAD_SIZE,
                CORE_THREAD_SIZE,
                10,
                TimeUnit.SECONDS,
                new LinkedBlockingQueue<>());
    
        public static void main(String[] args) throws InterruptedException {
        
    
            inheritableThreadLocal.set("main thread-1");
            executorService.submit(TtlRunnable.get(
                            () -> System.out.println("1 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
                    )
            );
    
            inheritableThreadLocal.set("main thread-2");
            executorService.submit(TtlRunnable.get(
                            () -> System.out.println("2 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
                    )
            );
    
            inheritableThreadLocal.set("main thread-3");
            executorService.submit(TtlRunnable.get(
                            () -> System.out.println("3 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
                    )
            );
    
            inheritableThreadLocal.set("main thread-4");
            executorService.submit(TtlRunnable.get(
                            () -> System.out.println("4 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
                    )
            );
    
        }
    }

**运行结果正常，如下所示：**  
![在这里插入图片描述][71791934f4d6419297628f37c2296ca4.png]

##### 2.2> TtlExecutors 修饰线程池： #####

public class TransmittableThreadLocalTest {
        
    
        public final static TransmittableThreadLocal<String> inheritableThreadLocal = new TransmittableThreadLocal<>();
        public static final int CORE_THREAD_SIZE = 2;
    
        public static ExecutorService executorService =
                TtlExecutors.getTtlExecutorService(new ThreadPoolExecutor(CORE_THREAD_SIZE,
                        CORE_THREAD_SIZE,
                        10,
                        TimeUnit.SECONDS,
                        new LinkedBlockingQueue<>())
                );
    
        public static void main(String[] args) throws InterruptedException {
        
    
            inheritableThreadLocal.set("main thread-1");
            executorService.submit(
                    () -> System.out.println("1 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
            );
    
            inheritableThreadLocal.set("main thread-2");
            executorService.submit(
                    () -> System.out.println("2 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
            );
    
            inheritableThreadLocal.set("main thread-3");
            executorService.submit(
                    () -> System.out.println("3 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
            );
    
            inheritableThreadLocal.set("main thread-4");
            executorService.submit(
                    () -> System.out.println("4 obtain inheritableThreadLocal in threadPool: " + inheritableThreadLocal.get())
            );
    
        }
    }

**运行结果和上述一致；**

##### 2.3> 使用Java Agent来修饰JDK线程池实现类； #####

该种方式我们在后续分析`TransmittableThreadLocal`的实现原理之后，再做演示、推论。

可以先参考：[github <–> 23-使用java-agent来修饰jdk线程池实现类][github _ 23-_java-agent_jdk]。

[https_blog.csdn.net_Saintmm_article_details_124461915]: https://blog.csdn.net/Saintmm/article/details/124461915
[51f17461b3064d45a34e064672500e45.png]: https://img-blog.csdnimg.cn/51f17461b3064d45a34e064672500e45.png
[ThreadPoolExecutor]: https://blog.csdn.net/Saintmm/article/details/117405885
[https_github.com_alibaba_transmittable-thread-local]: https://github.com/alibaba/transmittable-thread-local
[71791934f4d6419297628f37c2296ca4.png]: https://img-blog.csdnimg.cn/71791934f4d6419297628f37c2296ca4.png
[github _ 23-_java-agent_jdk]: https://github.com/alibaba/transmittable-thread-local#23-%E4%BD%BF%E7%94%A8java-agent%E6%9D%A5%E4%BF%AE%E9%A5%B0jdk%E7%BA%BF%E7%A8%8B%E6%B1%A0%E5%AE%9E%E7%8E%B0%E7%B1%BB