IT@程序员猿媛互联网科技Java

SpringBoot中并发定时任务的实现、动态定时任务的实现（看

2019-04-05  本文已影响23人  Java高级架构狮

一、在JAVA开发领域，目前可以通过以下几种方式进行定时任务

1、单机部署模式

2、分布式集群模式（不多介绍，简单提一下）

问题：

问题1的简单思考：

成熟的解决方案：

二、SpringTask实现定时任务（这里是基于springboot）

1、简单的定时任务实现

使用方式：
使用@EnableScheduling注解开启对定时任务的支持。
使用@Scheduled 注解即可，基于corn、fixedRate、fixedDelay等一些定时策略来实现定时任务。

使用缺点：

使用优点：

源码分析：

//默认使用的调度器
if(this.taskScheduler == null) {  
    this.localExecutor = Executors.newSingleThreadScheduledExecutor();
    this.taskScheduler = new ConcurrentTaskScheduler(this.localExecutor);
}
//可以看到SingleThreadScheduledExecutor指定的核心线程为1，说白了就是单线程执行
public static ScheduledExecutorService newSingleThreadScheduledExecutor() {
    return new DelegatedScheduledExecutorService
        (new ScheduledThreadPoolExecutor(1));
}
//利用了DelayedWorkQueue延时队列作为任务的存放队列，这样便可以实现任务延迟执行或者定时执行
public ScheduledThreadPoolExecutor(int corePoolSize) {
    super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS,
          new DelayedWorkQueue());
}

2、实现并发的定时任务

使用方式：
方式一：由1中我们知道之所以定时任务是阻塞执行，是配置的线程池决定的，那就好办了，换一个不就行了！直接上代码：

  @Configuration
  public class ScheduledConfig implements SchedulingConfigurer {

      @Autowired
      private TaskScheduler myThreadPoolTaskScheduler;

      @Override
      public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
          //简单粗暴的方式直接指定
          //scheduledTaskRegistrar.setScheduler(Executors.newScheduledThreadPool(5));
          //也可以自定义的线程池，方便线程的使用与维护，这里不多说了
          scheduledTaskRegistrar.setTaskScheduler(myThreadPoolTaskScheduler);
      }
  }

  @Bean(name = "myThreadPoolTaskScheduler")
  public TaskScheduler getMyThreadPoolTaskScheduler() {
      ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
      taskScheduler.setPoolSize(10);
      taskScheduler.setThreadNamePrefix("Haina-Scheduled-");
      taskScheduler.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
      //调度器shutdown被调用时等待当前被调度的任务完成
      taskScheduler.setWaitForTasksToCompleteOnShutdown(true);
      //等待时长
      taskScheduler.setAwaitTerminationSeconds(60);
      return taskScheduler;
  }

方式二：方式一的本质改变了任务调度器默认使用的线程池，接下来这种是不改变调度器的默认线程池，而是把当前任务交给一个异步线程池去执行

废话太多，直接上代码：

  @Scheduled(fixedRate = 1000*10,initialDelay = 1000*20)
  @Async("myThreadPoolTaskExecutor")
  //@Async
  public void scheduledTest02(){
      System.out.println(Thread.currentThread().getName()+"--->xxxxx--->"+Thread.currentThread().getId());
  }

  //自定义线程池
  @Bean(name = "myThreadPoolTaskExecutor")
  public TaskExecutor  getMyThreadPoolTaskExecutor() {
      ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
      taskExecutor.setCorePoolSize(20);
      taskExecutor.setMaxPoolSize(200);
      taskExecutor.setQueueCapacity(25);
      taskExecutor.setKeepAliveSeconds(200);
      taskExecutor.setThreadNamePrefix("Haina-ThreadPool-");
      // 线程池对拒绝任务（无线程可用）的处理策略，目前只支持AbortPolicy、CallerRunsPolicy；默认为后者
      taskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
      //调度器shutdown被调用时等待当前被调度的任务完成
      taskExecutor.setWaitForTasksToCompleteOnShutdown(true);
      //等待时长
      taskExecutor.setAwaitTerminationSeconds(60);
      taskExecutor.initialize();
      return taskExecutor;
  }

线程池的使用心得

三、动态定时任务的实现

