自定義線程池並拷貝線程上下文

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * 自定義的 TaskPoolConfig
 * 增加 TaskDecorator 屬性的配置
 *
 * @author wang
 */
@Configuration
public class TaskPoolConfig {
    /**
     *獲取當前機器CPU數量
     */
    private static final int CPU = Runtime.getRuntime().availableProcessors();
    /**
     *核心線程數（默認線程數）
     */
    private static final int CORE_POOL_SIZE = CPU;
    /**
     *最大線程數
     */
    private static final int MAX_POOL_SIZE = CPU * 2;
    /**
     *允許線程空閒時間（單位：默認爲秒）
     */
    private static final int KEEP_ALIVE_TIME = 60;
    /**
     *緩衝隊列數
     */
    private static final int QUEUE_CAPACITY = 200;
    /**
     *線程池名前綴
     */
    private static final String THREAD_NAME_PREFIX = "wflTaskExecutor-";


    @Bean("wflTaskExecutor")
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        // 核心線程數（默認線程數）
        executor.setCorePoolSize(CORE_POOL_SIZE);
        // 最大線程數
        executor.setMaxPoolSize(MAX_POOL_SIZE);
        // 緩衝隊列數
        executor.setQueueCapacity(QUEUE_CAPACITY);
        // 允許線程空閒時間 s
        executor.setKeepAliveSeconds(KEEP_ALIVE_TIME);
        // 線程池名前綴
        executor.setThreadNamePrefix(THREAD_NAME_PREFIX);
        // 線程池對拒絕任務的處理策略
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

        // 增加 TaskDecorator 屬性的配置
        executor.setTaskDecorator(new ContextDecorator());
        executor.initialize();

        return executor;
    }
}

import org.springframework.core.task.TaskDecorator;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;

/**
 * 線程上下文拷貝 TaskDecorator
 * @author wang
 */
public class ContextDecorator implements TaskDecorator {

    @Override
    public Runnable decorate(Runnable runnable) {
        RequestAttributes context = RequestContextHolder.currentRequestAttributes();
        return () -> {
            try {
                RequestContextHolder.setRequestAttributes(context);
                runnable.run();
            } finally {
                RequestContextHolder.resetRequestAttributes();
            }
        };
    }
}