Skip to content

Commit 0f9d4cc

Browse files
authored
Merge pull request #2561 from 1020325258/12-16
增加 ThreadLocal 异步线程场景下值传递内容
2 parents 1003ddf + 6d35f83 commit 0f9d4cc

File tree

1 file changed

+68
-0
lines changed

1 file changed

+68
-0
lines changed

docs/java/concurrent/java-concurrent-questions-03.md

+68
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,74 @@ static class Entry extends WeakReference<ThreadLocal<?>> {
141141
1. 在使用完 `ThreadLocal` 后,务必调用 `remove()` 方法。 这是最安全和最推荐的做法。 `remove()` 方法会从 `ThreadLocalMap` 中显式地移除对应的 entry,彻底解决内存泄漏的风险。 即使将 `ThreadLocal` 定义为 `static final`,也强烈建议在每次使用后调用 `remove()`
142142
2. 在线程池等线程复用的场景下,使用 `try-finally` 块可以确保即使发生异常,`remove()` 方法也一定会被执行。
143143

144+
### 如何跨线程传递 ThreadLocal 的值?
145+
146+
由于 `ThreadLocal` 的变量值存放在 `Thread` 里,而父子线程属于不同的 `Thread` 的。因此在异步场景下,父子线程的 `ThreadLocal` 值无法进行传递。
147+
148+
如果想要在异步场景下传递 `ThreadLocal` 值,有两种解决方案:
149+
150+
- `InheritableThreadLocal``InheritableThreadLocal` 是 JDK1.2 提供的工具,继承自 `ThreadLocal` 。使用 `InheritableThreadLocal` 时,会在创建子线程时,令子线程继承父线程中的 `ThreadLocal` 值,但是无法支持线程池场景下的 `ThreadLocal` 值传递。
151+
- `TransmittableThreadLocal``TransmittableThreadLocal` (简称 TTL) 是阿里巴巴开源的工具。`TTL` 可以在线程池的场景下支持 `ThreadLocal` 值传递。
152+
153+
#### `InheritableThreadLocal` 原理扩展
154+
155+
`InheritableThreadLocal` 实现了创建异步线程时,继承父线程 `ThreadLocal` 值的功能。该类是 JDK 团队提供的,通过改造 JDK 源码包中的 `Thread` 类来实现创建线程时,`ThreadLocal` 值的传递。
156+
157+
**`InheritableThreadLocal` 的值存储在哪里?**
158+
159+
`Thread` 类中添加了一个新的 `ThreadLocalMap` ,命名为 `inheritableThreadLocals` ,该变量用于存储需要跨线程传递的 `ThreadLocal` 值。如下:
160+
161+
```JAVA
162+
class Thread implements Runnable {
163+
ThreadLocal.ThreadLocalMap threadLocals = null;
164+
ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;
165+
}
166+
```
167+
168+
**如何完成 `ThreadLocal` 值的传递?**
169+
170+
通过改造 `Thread` 类的构造方法来实现,在创建 `Thread` 线程时,拿到父线程的 `inheritableThreadLocals` 变量赋值给子线程即可。相关代码如下:
171+
172+
```JAVA
173+
// Thread 的构造方法会调用 init() 方法
174+
private void init(/* ... */) {
175+
// 1、获取父线程
176+
Thread parent = currentThread();
177+
// 2、将父线程的 inheritableThreadLocals 赋值给子线程
178+
if (inheritThreadLocals && parent.inheritableThreadLocals != null)
179+
this.inheritableThreadLocals =
180+
ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
181+
}
182+
```
183+
184+
#### `TransmittableThreadLocal` 原理扩展
185+
186+
JDK 默认没有支持线程池场景下 `ThreadLocal` 值传递的功能,因此阿里巴巴开源了一套工具 `TransmittableThreadLocal` 来实现该功能。
187+
188+
阿里巴巴无法改动 JDK 的源码,因此他内部通过 **装饰器模式** 在原有的功能上做增强,以此来实现线程池场景下的 `ThreadLocal` 值传递。
189+
190+
TTL 改造的地方有两处:
191+
192+
- 实现自定义的 `Thread` ,在 `run()` 方法内部做 `ThreadLocal` 变量的赋值操作。
193+
194+
- 基于 **线程池** 进行装饰,在 `execute()` 方法中,不提交 JDK 内部的 `Thread` ,而是提交自定义的 `Thread`
195+
196+
如果想要查看相关源码,可以引入 Maven 依赖进行下载。
197+
198+
```XML
199+
<dependency>
200+
<groupId>com.alibaba</groupId>
201+
<artifactId>transmittable-thread-local</artifactId>
202+
<version>2.12.0</version>
203+
</dependency>
204+
```
205+
206+
#### 相关应用场景
207+
208+
**线上服务压测** 场景下,会使用 `ThreadLocal` 存储压测标记,来区分压测流量和线上真实流量。
209+
210+
如果使用默认的 `ThreadLocal` ,就会导致在异步线程、线程池场景下, `ThreadLocal` 存储的压测标记丢失,从而造成比较严重的后果。
211+
144212
## 线程池
145213

146214
### 什么是线程池?

0 commit comments

Comments
 (0)