MDC进阶 给异步线程添加唯一id

1
@EnableAsync
2
@Configuration
3
public class AsyncConfiguration {
4

5
    @Bean("async")
6
    public Executor taskExecutor() {
7
        // 线程池的具体配置根据实际情况去配置，不要直接复制粘贴就不管了
8
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
9
        //设置核心线程数量
10
        executor.setCorePoolSize(8);
11
        //设置最大线程数量
12
        executor.setMaxPoolSize(4);
13
        //设置队列最大长度
14
        executor.setQueueCapacity(16);
15
        //设置线程空闲时间
16
        executor.setKeepAliveSeconds(60);
17
        //设置线程前缀
18
        executor.setThreadNamePrefix("async-");
19
        //设置拒绝策略
20
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
21
        //设置线程装饰器
22
        executor.setTaskDecorator(runnable -> ThreadMdcUtils.wrapAsync(runnable, MDC.getCopyOfContextMap()));
23
        return executor;
24
    }
25
}
26

27
class ThreadMdcUtils {
28
    public static Runnable wrapAsync(Runnable task, Map<String, String> context) {
29
        return () -> {
30
            if (context == null) {
31
                MDC.clear();
32
            } else {
33
                MDC.setContextMap(context);
34
            }
35
            if (MDC.get("requestId") == null) {
36
                MDC.put("requestId", UUID.randomUUID().toString());
37
            }
38
            try {
39
                task.run();
40
            } finally {
41
                MDC.clear();
42
            }
43
        };
44
    }
45
}