ThreadLocal原理简介

ThreadLocal的原理

每个Thread内部维护着一个ThreadLocalMap，它是TheadLocal的一个内部类，如下为部分代码：

static class ThreadLocalMap {

        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
    //省略其它代码
}

由代码可知，这个Map的Key其实是一个弱引用，其实就是ThreadLocal本身，Value才真正存的线程变量Object。也就是说ThreadLocal本身并不真正存储线程的变量值，它只是一个工具，用来维护Thread内部的Map，帮助存和取，如下图所示：

由此可见，ThreadLocal的核心机制为：

1. 每个Thread线程内部都有一个Map。
2. Map里面存储线程本地对象（key）和线程的变量副本（value）。
3. 但是，Thread内部的Map是由ThreadLocal维护的，由ThreadLocal负责向Map获取和设置线程的变量值。

由代码已经知道，Map的Key是一个弱引用，如下图所示：

注意上图中的虚线，它代表一个弱引用类型，而弱引用只能存活到下次GC前。

ThreadLocal为什么会内存泄漏

ThreadLocal在ThreadLocalMap中是以一个弱引用身份被Entry中的Key引用的，因此如果ThreadLocal没有外部强引用来引用它，那么ThreadLocal会在下次JVM垃圾收集时被回收。这个时候就会出现Entry中Key已经被回收，出现一个null Key的情况，外部读取ThreadLocalMap中的元素是无法通过null Key来找到Value的。因此如果当前线程的生命周期很长，那么其内部的ThreadLocalMap对象也一直生存下来，这些null key就存在一条强引用链的关系一直存在：Thread --> ThreadLocalMap-->Entry-->Value，这条强引用链会导致Entry不会回收，Value也不会回收，但Entry中的Key却已经被回收的情况，造成内存泄漏。

但是JVM团队已经考虑到这样的情况，并做了一些措施来保证ThreadLocal尽量不会内存泄漏：在ThreadLocal的get()、set()、remove()方法调用的时候会清除掉线程ThreadLocalMap中所有Entry中Key为null的Value，并将整个Entry设置为null，利于下次内存回收。

为什么使用弱引用

官方文档的说法：

To help deal with very large and long-lived usages, the hash table entries use WeakReferences for keys.
为了处理非常大和生命周期非常长的线程，哈希表使用弱引用作为 key。

如何避免内存泄漏

每次使用完ThreadLocal，都调用它的remove()方法，清除数据。

 

文章主要内容来自：https://www.jianshu.com/p/a1cd61fa22da