跳到主要内容

ThreadLocal使用中致命的细节问题

我们在之前已经讲了很多关于 ThreadLocal 的设计与多线程的解决方案,而本章节介绍一个使用中容易被忽略的细节,保证你看完了后直拍大腿

BaseParameterHolder

先设计一个ThreadLocal的工具

public class BaseParameterHolder {

private static final ThreadLocal<Map<String, String>> THREAD_LOCAL_MAP = new ThreadLocal<>();


public static void setParameter(String name, String value) {
Map<String, String> map = THREAD_LOCAL_MAP.get();
if (map == null) {
map = new HashMap<>(64);
}
map.put(name, value);
THREAD_LOCAL_MAP.set(map);
}

public static String getParameter(String name) {
return Optional.ofNullable(THREAD_LOCAL_MAP.get()).map(map -> map.get(name)).orElse(null);
}

public static void removeParameter(String name) {
Map<String, String> map = THREAD_LOCAL_MAP.get();
if (map != null) {
map.remove(name);
}
}

public static ThreadLocal<Map<String, String>> getThreadLocal() {
return THREAD_LOCAL_MAP;
}

public static Map<String, String> getParameterMap() {
Map<String, String> map = THREAD_LOCAL_MAP.get();
if (map == null) {
map = new HashMap<>(64);
}
return map;
}

public static void setParameterMap(Map<String, String> map) {
THREAD_LOCAL_MAP.set(map);
}

public static void removeParameterMap(){
THREAD_LOCAL_MAP.remove();
}
}

BaseParameterHolder 中有个 ThreadLocal 类型成员属性 THREAD_LOCAL_MAP,里面放的数据是Map结构。

在操作 BaseParameterHolder 进行设值和取值的时候,其实就是操作 THREAD_LOCAL_MAP 中的Map结构

看着很简单,接下来用案例来详细解释

案例1

public class Test01 {

public static void main(String[] args) {
BaseParameterHolder.setParameter("name", "小红");
ExecutorService executors = Executors.newFixedThreadPool(10);

Map<String, String> parameterMap = BaseParameterHolder.getParameterMap();
System.out.println("主线程执行的数据获取:" + BaseParameterHolder.getParameter("name"));
executors.execute(() -> {
try {
Thread.sleep(500);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
BaseParameterHolder.setParameterMap(parameterMap);
System.out.println("线程池执行的数据获取:" + BaseParameterHolder.getParameter("name"));
});
}
}

先在主线程中,向BaseParameterHolder中存放数据,然后将数据取出,进行输出。

在线程池执行中,再把数据放入BaseParameterHolder中,然后将数据取出,进行输出。

结果

主线程执行的数据获取:小红
线程池执行的数据获取:小红

这样做肯定主线程和子线程都能取到数据

案例2

一般在操作 ThreadLocal 时,为了防止内存泄露,通常会在使用完后,手动的删除,所以在案例2中多了个删除的动作

public class Test02 {

public static void main(String[] args) {
BaseParameterHolder.setParameter("name", "小红");
ExecutorService executors = Executors.newFixedThreadPool(10);

Map<String, String> parameterMap = BaseParameterHolder.getParameterMap();
System.out.println("主线程执行的数据获取:" + BaseParameterHolder.getParameter("name"));
executors.execute(() -> {
try {
Thread.sleep(500);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
BaseParameterHolder.setParameterMap(parameterMap);
System.out.println("线程池执行的数据获取:" + BaseParameterHolder.getParameter("name"));
});
System.out.println("主线程执行删除数据");
BaseParameterHolder.removeParameter("name");
}
}

大家可以先思考一下结果是什么,然后再看结果

付费内容提示

该文档的全部内容仅对「码力全开」项目实战&技术讲解 知识星球用户开放

加入星球,一次获得完整项目资料、全栈技术知识库和长期答疑服务。

100万+字全栈技术知识库深入讲解技术核心、数据库、中间件和分布式等内容
8套热门的实战项目持续更新的企业级项目覆盖高并发、微服务、数据中台 和 AI Agent 等方向
AI 技术知识大模型面试详解覆盖 AI 模型原理、Agent、RAG、MCP、Skills、Harness 等核心知识点
文档 + 视频两种讲解形式既能系统阅读,也能跟随视频理解核心业务

完整项目实战资料

每套项目均包含从 0 到 1 讲解文档核心业务讲解视频

从基础项目到复杂业务场景,项目资料会持续更新。

8 套项目
  • 01Nexus Agent AI 智能体
  • 02Nexus Agent Pro 完全版
  • 03黑马点评Plus
  • 04大麦
  • 05大麦Pro
  • 06大麦AI
  • 07流量切换
  • 08数据中台

加入后还能获得

进入星球后,即可享受上述所有服务,保证不会再有其他隐藏费用。

从学习、面试到项目启动,都可以继续获得支持。

  • 1 对 1 解答项目和技术问题都可以提问
  • 针对性补充没有讲清楚的内容会继续补充
  • 面试与简历指导梳理回答技巧和项目亮点
  • 中间件云环境项目依赖可以直接接入使用
  • 面试后复盘被问住的问题可以继续交流
  • 远程问题解决项目启动问题可协助排查
知识星球二维码

扫码进入知识星球

  1. 打开微信,扫描左侧二维码,加入「码力全开」项目实战&技术讲解 知识星球
  2. 查看星球使用指导,获取完整项目讲解资料索引
解锁全部付费内容
🎁优惠