线程池中传递MDC

线程池中传递MDC

@Bean
public Executor asyncServiceExecutor() {
    logger.info("start asyncServiceExecutor");
    // 使用VisiableThreadPoolTaskExecutor
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
    // 配置核心线程数
    executor.setCorePoolSize(20);
    // 配置最大线程数
    executor.setMaxPoolSize(100);
    // 配置队列大小
    executor.setQueueCapacity(99999);
    // 配置线程池中的线程的名称前缀
    executor.setThreadNamePrefix("async-service-");

    // rejection-policy：当pool已经达到max size的时候，如何处理新任务
    // CALLER_RUNS：不在新线程中执行任务，而是有调用者所在的线程来执行
    executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
    // 填充装饰器
    executor.setTaskDecorator(new MdcTaskDecorator());
    // 执行初始化
    executor.initialize();
    return TtlExecutors.getTtlExecutor(executor);
}

class MdcTaskDecorator implements TaskDecorator {
    @Override
    public Runnable decorate(Runnable runnable) {
        Map cOntextMap= MDC.getCopyOfContextMap();
        return () -> {
            try {
                if (contextMap != null) {
                    MDC.setContextMap(contextMap);
                }
                runnable.run();
            } finally {
                MDC.clear();
            }
        };
    }
}