Spring/Spring Boot下如何动态配置计划任务

Spring/Spring Boot下如何动态配置计划任务

在Spring/Spring Boot下实现计划任务是很简单的,我们只需通过@EnableScheduling 开启计划任务的支持,然后通过@Scheduled 注解来制定计划任务,这样的实现解决了我们对计划任务的绝大部分需求。

但在很多时候,通过上述方式实现的计划任务是在代码里写死的,我们需要修改计划任务只能通过修改代码的方式实现。很多时候,我们需要从外部来设置计划任务执行的时间和方式。所以在本文中,我们着重讲解一下如何动态地配置计划任务。

和Spring对异步任务的支持一样,通过@EnableAsync 注解开启异步的支持,然后通过@Async 注解来指定异步的方法,而真正的异步任务的执行者是TaskExecutor接口 ,它的实现是ThreadPoolTaskExecutor。(若感兴趣可参考《Spring Boot 实战21 - Spring Boot的多线程异步任务》,此处只为比较加深记忆)

同样,Spring对计划任务的支持也是通过@EnableScheduling 开启计划任务的支持,然后通过@Scheduled 注解来制定计划任务,而真正的任务执行者是TaskScheduler 接口,它的实现是ThreadPoolTaskScheduler 。(若对于固定配置的计划任务感兴趣可参考《Spring Boot 实战22 - Spring Boot的计划任务》)

TaskScheduler的接口有三类主要方法:

public interface TaskScheduler {

  //指定时间启动计划任务
   ScheduledFuture<?> schedule(Runnable task, Trigger trigger);
  ScheduledFuture<?> schedule(Runnable task, Date startTime);

  //指定频率启动计划任务
   ScheduledFuture<?> scheduleAtFixedRate(Runnable task, Date startTime, long period);
   ScheduledFuture<?> scheduleAtFixedRate(Runnable task, long period);

  //指定延迟启动计划任务
   ScheduledFuture<?> scheduleWithFixedDelay(Runnable task, Date startTime, long delay);
   ScheduledFuture<?> scheduleWithFixedDelay(Runnable task, long delay);
}

方法的返回值都是ScheduledFuture 接口,可以使用它来取消任务或者检查任务是否完成。而实际要执行的任务是Runnable的实现类。

那我们可以通过TaskScheduler来实现动态的计划任务的实现。Spring Boot为我们提供了 TaskSchedulingAutoConfiguration自动配置来自动注册TaskScheduler 的Bean。所以我们无需配置可直接注入TaskScheduler的Bean即可使用。

image.png

Spring Boo同样提供了TaskSchedulingPropertiesTaskScheduler 进行配置,我们可以通过:spring.task.scheduling的属性进行配置。

  • 新建一个演示项目

image.png

  • 构建需要定时启动的任务
public class RunnableTask implements Runnable{
    private String taskName;
    private WorkingService workingService;

    public RunnableTask(String taskName, WorkingService workingService) {
        this.taskName = taskName;
        this.workingService = workingService;
    }

    @Override
    public void run() {
        workingService.work( taskName );
    }
}

这里为了还原实际,一般我们会在任务中调用Spring容器中的其他类进行操作,在此我们引入了一个WorkingService来真正执行任务。

  • 简单的WorkingService
@Service
@Slf4j
public class WorkingService {

    public void work(String taskName){
        log.info("当前任务:" + taskName +" 在线程" + Thread.currentThread().getName() + "上运行");
    }
}
  • 使用TaskScheduler 的Bean启动定时任务
@Service
public class SchedulingService {
    private final TaskScheduler taskScheduler; // 可直接注入TaskScheduler的Bean
    private final WorkingService workingService; 

    //构造器注入,免@Autowired
    public SchedulingService(TaskScheduler taskScheduler, WorkingService workingService) { 
        this.taskScheduler = taskScheduler;
        this.workingService = workingService;
    }

    public void schedule(String taskName, String cronExpression){
          //RunnableTask不是一个Bean,不能直接注入workingService,在它的构造器直接传入
        RunnableTask task = new RunnableTask(taskName,workingService);
        CronTrigger cronTrigger = new CronTrigger(cronExpression);
       //api接受实际执行的任务,和定时启动的cron表达式
        taskScheduler.schedule(task,cronTrigger);
    }

    public void scheduleAtFixedRate(String taskName, long period){
        RunnableTask task = new RunnableTask(taskName,workingService);
       //api接受实际执行的任务,和间隔时间
        taskScheduler.scheduleAtFixedRate(task, period);
    }

    public void scheduleWithFixedDelay(String taskName, long delay){
        RunnableTask task = new RunnableTask(taskName,workingService);
       //api接受实际执行的任务,和延迟时间
        taskScheduler.scheduleWithFixedDelay(task, delay);
    }
}
  • 在启动类设置动态配置
@SpringBootApplication
@EnableScheduling
public class DynamicTaskSchedulerApplication {

   public static void main(String[] args) {
      SpringApplication.run(DynamicTaskSchedulerApplication.class, args);
   }

   @Bean
   CommandLineRunner commandLineRunner(SchedulingService schedulingService, WorkingService workingService){
      return args -> {
         schedulingService.schedule("指定时间任务", "50 * * * * THU");
         schedulingService.scheduleAtFixedRate("指定频率任务", 1000);
         schedulingService.scheduleWithFixedDelay("指定延迟任务", 1000);
      } ;
   }
}
  • application.properties 的配置
spring.task.scheduling.pool.size=3
spring.task.scheduling.thread-name-prefix=wisely-
  • 启动程序验证

image.png

头条原文地址:https://www.toutiao.com/article/7158705923182182927/

感谢对我的书《从企业级开发到云原生微服务:Spring Boot实战》的支持。