ThreadPoolTaskExecutor怎么设置任务超时时间

ThreadPoolTaskExecutor是Spring框架中的一个线程池实现类，可以通过设置TaskExecutor的属性来设置任务超时时间。具体方法如下：

ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(2); // 核心线程数
executor.setMaxPoolSize(3); // 最大线程数
executor.setKeepAliveSeconds(30); // 线程空闲时间
executor.setQueueCapacity(10); // 阻塞队列大小
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); // 拒绝策略
executor.setAwaitTerminationSeconds(60); // 等待所有任务执行完毕的时间
executor.setWaitForTasksToCompleteOnShutdown(true); // 是否等待所有任务执行完毕后再关闭线程池
executor.setTaskDecorator(new TimeoutTaskDecorator(30)); // 设置任务超时时间为30秒
executor.initialize(); // 初始化线程池

// 往线程池中添加任务
for (int i = 0; i < 5; i++) {
    executor.execute(new MyTask());
}

// 扩大线程池
executor.setCorePoolSize(3);

// 每三秒往线程池中添加一个任务
for (int i = 0; i < 15; i++) {
    Thread.sleep(3000);
    executor.execute(new MyTask());
}

// 等待所有任务执行完毕
executor.shutdown();
executor.awaitTermination(60, TimeUnit.SECONDS);

// 获取线程池中存活的线程数
int activeCount = executor.getActiveCount();
System.out.println("线程池中存活的线程数：" + activeCount);

其中，TimeoutTaskDecorator是一个自定义的任务装饰器，用于设置任务超时时间。代码如下：

public class TimeoutTaskDecorator implements TaskDecorator {
    private int timeout;

    public TimeoutTaskDecorator(int timeout) {
        this.timeout = timeout;
    }

    @Override
    public Runnable decorate(Runnable runnable) {
        return new TimeoutRunnable(runnable, timeout);
    }

    private static class TimeoutRunnable implements Runnable {
        private Runnable runnable;
        private int timeout;

        public TimeoutRunnable(Runnable runnable, int timeout) {
            this.runnable = runnable;
            this.timeout = timeout;
        }

        @Override
        public void run() {
            Future<?> future = null;
            try {
                future = Executors.newSingleThreadExecutor().submit(runnable);
                future.get(timeout, TimeUnit.SECONDS);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (ExecutionException e) {
                e.printStackTrace();
            } catch (TimeoutException e) {
                e.printStackTrace();
            } finally {
                if (future != null) {
                    future.cancel(true);
                }
            }
        }
    }
}