ThreadLocal 内存泄漏问题

本文说说ThreadLocal由于使用不当造成的内存泄漏问题

Thead和ThreadLocal的内存状况如下图，不了解的同学参考：图解ThreadLocal核心原理

如果线程太多，每个线程的value值都很大，是不是会造成内存溢出，写个程序模拟一下OOM的场景

1、程序模拟一下OOM

public class ThreadLocalTest {public static void main(String[] args) throws InterruptedException {testOOM();}private static void testOOM() throws InterruptedException {int nThreads = 101;ExecutorService executorService = Executors.newFixedThreadPool(nThreads);for (int i = 0; i < nThreads; i++) {executorService.execute(new Runnable() {@Overridepublic void run() {System.out.println(Thread.currentThread().getName());threadLocal.set(new BigObject());// 业务处理}});}Thread.sleep(10 * 1000);}static class BigObject {byte[] _1M_bytes = new byte[1024 * 1024];}
}
//

使用 -Xms100m -Xmx100m 参数运行程序

java.lang.OutOfMemoryError: Java heap spaceat com.yh.stu.rocketmq.ThreadLocalTest$BigObject.<init>(ThreadLocalTest.java:67)at com.yh.stu.rocketmq.ThreadLocalTest$1.run(ThreadLocalTest.java:28)

2、ThreadLocal OOM原因分析

为什么会OOM？因为线程池有101个线程，每一个线程都会有个Map，每个Map中存放了1M大小的对象，当循环执行到100左右的时候，堆内存就放满了，再向线程中放入Map时就OOM了

解决办法：在// 业务处理的后面加入threadLocal.remove()，就不会OOM了，垃圾回收器会回收线程中的new BigObject()对象的空间

3、Thread中的 Map

从前文中我们感觉到，用户的数据是存在Thread的Map中的，ThreadLocal对象只是作为Map的key的方式存在着

key是虚引用，内存不够垃圾回收时会回收，但value却不会，看下图

执行 threadLocal.remove() 方法，将引用置为null，value对应的对象BigObject没有引用指向它，可以被回收

附录-WeakReference使用

运行参数 -Xms100m -Xmx100m

 public static void testWeakReference() throws InterruptedException {while(true){new Thread(){@Overridepublic void run() {WeakReference<BigObject> wref= new WeakReference<>(new BigObject());try {Thread.sleep(Long.MAX_VALUE);} catch (InterruptedException e) {e.printStackTrace();}}}.start();Thread.sleep(100);}
}

这个程序不会造成内存溢出，因为弱引用在内存不足时会被垃圾回收器回收