TransmittableThreadLocal
一、背景
上文说到父子线程传递本地变量可以通过InheritableThreadlocoal
进行传递,但是如果采用线程池,不一定能传递,因为在线程在线程池中的存在不是每次使用都会进行创建,InheritableThreadlocal
是在线程初始化时intertableThreadLocals=true
才会进行拷贝传递。所以若本次使用的子线程是已经被池化的线程,从线程池中取出线下进行使用,是没有经过初始化的过程,也就不会进行父子线程的本地变量拷贝。
由于在日常应用场景中,绝大多数都是会采用线程池的方式进行资源的有效管理。目前知道的阿里有一个开源项目就是为了解决这个问题ThansmittableThreadLocal
。
二、简介
在使用线程池等会池化复用线程的执行组件情况下,提供ThreadLocal
值的传递功能,解决异步执行时上下文传递的问题。
JDK
的InheritableThreadLocal
类可以完成父线程到子线程的值传递。但对于使用线程池等会池化复用线程的执行组件的情况,线程由线程池创建好,并且线程是池化起来反复使用的;这时父子线程关系的ThreadLocal
值传递已经没有意义,应用需要的实际上是把 任务提交给线程池时的ThreadLocal
值传递到 任务执行时。
本章主要介绍使用线程池场景下的问题,TransmittableThreadLocal
还有很多其他的应用场景。传送门
三、基本使用
1.引入
TransimittableThreadLocal
依赖
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>transmittable-thread-local</artifactId>
<version>2.11.5</version>
</dependency>
2.简单使用
public static void main(String[] args) throws InterruptedException {
// 创建线程池
ExecutorService executorService = Executors.newSingleThreadExecutor();
// 线程池经过TtlExecutors工具类包装,返回包装类ExecutorServiceTtlWrapper
executorService = TtlExecutors.getTtlExecutorService(executorService);
// 创建需要传递给线程池本地变量
TransmittableThreadLocal<String> username = new TransmittableThreadLocal<>();
// 首次调用,这时候线程池中还未有线程,就算不使用TTL也可以通过InheritableThreadLocal获取到
// 父线程的本地变量。
username.set("zhangShang");
executorService.submit(new Runnable() {
@Override
public void run() {
System.out.println(username.get());
}
});
// 第二次调用,由于使用的是单一线程的线程池,这时候是复用了上面创建的线程,所以这时通过
// inheritableThreadLocal是获取不到本地变量的。
Thread.sleep(3000);
username.set("liSi");
executorService.submit(new Runnable() {
@Override
public void run() {
System.out.println(username.get());
}
});
Thread.sleep(3000);
username.set("wangWu");
executorService.submit(new Runnable() {
@Override
public void run() {
System.out.println(username.get());
}
});
}
输出结果为:
zhangShang
liSi
wangWu
四、原理
从定义来看,
TransimittableThreadLocal
继承于InheritableThreadLocal
,并实现TtlCopier
接口,它里面只有一个copy
方法。所以主要是对InheritableThreadLocal
的扩展。
public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> implements TtlCopier<T>
在
TransimittableThreadLocal
中添加holder
属性。这个属性的作用就是被标记为具备线程传递资格的对象都会被添加到这个对象中。
// 1. holder本身是一个InheritableThreadLocal对象
// 2. 这个holder对象的value是WeakHashMap<TransmittableThreadLocal<Object>, ?>
// 2.1 WeekHashMap的value总是null,且不可能被使用。
// 2.2 WeekHasshMap支持value=null
private static InheritableThreadLocal<WeakHashMap<TransmittableThreadLocal<Object>, ?