ThreadPoolTaskExecutor原理解析及使用实践

@Service
public class AsyncService {
    @Async("myTaskExecutor") // 指定线程池Bean名称
    public void asyncMethod() {
        // 耗时操作
    }
}

@Autowired
private TaskExecutor taskExecutor;

public void processBatch(List<Data> dataList) {
    dataList.forEach(data -> taskExecutor.execute(() -> {
        // 处理单个数据
    }));
}

@Configuration
public class ThreadPoolConfig {
    @Bean("myTaskExecutor")
    public ThreadPoolTaskExecutor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(10);
        executor.setQueueCapacity(20);
        executor.setKeepAliveSeconds(60);
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.setThreadNamePrefix("MyExecutor-");
        executor.initialize(); // 必须调用initialize()初始化
        return executor;
    }
}

@Service
public class OrderService {
    @Async("myTaskExecutor")
    public void createOrder(Order order) {
        // 模拟耗时操作
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        System.out.println("Order created: " + order.getId());
    }
}

@RestController
public class TaskController {
    @Autowired
    private TaskExecutor taskExecutor;

    @GetMapping("/run-task")
    public String runTask() {
        taskExecutor.execute(() -> {
            System.out.println("Task running in thread: " + Thread.currentThread().getName());
        });
        return "Task submitted";
    }
}

@Bean
public ThreadPoolTaskExecutor taskExecutor() {
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
    executor.setTaskDecorator(runnable -> {
        // 保存当前线程上下文
        Map<String, String> context = new HashMap<>();
        // ... 填充上下文数据
        return () -> {
            // 恢复上下文到子线程
            // ... 恢复逻辑
            runnable.run();
        };
    });
    // 其他配置...
    return executor;
}