ThreaLocal的内存泄露问题解析

简介

本文分析了ThreaLocal为什么会存在内存泄露，以及解决该问题的方法。

问题分析

ThreadLoacl的定义方法

public static ThreadLocal<User> LOGIN_RECORD= new ThreadLocal<>();

ThreadLoacl的设置值方法：

    public void set(T value) {
    	// 获得当前线程对象
        Thread t = Thread.currentThread();
        // 获得当前线程中的ThreadLoacalMap对象
        ThreadLocalMap map = getMap(t);
        // 如果ThreadLocalMap已经定义好了直接设置，否则初始化
        if (map != null) {
            map.set(this, value);
        } else {
            createMap(t, value);
        }
    }

ThreadLocalMap的实现是使用弱引用实现的！
注意：以下代码中只有Entry的key是弱引用

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

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

我们可以发现，ThreadLocal的值是存放在线程中的一个map对象里面的，key为ThreadLocal对象，value为具体的值。

这里使用一张图来解释ThreadLocal，上图中，左侧是线程的栈，栈中存在两个强引用，分别是对于ThreadLocal的强引用，以及对于当前线程的强引用。右边是堆中ThreadLocal的存储结构，一个线程有一个ThreadLocalMap，在ThreadLocalMap的entry中的key对于ThreadLocal有一个弱引用。

当线程中ThreadLocal对象引用为null之后，在栈中对于ThreadLocal对象的强引用断了，只剩ThreadLocalMap中key值对于对象的一个引用。我们知道，当弱引用只有一个引用的时候会被垃圾回收，这就造成了ThreadLocal对象被垃圾回收了，这个时候ThreadLocalMap中的key变成了null，value仍然是强引用所以不会被回收，所以造成了内存泄露。

要解决这个问题就直接调用threadLocal中的remove方法，直接干掉k-v对消除强引用。

 public void remove() {
     ThreadLocalMap m = getMap(Thread.currentThread());
     if (m != null) {
         m.remove(this);
     }
 }

特别是在使用线程池的场景中，ThreadLocal会一直存在，更要小心内存泄漏