Java异步线程池中处理logback MDC

最新推荐文章于 2024-05-16 18:21:41 发布

weixin_33739646

最新推荐文章于 2024-05-16 18:21:41 发布

阅读量2.3k

点赞数 1

文章标签： java python

原文链接：https://my.oschina.net/eonezhang/blog/3045104

版权

2019独角兽企业重金招聘Python工程师标准>>>

https://moelholm.com/2017/07/24/spring-4-3-using-a-taskdecorator-to-copy-mdc-data-to-async-threads/

@EnableAsync(proxyTargetClass = true)
@SpringBootApplication
public class Application extends AsyncConfigurerSupport {
 
  @Override
  public Executor getAsyncExecutor() {
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
    executor.setTaskDecorator(new MdcTaskDecorator()); //重点
    executor.initialize();
    return executor;
  }
 
  public static void main(String[] args) {
    SpringApplication.run(Application.class, args);
  }
}

class MdcTaskDecorator implements TaskDecorator {
 
  @Override
  public Runnable decorate(Runnable runnable) {
    // Right now: Web thread context !
    // (Grab the current thread MDC data)
    Map<String, String> contextMap = MDC.getCopyOfContextMap();
    return () -> {
      try {
        // Right now: @Async thread context !
        // (Restore the Web thread context's MDC data)
        MDC.setContextMap(contextMap);
        runnable.run();
      } finally {
        MDC.clear();
      }
    };
  }
}

转载于:https://my.oschina.net/eonezhang/blog/3045104