文章目录
ThreadLocal 介绍
ThreadLocal 是 Java 中用于实现线程局部变量的一个类。它允许每个线程都拥有自己的变量副本,从而避免了多个线程之间的共享和竞争问题。在多线程环境下,当需要保持线程安全性的数据访问,或者需要在多个方法之间共享数据但又不想通过传递参数的方式时,ThreadLocal 提供了一个很好的解决方案。
当set的时候,获取的是当前线程对象中的ThreadLocalMap对象,然后将当前threadLocal对象本身当做key存入
使用场景
- 用户会话管理:在 Web 应用中,ThreadLocal 可以用来存储用户会话信息,确保每个请求的处理线程都能访问到对应的会话数据。
- 数据库连接管理:在数据库连接池中,可以使用 ThreadLocal 来存储每个线程的数据库连接,避免线程之间的冲突。
- 事务管理:在事务处理时,可以使用 ThreadLocal 来存储当前事务的状态,以确保在整个事务处理过程中数据的一致性。
为什么 ThreadLocalMap 中的 Entry 继承 WeakReference
- ThreadLocal 内部通过 ThreadLocalMap 来存储每个线程的变量副本。ThreadLocalMap 的 Entry 继承自 WeakReference,主要是为了减少内存泄漏的风险。
原因:
- 当 ThreadLocal 变量不再被引用时(即没有任何强引用指向它),它可能会被垃圾回收器回收。由于 Entry 的键(即 ThreadLocal 变量)是弱引用,因此即使 ThreadLocal 变量被回收,其对应的 Entry 也不会立即被回收,但是键(ThreadLocal 变量)的引用会被清除。
- 如果没有弱引用,当 ThreadLocal 变量被设置为 null 并希望被回收时,由于 ThreadLocalMap 还持有 ThreadLocal 变量的强引用,导致 ThreadLocal 变量无法被回收,进而可能导致 ThreadLocalMap 中存储的大量数据也无法被回收,造成内存泄漏。 Entry 也会因为失去了键的引用而被移除,从而避免了内存泄漏。
总结:
- 通过使用弱引用,ThreadLocalMap中的键(ThreadLocal对象)可以在不再需要时被垃圾回收器回收,从而避免了上述问题。当ThreadLocal对象被回收后,虽然ThreadLocalMap中对应的Entry的键会变成null,但是只要Entry对象本身还存在(因为它还持有对值的强引用),那么值对象就不会被立即回收。不过,ThreadLocalMap在后续的get、set或rehash操作中会清理这些键为null的Entry,从而避免内存泄漏。
ThreadLocal 内存泄漏问题及解决方法
内存泄漏问题
- 如果 ThreadLocal 变量在不再需要时没有被显式移除(调用 remove() 方法),且线程长时间运行(如在线程池中),那么 ThreadLocalMap 中的 Entry 可能会持续存在,导致内存泄漏。
解决方法 - 显式移除:在 ThreadLocal 变量不再需要时,调用 remove() 方法移除。
- 使用线程池时的注意:确保在任务执行完毕后,调用 remove() 方法清理 ThreadLocal 变量。
阿里的 TransmittableThreadLocal
TransmittableThreadLocal 介绍:
- TransmittableThreadLocal(TTL)是阿里巴巴开源的一个扩展 ThreadLocal,它解决了在线程池等场景下,ThreadLocal 变量在线程间传递时失效的问题。TTL 允许在线程池等场景下,子线程能够继承父线程中的 ThreadLocal 变量。
import com.alibaba.ttl.TransmittableThreadLocal;
public class TransmittableThreadLocalExample {
private static final TransmittableThreadLocal<String> ttl = new TransmittableThreadLocal<>();
public static void main(String[] args) {
ttl.set("Initial Value");
Thread thread = new Thread(() -> {
System.out.println(Thread.currentThread().getName() + ": " + ttl.get()); // 继承父线程的 TTL 值
// 修改 TTL 值
ttl.set("New Value in Child Thread");
// 提交到线程池
ExecutorService executor = Executors.newSingleThreadExecutor();
executor.submit(() -> {
// 默认情况下,子线程不会继承 TTL 值
// 但如果使用 TTL 提供的装饰器包装线程池,则可以继承
System.out.println(Thread.currentThread().getName() + ": " + ttl.get()); // 如果没有包装线程池,这里将输出 null
// ...
});
executor.shutdown();
});
thread.start();
// 注意:实际使用中,应该使用 TTL 提供的线程池包装器来包装你的线程池
}
}
// 注意:上面的代码示例并没有展示如何使用 TTL 提供的线程池包装器,
// 因为它需要额外的依赖和配置。通常,你会在配置线程池时,使用 TTL 提供的 `TtlExecutors` 类来包装你的线程池。