线程池场景下使用InheritableThreadLocal出现数据泄漏问题求解

1. 线程池的核心逻辑是线程复用，线程一旦创建完成后会被循环复用，不会重新初始化。而InheritableThreadLocal的上下文继承逻辑仅在子线程实例化的那一刻执行，当你提交新任务给已经存在的线程池线程运行时，不会重新从提交任务的父线程拷贝最新的TaskContext，线程里存储的还是之前运行任务留下的上下文。
2. 你当前的清理逻辑存在执行风险：如果任务运行过程中抛出异常，没有包裹在finally块中的setTaskContext(null)逻辑会被跳过，导致旧的上下文残留在复用线程中，下次调用getTaskContext()就会拿到上一次任务的数据。

ThreadLocal的底层实现是每个线程持有一个ThreadLocalMap，Map的Entry是弱引用类型，key为ThreadLocal实例，value为存储的上下文值：

• 调用set(null)仅会将当前ThreadLocal对应的Entry的value设置为null，Entry本身仍然会保留在ThreadLocalMap中。你代码中的taskContextTL是静态变量，生命周期和类一致，永远不会被垃圾回收，这个残留的Entry会一直占用内存，也就是资料里提到的内存泄漏风险。
• 调用remove()会直接将整个Entry从ThreadLocalMap中彻底删除，不会残留任何引用，不会有内存泄漏隐患，清理更彻底。

• 首先修正上下文的设置、清理逻辑：将上下文的生命周期控制在任务执行的同一个线程内，用try-finally块包裹，确保无论任务是否抛出异常，清理逻辑都能执行，并且把set(null)替换为remove()：

// 建议先在TaskContext类中新增清理方法
public static void clear() {
    taskContextTL.remove();
}

// 提交到线程池的任务按如下格式编写
Runnable yourTask = () -> {
    try {
        TaskContext.setTaskContext(new TaskContext("taskName", "user"));
        // 业务逻辑
        System.out.println(TaskContext.getTaskContext().getTaskName());
    } finally {
        TaskContext.clear();
    }
};

• 如果需要将父线程的上下文正确传递给线程池的复用线程，不要依赖InheritableThreadLocal的默认继承逻辑，用任务装饰器主动捕获、传递上下文：

public class ContextRunnable implements Runnable {
    private final Runnable target;
    private final TaskContext parentContext;

    public ContextRunnable(Runnable target) {
        this.target = target;
        // 提交任务时在父线程捕获上下文
        this.parentContext = TaskContext.getTaskContext();
    }

    @Override
    public void run() {
        try {
            TaskContext.setTaskContext(parentContext);
            target.run();
        } finally {
            TaskContext.clear();
        }
    }
}

// 提交任务时用装饰器包裹即可
executor.submit(new ContextRunnable(originalTask));

• 若上下文传递场景复杂，可使用专门解决线程池上下文传递问题的TransmittableThreadLocal组件简化开发。

内容的提问来源于stack exchange，提问作者user16076953