【JVM】为什么ThreadLocal被设计为弱引用

 

目录

引言

ThreadLocal的弱引用

最佳实践

结论

---

引言

        ThreadLocal提供线程局部变量。这些变量与正常的变量不同，因为每一个线程在访问ThreadLocal实例的时候（通过其get或set方法）都有自己的、独立初始化的变量副本。ThreadLocal实例通常是类中的私有静态字段，使用它的目的是希望将状态（例如，用户ID或事务ID）与线程关联起来。

        实现每一个线程都有自己专属的本地变量副本(自己用自己的变量不麻烦别人，不和其他人共享，人人有份，人各一份)，主要解决了让每个线程绑定自己的值，通过使用get() 和set() 方法，获取默认值或将其值更改为当前线程所存的副本的值从而避免了线程安全问题。

ThreadLocal的弱引用

        ThreadLocal是一个壳子，真正的存储结构是ThreadLocal里ThreadLocalMap内部类，每个Thread对象维护着一个ThreadLocalMap的引用，ThreadLocalMap是ThreadLocal的内部类，用**Entry(k，v)**来进行存储。

        ThreadLocal的set()方法实际上就是往ThreadLocalMap设置值，key是ThreadLocal对象，值Value是传递进来的对象；ThreadLocal的get()方法实际上就是往ThreadLocalMap获取值，key是ThreadLocal对象。

        ThreadLocal本身并不存储值，它只是自己作为一个key来让线程从ThreadLocalMap获取value，正因为这个原理，所以ThreadLocal能够实现“数据隔离”，获取当前线程的局部变量值，不受其他线程影响。

        那么言归正传，为什么要用弱引用?不用又会如何？我们通过一段代码分析一下：

public void function01()
{
    //新建了一个ThreadLocal对象，t1 是强引用指向这个对象；
    ThreadLocal tl = new ThreadLocal<Integer>();    
    //调用set()方法后新建一个Entry，通过源码可知Entry对象里的k是弱引用指向这个对象
    tl.set(2021);                                   
    tl.get();                                       
}

        上图中K,V就是ThreadLocalMap中的一条Entry。

        当func1方法执行完毕后，栈帧销毁强引用 tl 也就没有了。但此时线程的ThreadLocalMap里某个entry的key引用还指向这个对象。
        若这个key引用是强引用，就会导致key指向的ThreadLocal对象及v指向的对象不能被gc回收，造成内存泄漏；
        若这个key引用是弱引用，就大概率会减少内存泄漏的问题(还有一个key为null的雷)。使用弱引用，就可以使ThreadLocal对象在方法执行完毕后顺利被回收且Entry的key引用指向为null。

        此后我们调用get,set或remove方法时，就会尝试删除key为null的entry，可以释放value对象所占用的内存。

        但是弱引用也不是万事大吉了。当我们为threadLocal变量赋值，实际上就是当前的Entry(threadLocal实例为key，值为value)往这个threadLocalMap中存放。Entry中的key是弱引用，当threadLocal外部强引用被置为null(tl=null),那么系统 GC 的时候，根据可达性分析，这个threadLocal实例就没有任何一条链路能够引用到它，这个ThreadLocal势必会被回收，这样一来，ThreadLocalMap中就会出现key为null的Entry，就没有办法访问这些key为null的Entry的value，如果当前线程再迟迟不结束的话(使用线程池的情况下)，这些key为null的Entry的value就会一直存在一条强引用链：Thread Ref -> Thread -> ThreaLocalMap -> Entry -> value永远无法回收，造成内存泄漏。

        当然，如果当前thread运行结束，threadLocal，threadLocalMap，Entry没有引用链可达，在垃圾回收的时候都会被系统进行回收。

        但在实际使用中我们有时候会用线程池去维护我们的线程，比如在Executors. newFixedThreadPool()时创建线程的时候，为了复用线程是不会结束的，所以threadLocal内存泄漏就值得我们小心。

最佳实践

1）尽量一定要先初始化：ThreadLocal.withInitial(() -> 初始化)

2）建议：把ThreadLocal修饰为static

3）强制：用完记得手动remove()

ThreadLocal能实现线程的数据隔离，不在于它自己本身，而在于Thread的ThreadLocalMap
所以，ThreadLocal可以只初始化一次，只分配一块空间就足够了，没必要作为成员变量多次被初始化。

结论

ThreadLocal 并不解决线程间共享数据的问题。
ThreadLocal 适用于变量在线程间隔离且在方法间共享的场景。
ThreadLocal 通过隐式的在不同线程内创建独立实例副本避免了实例线程安全的问题。
每个线程持有一个只属于自己的专属Map并维护了ThreadLocal对象与具体实例的映射，该Map由于只被持有它的线程访问，故不存在线程安全以及锁的问题。
ThreadLocalMap的Entry对ThreadLocal的引用为弱引用，避免了ThreadLocal对象无法被回收的问题。