Java Code Examples for org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler#setAwaitTerminationSeconds()

@Bean("taskExecutor")
public Executor taskExecutor() {
    ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
    //最大线程数30
    executor.setPoolSize(30);
    //线程池名的前缀
    executor.setThreadNamePrefix("taskExecutor-");
    //设置线程池关闭的时候等待所有任务都完成再继续销毁其他的Bean
    executor.setWaitForTasksToCompleteOnShutdown(true);
    //设置线程池中任务的等待时间
    executor.setAwaitTerminationSeconds(60);
    //当线程池没有处理能力的时候，该策略会直接在 execute 方法的调用线程中运行被拒绝的任务；如果执行程序已关闭，则会丢弃该任务
    executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

    return executor;
}
 

/**
 * thread pool for scheduler parallel task (not async):
 * pull block, trans monitor, statistic trans, delete info, reset groupList
 * @return ThreadPoolTaskScheduler
 */
@Bean(destroyMethod = "shutdown")
public ThreadPoolTaskScheduler mgrTaskScheduler() {
    log.info("start mgrTaskScheduler init..");
    ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
    scheduler.setPoolSize(schedulerProperties.getPoolSize());
    scheduler.setThreadNamePrefix(schedulerProperties.getThreadNamePrefix());
    scheduler.setAwaitTerminationSeconds(schedulerProperties.getAwaitTerminationSeconds());
    scheduler.setWaitForTasksToCompleteOnShutdown(
            schedulerProperties.getWaitForTasksToCompleteOnShutdown());
    return scheduler;
}
 

public static ThreadPoolTaskScheduler createTaskScheduler(
    int poolSize, String threadPrefix, int terminationTime) {
  ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
  scheduler.setBeanName(threadPrefix.substring(0, threadPrefix.length() - 1));
  scheduler.setPoolSize(poolSize);
  scheduler.setThreadNamePrefix(threadPrefix);
  scheduler.setAwaitTerminationSeconds(terminationTime);
  scheduler.setRemoveOnCancelPolicy(true);
  scheduler.afterPropertiesSet();
  return scheduler;
}
 

@Bean(destroyMethod = "shutdown")
public ThreadPoolTaskScheduler taskScheduler() {
    ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
    scheduler.setPoolSize(systemSchedulePoolsize);
    scheduler.setThreadNamePrefix("dispatch-");
    scheduler.setAwaitTerminationSeconds(600);
    scheduler.setErrorHandler(throwable -> logger.error("调度任务发生异常", throwable));
    scheduler.setWaitForTasksToCompleteOnShutdown(true);
    return scheduler;
}
 

@Bean(destroyMethod = "shutdown")
public ThreadPoolTaskScheduler taskScheduler() {
    ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
    // 设置默认线程名称
    executor.setThreadNamePrefix(env.getProperty("spring.application.name") + "-task-");
    // 设置最大线程数
    executor.setPoolSize(new Double(Math.floor(Runtime.getRuntime().availableProcessors() / (1 - 0.9))).intValue());
    // 设置线程活跃时间（秒）
    executor.setAwaitTerminationSeconds(60);
    // 设置拒绝策略
    executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
    // 等待所有任务结束后再关闭线程池
    executor.setWaitForTasksToCompleteOnShutdown(true);
    return executor;
}
 

@Bean(destroyMethod = "shutdown")
public ThreadPoolTaskScheduler taskScheduler() {
    ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
    scheduler.setPoolSize(10);
    scheduler.setThreadNamePrefix("task-"); //设置线程名开头
    scheduler.setAwaitTerminationSeconds(120);
    scheduler.setWaitForTasksToCompleteOnShutdown(true);
    return scheduler;
}
 

@Bean(SCHEDULER)
public TaskScheduler taskScheduler() {
    ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
    scheduler.setPoolSize(commonProperties.getExecution().getSchedulerPoolSize());
    scheduler.setWaitForTasksToCompleteOnShutdown(true);
    scheduler.setAwaitTerminationSeconds(30);
    scheduler.setThreadNamePrefix(SCHEDULER);
    return scheduler;
}
 

@Bean
public TaskScheduler sqlTaskScheduler() {
	ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
	taskScheduler.setWaitForTasksToCompleteOnShutdown(true);
	taskScheduler.setAwaitTerminationSeconds(properties.getBatchTimeout());
	return taskScheduler;
}