【Takin使用日记】记一次TransmittableThreadLocal引起的业务异常

对于常见的 WEB 容器，Takin 通过增强 org.apache.catalina.core.StandardHostValve#invoke 方法，拦截并解析方法入参的 Request 对象中的 header 数据，判断压测标的信息并将压测标以及链路信息设置到上下文对象，并通过 ThreadLocal 保存该上下文对象，以此来保证当前线程处理的过程中都能获取到压测标以及链路信息。

但是不管是 WEB 容器还是业务代码经常会使用线程池运行代码，这时候 ThreadLocal 就有局限性了，如何做到跨线程池传递我们的上下文对象就显得异常重要了，好在 Alibaba 的解决了这个问题，具体 transmittable-thread-local 的详细内容大家可以通过 git 地址查看，这里我们主要以 git 上的 demo 看看 transmittable-thread-local 如何实现跨线程传递的，主要通过将线程池运行的 Runnable 对象包装成 TtlRunnable 对象并保存当前的上下文信息。这样就完成了对线程池运行任务的压测标以及链路数据信息的透传，至此似乎一切都很完美。

直到某次出现了一个诡异的现象，在使用 Takin 的过程中，有一应用接入 Takin 之后，某个功能无法正常使用，但是无论是应用的日志还是 Takin 的日志都未显示任何异常信息，再追问了无法使用的功能是一个线程池执行任务，当时扩展了 ThreadPoolExecutor 的 beforeExecute 方法和 afterExecute 方法，用以记录任务执行的耗时，并且会打印任务的 taskId 等信息，通过截图的 Debug 信息便可以看到问题出现了，由于使用了 transmittable-thread-local 导致原始的 Runnable 被包装成了 TtlRunnable 对象，在这个过程中如果业务想要再获取 Runnable 对象强转获取一些任务信息，就会出现类无法转换的异常。

解决方案的话目前可以通过增强 beforeExecute 和 afterExecute 方法，对 Runnable 对象进行判断，如果是 TtlRunnable 对象，则通过 TtlRunnable.getRunnable()方法返回原是的 Runnable 对象

Takin 开源详情