TaskDecorator——异步多线程中传递上下文等变量

目录

TaskDecorator

定义TaskDecorator实例

线程池使用TaskDecorator

开发中很多数据如oauth2的认证信息，日志TracerId都是在请求线程中的，如果内部使用多线程处理就存在获取不到认证信息或TraceId的问题。这时候就需要处理子线程与主线程间数据传递的问题。

TaskDecorator

这个问题需要使用线程的ThreadLocal和TaskDecorator来处理。官方文档中描述意思是TaskDecorator是一个执行回调方法的装饰器，主要应用于传递上下文，或者提供任务的监控/统计信息。

实现方式就是定义一个TaskDecorator，在线程池中设置使用这个TaskDecorator。

注意线程池中有的线程是一直存在一直被复用的，所以线程执行完成后需要在TaskDecorator的finally方法中移除传递的上下文对象，否则就存在内存泄漏的问题。

定义TaskDecorator实例

继承TaskDecorator接口创建ContextCopyingDecorator 实现类，重写decorate方法，设置需要传递的上下文和变量值。注意finally代码块。

1.  
   public class ContextCopyingDecorator implements TaskDecorator {
2.  
   @Override
3.  
   public Runnable decorate(Runnable runnable) {
4.  
   try {
5.  
   RequestAttributes context = RequestContextHolder.currentRequestAttributes(); //1
6.  
   Map<String,String> previous = MDC.getCopyOfContextMap(); //2
7.  
   SecurityContext securityContext = SecurityContextHolder.getContext(); //3
8.  
   return () -> {
9.  
   try {
10.  
    RequestContextHolder.setRequestAttributes(context); //1
11.  
    MDC.setContextMap(previous); //2
12.  
    SecurityContextHolder.setContext(securityContext); //3
13.  
    runnable.run();
14.  
    } finally {
15.  
    RequestContextHolder.resetRequestAttributes(); // 1
16.  
    MDC.clear(); // 2
17.  
    SecurityContextHolder.clearContext(); // 3
18.  
    }
19.  
    };
20.  
    } catch (IllegalStateException e) {
21.  
    return runnable;
22.  
    }
23.  
    }
24.  
    }

线程池使用TaskDecorator

1.  
   @Bean
2.  
   public TaskExecutor taskExecutor() {
3.  
   ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
4.  
   executor.setCorePoolSize(corePoolSize);
5.  
   executor.setMaxPoolSize(maxPoolSize);
6.  
   executor.setQueueCapacity(queueCapacity);
7.  
   executor.setThreadNamePrefix("MyExecutor-");
8.  
    
9.  
   // for passing in request scope context
10.  
    executor.setTaskDecorator(new ContextCopyingDecorator());
11.  
     
12.  
    executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
13.  
    executor.setWaitForTasksToCompleteOnShutdown(true);
14.  
    executor.initialize();
15.  
    return executor;
16.  
    }