21.导致JVM内存泄露的ThreadLocal详解

为什么要有 ThreadLocal

在原生 JDBC 代码中，为了使事务使用同一个连接。

①把连接对象使用参数传递的方式

Spring 是如何解 决这个问题的？

②使用一个 ThreadLocal 绑定连接到线程

ThreadLocal 为每个线程分配变量副本（跨方法的参数传递），这样就隔离了多个线程对数据的数据共享。

应用

在微服务领域，链路跟踪中的 traceId 传递也是利用了 ThreadLocal。

ThreadLocal 的使用

• void set(Object value)
  • 设置当前线程的线程局部变量的值。
• public Object get()
  • 该方法返回当前线程所对应的线程局部变量。
• public void remove()
  • 将当前线程局部变量的值删除，目的是为了减少内存的占用，可以加快内存回收的速度。
• protected Object initialValue()
  • 这个方法是一个延迟调用方法，在线程第 1 次调用 get() 或 set(Object)时才执行，并且仅执行 1 次。ThreadLocal 中的缺省实现直接返回一 个 null。

//基本使用案例
public class ThreadLocalTest {

    public static ThreadLocal<String> local = new ThreadLocal<>();
    public static void main(String[] args) {

        for (int i = 0; i < 5; i++) {
            new MyThread(i).start();
        }
    }

    static class MyThread extends Thread {
        private int id;

        public MyThread() {
        }

        public MyThread(int id) {
            this.id = id;
        }
        @Override
        public void run() {
            local.set("唯一id：" + id);
            String name = this.getName();
            System.out.println("Thread[" + name + "] " + local.get() + " is running");
        }
    }
}

实现解析

如何实现 ThreadLocal，既然说让每个线程都拥有自己变量的副本，最容易的方式就是用一个 Map 将线程的副本存放起来，Map 里 key 就是每个线程的唯 一性标识，比如线程 ID，value 就是副本值。

public class MyThreadLocal<T> {

    private Map<Thread,T> map = new HashMap<>();

    public synchronized T get() {
        return map.get(Thread.currentThread());
    }

    public synchronized void set(T value) {
        map.put(Thread.currentThread(),value);
    }
}

在《并发编程实战》中为我们做过性能测试

ThreadLocal 的性能远超类似 synchronize 的锁实现 ReentrantLock， 比AtomicInteger 也要快很多。

具体实现

//线程中的一个成员变量
/* ThreadLocal values pertaining to this thread. This map is maintained
     * by the ThreadLocal class. */
ThreadLocal.ThreadLocalMap threadLocals = null;

//ThreadLocalMap中包含了一个Entry[]
static class Entry extends WeakReference<ThreadLocal<?>> {
    /** The value associated with this ThreadLocal. */
    Object value;

    Entry(ThreadLocal<?> k, Object v) {
        super(k);
        value = v;
    }
}

Hash 冲突的解决

什么是 Hash，就是把任意长度的输入（又叫做预映射， pre-image），通过 散列算法，变换成固定长度的输出，该输出就是散列值，输入的微小变化会导致 输出的巨大变化。

开放定址法

基本思想是，出现冲突后按照一定算法查找一个空位置存放，根据算法的不 同又可以分为线性探测再散列、二次探测再散列、伪随机探测再散列。 线性探测再散列即依次向后查找，二次探测再散列，即依次向前后查找，增 量为 1、2、3 的二次方，伪随机，顾名思义就是随机产生一个增量位移。

//ThreadLocal 里用的则是线性探测再散列
/**
* Increment i modulo len.
*/
private static int nextIndex(int i, int len) {
    return ((i + 1 < len) ? i + 1 : 0);
}

链地址法

这种方法的基本思想是将所有哈希地址为 i 的元素构成一个称为同义词链的 单链表，并将单链表的头指针存在哈希表的第 i 个单元中，因而查找、插入和删 除主要在同义词链中进行。链地址法适用于经常进行插入和删除的情况。Java里的 HashMap 用的就是链地址法，为了避免 hash 洪水攻击，1.8 版本开始还引 入了红黑树。

再哈希法

这种方法是同时构造多个不同的哈希函数：Hi=RH1（key） i=1，2，…，k 当哈希地址 Hi=RH1（key）发生冲突时，再计算 Hi=RH2（key）……，直到冲突 不再产生。这种方法不易产生聚集，但增加了计算时间。

建立公共溢出区

这种方法的基本思想是：将哈希表分为基本表和溢出表两部分，凡是和基本 表发生冲突的元素，一律填入溢出表。

引发的内存泄漏分析

总结

使用线程池+ ThreadLocal时，主动使用remove()方法释放内存。

错误使用ThreadLocal导致线程不安全

import java.util.Random;

public class ThreadLocalErrTest implements Runnable {


    public static Number number = new Number(0);
//1.所有的线程持有的对象引用所指向的同一个对象实例,会导致随机数一致
    public static ThreadLocal<Number> value = new ThreadLocal<>();
    //2.让每个线程中的ThreadLocal都应该持有一个新的Number对象。
//    public static ThreadLocal<Number> value = ThreadLocal.withInitial(() -> new Number(0));

    public static void main(String[] args) {
        for (int i = 0; i < 5; i++) {
            new Thread(new ThreadLocalErrTest()).start();
        }
    }
    public void run() {
        Random r = new Random();
        //配合2使用
//        Number number = value.get();
        //每个线程计数加随机数
        number.setNum(number.getNum()+r.nextInt(100));
        //将其存储到ThreadLocal中
        value.set(number);
        try {
            Thread.sleep(20);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        //输出num值
        System.out.println(Thread.currentThread().getName()+"="+value.get().getNum());
    }

    private static class Number {
        public Number(int num) {
            this.num = num;
        }

        private int num;

        public int getNum() {
            return num;
        }

        public void setNum(int num) {
            this.num = num;
        }

        @Override
        public String toString() {
            return "Number [num=" + num + "]";
        }
    }

}

public void setNum(int num) {
this.num = num;
}

    @Override
    public String toString() {
        return "Number [num=" + num + "]";
    }
}

}