问题：
使用@Scheduled注解来完成设置定时任务，但是有时候我们往往需要对周期性的时间的设置会做一些改变，或者要动态的启停一个定时任务，那么这个时候使用此注解就不太方便了，原因在于这个注解中配置的cron表达式必须是常量，那么当我们修改定时参数的时候，就需要停止服务，重新部署。

解决办法：
方式一：实现SchedulingConfigurer接口，重写configureTasks方法，重新制定Trigger,核心方法就是addTriggerTask(Runnable task, Trigger trigger) ，不过需要注意的是，此种方式修改了配置值后，需要在下一次调度结束后，才会更新调度器，并不会在修改配置值时实时更新，实时更新需要在修改配置值时额外增加相关逻辑处理。

  @Configuration
  public class ScheduledConfig implements SchedulingConfigurer {

  @Autowired
  private TaskScheduler myThreadPoolTaskScheduler;

  @Override
  public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
      //scheduledTaskRegistrar.setScheduler(Executors.newScheduledThreadPool(5));
      scheduledTaskRegistrar.setTaskScheduler(myThreadPoolTaskScheduler);
      //可以实现动态调整定时任务的执行频率
      scheduledTaskRegistrar.addTriggerTask(
              //1.添加任务内容(Runnable)
              () -> System.out.println("cccccccccccccccc--->" + Thread.currentThread().getId()),
              //2.设置执行周期(Trigger)
              triggerContext -> {
                  //2.1 从数据库动态获取执行周期
                  String cron = "0/2 * * * * ? ";
                  //2.2 合法性校验.
  //                    if (StringUtils.isEmpty(cron)) {
  //                        // Omitted Code ..
  //                    }
                      //2.3 返回执行周期(Date)
                      return new CronTrigger(cron).nextExecutionTime(triggerContext);
                  }
          );
  }
  }

方式二：使用threadPoolTaskScheduler类可实现动态添加删除功能，当然也可实现执行频率的调整

首先，我们要认识下这个调度类，它其实是对java中ScheduledThreadPoolExecutor的一个封装改进后的产物，主要改进有以下几点：

顺便说下ThreadPoolTaskExecutor相对于ThreadPoolExecutor的改进点：

扯了这么多，还是直接上代码：

  @Component
  public class DynamicTimedTask {

      private static final Logger logger = LoggerFactory.getLogger(DynamicTimedTask.class);

      //利用创建好的调度类统一管理
      //@Autowired
      //@Qualifier("myThreadPoolTaskScheduler")
      //private ThreadPoolTaskScheduler myThreadPoolTaskScheduler;


      //接受任务的返回结果
      private ScheduledFuture<?> future;

      @Autowired
      private ThreadPoolTaskScheduler threadPoolTaskScheduler;

      //实例化一个线程池任务调度类,可以使用自定义的ThreadPoolTaskScheduler
      @Bean
      public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
          ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
          return new ThreadPoolTaskScheduler();
      }


      /**
       * 启动定时任务
       * @return
       */
      public boolean startCron() {
          boolean flag = false;
          //从数据库动态获取执行周期
          String cron = "0/2 * * * * ? ";
          future = threadPoolTaskScheduler.schedule(new CheckModelFile(),cron);
          if (future!=null){
              flag = true;
              logger.info("定时check训练模型文件,任务启动成功！！！");
          }else {
              logger.info("定时check训练模型文件,任务启动失败！！！");
          }
          return flag;
      }

      /**
       * 停止定时任务
       * @return
       */
      public boolean stopCron() {
          boolean flag = false;
          if (future != null) {
              boolean cancel = future.cancel(true);
              if (cancel){
                  flag = true;
                  logger.info("定时check训练模型文件,任务停止成功！！！");
              }else {
                  logger.info("定时check训练模型文件,任务停止失败！！！");
              }
          }else {
              flag = true;
              logger.info("定时check训练模型文件，任务已经停止！！！");
          }
          return flag;
      }


      class CheckModelFile implements Runnable{

          @Override
          public void run() {
              //编写你自己的业务逻辑  
              System.out.print("模型文件检查完毕！！！")
          }
      }

  }

四、总结

到此基于springtask下的定时任务的简单使用算是差不多了，其中不免有些错误的地方，或者理解有偏颇的地方欢迎大家提出来！

上一篇下一篇

猜你喜欢

热点阅读