业务开发中,一般都会使用ThreadLocal保存一些上下文信息,但是在线程池中执行对应逻辑时,由于是不同线程所以无法获取之前线程的上下文信息。
线程池的线程上下文传递,实现方案就是在提交任务时记录当前线程上下文信息,在线程池中线程执行用户任务前将之前保存的上下文塞到当前线程的上下文中,在执行用户任务之后移除该上下文即可。简单来说就是,外部线程提交任务时要记录上下文信息,内部线程执行任务时获取之前记录的上下文信息设置到当前线程上下文中。
实现线程上下文传递的2种方式:
一种是在用户任务中直接进行手动获取/设置上下文逻辑。
另一种是实现一个自定义的线程池,在提交任务时对任务进行包装并保存上下文信息,然后任务执行前设置上下文信息。
两种实现方式的代码如下:
private static ThreadLocal CONTEXT = new ThreadLocal<>();
private static ExecutorService executor = new ThreadPoolExecutor(1, 1,
60, TimeUnit.SECONDS, new ArrayBlockingQueue<>(512));
private static ExecutorService executorWrap = new ThreadPoolExecutorWrap(1, 1,
60, TimeUnit.SECONDS, new ArrayBlockingQueue<>(512));
public static void main(String[] args){
CONTEXT.set("main context");
//