阿里开源TransmittableThreadLocal使用经验记录

简介： 本文章主要记录我在一次业务优化中，使用线程池带来的子父线程值传递问题，及使用TransmittableThreadLocal来解决该问题的经验，并对TransmittableThreadLocal原理做些梳理。

业务背景

前段时间在工作项目中，被指派一项优化任务，由于原功能耗时较久，影响使用体感；梳理完该功能涉及到的业务后，发现具备复杂、量大等特点。

将一个大体量数据循环分割成若干小循环，并行执行，结果汇总，是该优化方案中的一环。但原业务中使用ThreadLocal。

ThreadLocal是解决线程间数据隔离性的，源于它将值存储在每个线程Thread的自己变量(ThreadLocal.ThreadLocalMap)中，但是ThreadLocal有个弊端，由于线程间隔离性，子线程无法捕获父线程ThreadLocalMap里的值。一旦使用多线程，需要解决值传递问题。

尝试使用InheritableThreadLocal（Itl）解决问题

Thread线程有两个ThreadLocal.ThreadLocalMap 类型变量threadLocals 和 inheritableThreadLocals。

  ThreadLocal.ThreadLocalMap threadLocals = null;

    ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;

inheritableThreadLocals 是在Thread 初始化方法init中赋值。

if (inheritThreadLocals && parent.inheritableThreadLocals != null)
            this.inheritableThreadLocals =
                ThreadLocal.createInheritedMap(parent.inheritableThreadLocals); // 创建一个新ThreadLocalMap 并将父线程值传递过去

       InheritableThreadLocal是可以解决子父线程值传递问题。我使用Itl 代替了Tl。待优化工作完毕，本地调试及测试通过后。发布到测试环境验证中，在测试环境中发现功能出现了异常，重新回归到本地测试，本地功能多次反复点测却又是完好的。我一度认为环境问题导致，却毫无根据。开始输出日志定位排查，发现测试环境部分线程没有获取到主线程set的值。

    开始思考其中原因。想到了线程池问题，项目中使用了线程池，点测该功能前，线程池是有部分线程存于活跃状态，该部分线程没有经过创建时值传递过程，在执行该功能时，获取不到父线程set的值。

     本地环境却是好的，这也不奇怪了。在我本地程序重启后，直接点测了该功能，线程池中的线程都是在该任务中创建的，全部携带了父线程set的值。

使用TransmittableThreadLocal解决线程池引起的问题

    查阅了资源，决定使用Alibaba开源的TransmittableThreadLocal，根据官网描述：TransmittableThreadLocal(简称TTL)，"在使用线程池等会池化复用线程的执行组件情况下，提供ThreadLocal值的传递功能，解决异步执行时上下文传递的问题"。在使用TransmittableThreadLocal后，测试环境线程池引发的问题确实得到了解决。

官网地址：GitHub - alibaba/transmittable-thread-local: 📌 TransmittableThreadLocal (TTL), the missing Java™ std lib(simple & 0-dependency) for framework/middleware, provide an enhanced InheritableThreadLocal that transmits values between threads even using thread pooling components.

知其然知其所以然

Ttl定义：

TransmittableThreadLocal<T> extends InheritableThreadLocal<T> //集成了Itl,具备Itl的功能.子线程被创建后，继承了父线程的值

 Ttl有个类变量hold：

/**
     * hold是一个InheritableThreadLocal类型变量，持有WeakHashMap,重写了initialValue()和childValue()方法，
     * 异步时父线程的属性是直接作为初始化值赋值给子线程的本地变量对象
     */
    private static final InheritableThreadLocal<WeakHashMap<TransmittableThreadLocal<Object>, ?>> holder =
            new InheritableThreadLocal<WeakHashMap<TransmittableThreadLocal<Object>, ?>>() {
                @Override
                protected WeakHashMap<TransmittableThreadLocal<Object>, ?> initialValue() {
                    return new WeakHashMap<TransmittableThreadLocal<Object>, Object>();
                }