将ThreadLocal传播到从ExecutorService提取的新线程

Question

我正在使用ExecutorService和Future（在此处为示例代码）在具有超时的单独线程中运行进程（线程“生成”发生在AOP Aspect中）。

现在，主线程是一个Resteasy请求。Resteasy使用一个或多个ThreadLocal变量来存储一些上下文信息，这些信息需要在Rest方法调用中的某个时刻检索。问题是，由于Resteasy线程在新线程中运行，因此ThreadLocal变量丢失。

将Resteasy使用的任何ThreadLocal变量“传播”到新线程的最佳方法是什么？似乎Resteasy使用了多个ThreadLocal变量来跟踪上下文信息，我想“盲目”将所有信息转移到新线程中。

我已经研究了子类化，ThreadPoolExecutor并使用beforeExecute方法将当前线程传递给池，但是我找不到将ThreadLocal变量传递给池的方法。

有什么建议吗？

谢谢

Answer 1

ThreadLocal与线程关联的实例集保存在每个html" target="_blank">实例的私有成员中Thread。列举这些的唯一机会就是对问题进行一些思考Thread;这样，您可以覆盖线程字段上的访问限制。

一旦获得的集合ThreadLocal，就可以使用beforeExecute()和的afterExecute()钩子在后台线程中进行复制ThreadPoolExecutor，或者通过Runnable为任务创建一个包装器来拦截run()调用以设置一个未设置的必要ThreadLocal实例。实际上，后一种技术可能会更好，因为它会为您提供一个方便的位置，以便ThreadLocal在排队任务时存储值。

更新： 这是第二种方法的更具体的说明。与我的原始描述相反，包装器中存储的全部是调用线程，在执行任务时会对其进行查询。

static Runnable wrap(Runnable task)
{
  Thread caller = Thread.currentThread();
  return () -> {
    Iterable<ThreadLocal<?>> vars = copy(caller);
    try {
      task.run();
    }
    finally {
      for (ThreadLocal<?> var : vars)
        var.remove();
    }
  };
}

/**
 * For each {@code ThreadLocal} in the specified thread, copy the thread's 
 * value to the current thread.  
 * 
 * @param caller the calling thread
 * @return all of the {@code ThreadLocal} instances that are set on current thread
 */
private static Collection<ThreadLocal<?>> copy(Thread caller)
{
  /* Use a nasty bunch of reflection to do this. */
  throw new UnsupportedOperationException();
}