Java并发编程Bug：ThreadLocal已用完但未清除

最新推荐文章于 2024-10-14 20:00:00 发布

原创最新推荐文章于 2024-10-14 20:00:00 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #jvm #servlet

本文探讨了Java中的ThreadLocal特性，它为每个线程提供变量副本以实现线程安全，但未正确清理可能导致内存泄漏。在高并发环境中，线程局部变量ThreadLocal使用后未调用remove()会留下内存隐患。解决方案是在finally块中确保执行ThreadLocal.remove()以释放资源。

在Java中，有许多技术可以确保线程安全。你可以使用synchronized和Lock等关键字来锁定代码块。

但它们有一个共同的特点，那就是锁定会对代码的性能产生一定的损失。

其实，JDK中还提供了另一种思路，即：以空间换时间。

没错，使用ThreadLocal类就是这种思想的具体体现。

ThreadLocal为每个使用该变量的线程提供了一个独立的副本，这样每个线程就可以独立地改变自己的副本而不影响其他线程的相应副本。

ThreadLocal的用法大致是这样的：

首先创建一个CurrentUser类，里面包含了ThreadLocal的逻辑。

<b>public</b> <b>class</b> CurrentUser {
    <b>private</b> <b>static</b> <b>final</b> ThreadLocal<UserInfo> THREA_LOCAL = <b>new</b> ThreadLocal();
    
    <b>public</b> <b>static</b> <b>void</b> set(UserInfo userInfo) {
        THREA_LOCAL.set(userInfo);
    }
    
    <b>public</b> <b>static</b> UserInfo get() {
       THREA_LOCAL.get();
    }
    
    <b>public</b> <b>static</b> <b>void</b> remove() {
       THREA_LOCAL.remove();
    }
}

在业务代码中调用CurrentUser类。

<b>public</b> <b>void</b> doSamething(UserDto userDto) {
   UserInfo userInfo = convert(userDto);
   CurrentUser.set(userInfo);
   ...

   <font><i>// Business code</i></font><font>
   UserInfo userInfo = CurrentUser.get();
   ...
}
</font>

在业务代码的第一行，将userInfo对象设置为CurrentUser，这样在业务代码中，userInfo就可以通过获取刚刚设置的对象CurrentUser.get()。

尤其是在业务代码调用层次比较深的时候，这种用法非常有用，可以减少很多不必要的参数传递。

但是在高并发的场景下，这段代码就有问题了。数据只存储在中ThreadLocal，数据用完后不及时清理。

即使ThreadLocal使用弱引用WeakReference，也可能存在内存泄漏问题，因为在 entry 对象中只有 key 设置为弱引用，但value没有。

那么，如何解决这个问题呢？

解决方案非常简单，只需在finally块中调用remove方法来清理无用的数据。

<b>public</b> <b>void</b> doSamething(UserDto userDto) {
   UserInfo userInfo = convert(userDto);
   
   <b>try</b>{
     CurrentUser.set(userInfo);
     ...
     
     <font><i>// Business code</i></font><font>
     UserInfo userInfo = CurrentUser.get();
     ...
   } <b>finally</b> {
      CurrentUser.remove();
   }
}
</font>