关于 ThreadLocal 内存泄露

﹏ヽ暗。殇╰゛Y 2022-12-05 10:28 221阅读 0赞

在使用 ThreadLocal 的时候，一般我们的代码都是这样写的：

public class ThreadLocalDemo { 
        private static final ThreadLocal<Map<String, Object>> context = ThreadLocal.withInitial(HashMap::new); 
        
        public static void setUserId(Long userId) { 
            context.get().put("userId", userId);
        }
        
        public static Long getUserId() { 
            return (Long) context.get().get("userId");
        }
        
        public static void remove() { 
            context.remove();
        }
    }

然后处理业务的是一个线程池，有一个结果就是 「ThreadLocal 引用」 context（静态变量） 和 「Thread 引用」（线程池里的线程）会常驻内存。引用关系如图所示：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW9idWNob25n_size_16_color_FFFFFF_t_70]  
虽然 key 使用了弱引用，但是 key 指向的 「ThreadLocal 对象」还一直被「ThreadLocal 引用」给引用着，这样 ThreadLocal 里的 expungeStaleEntry 方法是无法进行清理的（清理 key 为 null 的元素）。

ThreadLocal 的使用场景不应该图省事，就用来传递业务参数，这样增加了程序复杂度，代码可读性降低，处理不当还会内存泄漏，适用于一些日志、监控类的场景。

推荐的做法：

try { 
          ThreadLocalDemo.setUserId(124132523);
           // 业务逻辑
       } finally { 
           // 当前线程处理完数据以后清理掉 ThreadLocal 里面的数据，
           // 处理业务的线程一般都是线程池，同一个线程可以处理多个请求，ThreadLocal 的数据是和请求相关的，及时清掉 ThreadLocal 里的数据，避免请求之间数据污染
           ThreadLocalDemo.remove();
       }

WeakReference:  
[软引用、弱引用、虚引用-他们的特点及应用场景][-]  
更多参考：  
[对ThreadLocal实现原理的一点思考][ThreadLocal]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW9idWNob25n_size_16_color_FFFFFF_t_70]: /images/20221123/931c370e879044d5b32595fc582dd86c.png
[-]: https://www.jianshu.com/p/825cca41d962?utm_campaign=maleskine&utm_content=note&utm_medium=seo_notes&utm_source=recommendation
[ThreadLocal]: https://www.jianshu.com/p/ee8c9dccc953