spring动态控制定时任务的实现
在spring框架中，对于简单的定时任务，可以使用 @Scheduled 注解实现，在实际项目中，经常需要动态的控制定时任务，比如通过接口增加、启动、停止、删除定时任务，动态的改变定时任务的执行时间等。
我们可以通过编码的方式动态控制定时任务，具体的代码参照 示例项目 https://github.com/qihaiyan/springcamp/tree/master/spring-dynamic-scheduler
一、概述
在spring框架可以通过 CronTask 和 TaskScheduler 动态控制定时任务，实现定时任务的动态更新，比如修改定时任务的执行时间，这个是 @Scheduled 无法实现的。采用编码控制动态任务的方式，我们还可以把动态任务执行信息保存到数据库中，通过数据库里的任务配置数据来动态控制定时任务，也可以通过接口来动态控制定时任务。
二、配置定时任务
首先，同 @Scheduled 注解的方式一样，动态控制定时任务也需要使用 @EnableScheduling 注解来开启定时任务功能：
然后通过实现 SchedulingConfigurer 接口来对动态任务进行配置：
@Component
public class MyScheduler implements SchedulingConfigurer {
private ScheduledTaskRegistrar taskRegistrar;
private final ConcurrentHashMap<Long, ScheduledFuture<?>> scheduledFutures = new ConcurrentHashMap<>();
@Override
public void configureTasks(@NonNull ScheduledTaskRegistrar taskRegistrar) {
ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();
threadPoolTaskScheduler.setPoolSize(10);// Set the pool of threads
threadPoolTaskScheduler.setThreadNamePrefix("sys-scheduler");
threadPoolTaskScheduler.initialize();
this.taskRegistrar = taskRegistrar;
this.taskRegistrar.setTaskScheduler(threadPoolTaskScheduler);
}
@PreDestroy
public void destroy() {
this.taskRegistrar.destroy();
}
}
通过上面的代码，我们就启用了动态任务的基本能力，为动态任务指定了执行线程池。
三、动态更新定时任务
更新定时任务通过 CronTask 和 TaskScheduler 来实现，我们新增一个注册定时任务的方法：
public void registerTask(TaskData taskData) {
//如果配置一致，则不需要重新创建定时任务
if (scheduledFutures.containsKey(taskData.getId())
&& cronTasks.get(taskData.getId()).getExpression().equals(taskData.getExpression())) {
return;
}
//如果策略执行时间发生了变化，则取消当前策略的任务
if (scheduledFutures.containsKey(taskData.getId())) {
scheduledFutures.remove(taskData.getId()).cancel(false);
cronTasks.remove(taskData.getId());
}
CronTask task = new CronTask(taskData, taskData.getExpression());
TaskScheduler scheduler = taskRegistrar.getScheduler();
if (scheduler != null) {
ScheduledFuture<?> future = scheduler.schedule(task.getRunnable(), task.getTrigger());
if (future != null) {
scheduledFutures.put(taskData.getId(), future);
}
}
}
我们新增了一个 registerTask 方法用于注册定时任务，入参中 TaskData 是定时任务的配置数据，为了简单，我们把配置数据和执行代码放到了一起：
@Slf4j
@Data
@Entity
public class TaskData implements Runnable {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private String name;
private String expression;
@Transient
@Override
public void run() {
log.info("{} is running with expression {}", this.getName(), this.getExpression());
}
}
核心代码是创建一个 CronTask 对象，该对象包含两个参数：Runnable 方法和 cron 表达式。
CronTask 对象创建好后，通过 ScheduledTaskRegistrar 对定时任务进行注册，注册完成后，定时任务就会在cron表达式指定的时间点开始执行了。
执行的代码就是 Runnable 参数指定的方法。
四、动态停止定时任务
为了能够动态停止定时任务，我们在注册定时任务时，把注册结果放到了一个Map中：
private final ConcurrentHashMap<Long, ScheduledFuture<?>> scheduledFutures = new ConcurrentHashMap<>();
ScheduledFuture<?> future = scheduler.schedule(task.getRunnable(), task.getTrigger());
if (future != null) {
scheduledFutures.put(taskData.getId(), future);
}
新增停止定时任务的方法：
public void stop(Long id) {
if (scheduledFutures.containsKey(id)) {
scheduledFutures.remove(id).cancel(false);
}
}
该方法需要传入定时任务的id，由于我们把定时任务信息保存到了 scheduledFutures 这个Map中，所以可以根据任务id参数查找到对应的定时任务信息，然后调用对应的 cancel方法来停止定时任务。
五、通过接口控制定时任务
通过上面的步骤我们已经具备了动态控制定时任务的基本能力，下面增加接口来控制定时任务：
@EnableScheduling
@SpringBootApplication
@RestController
public class DemoApplication {
@Autowired
private MyScheduler myScheduler;
@Autowired
private TaskDataRepository taskDataRepository;
public static void main(String[] args) {
SpringApplication.run(DemoApplication.class, args);
}
@RequestMapping("/register")
public TaskData register(
String name,
@RequestParam(name = "expression", required = false, defaultValue = "0/1 * * * * ?") String expression
) {
TaskData taskData = taskDataRepository.findOneByName(name).orElse(new TaskData());
taskData.setName(name);
taskData.setExpression(expression);
taskData = taskDataRepository.save(taskData);
myScheduler.registerTask(taskData);
return taskData;
}
@RequestMapping("/stop")
public void stop(Long id) {
taskDataRepository.findById(id).ifPresent(taskData -> {
myScheduler.stop(id);
});
}
}
我们提供了 register 和 stop 两个接口，这两个接口会在改变动态任务执行数据时，先将数据保存到数据库中，对定时任务进行持久化，避免程序重启后定时任务都丢失。
程序启动后，我们首先调用 register 接口新增一个定时任务：
http://localhost:8080/register?name=test
接口调用后，在日志中可以看到定时任务开始执行了：
2024-01-07T18:02:09.003+08:00 INFO 23012 --- [ sys-scheduler5] c.s.springdynamicscheduler.TaskData : test is running with expression 0/1 * * * * ? 2024-01-07T18:02:10.005+08:00 INFO 23012 --- [ sys-scheduler3] c.s.springdynamicscheduler.TaskData : test is running with expression 0/1 * * * * ? 2024-01-07T18:02:11.012+08:00 INFO 23012 --- [ sys-scheduler3] c.s.springdynamicscheduler.TaskData : test is running with expression 0/1 * * * * ?
再调用 stop 接口，通过日志可以发现定时任务停止了执行：
http://localhost:8080/stop?id=1
到此这篇关于spring动态控制定时任务的实现的文章就介绍到这了,更多相关spring动态控制定时任务内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Java数据结构中七种排序算法的实现方法,排序算法可分为两大类,比较类排序和非比较类排序,顾名思义可知它们是通过比较来决定元素间的相对次序,需要详细了解排序算法的朋友可以参考下2024-02-02
本文主要介绍了Mybatis拦截器实现自定义需求，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-05-05
SpringBoot注册服务到Nacos上，由Nacos来做服务的管理，本文主要介绍了springboot读取nacos配置文件的实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
浅谈StringEntity 和 UrlEncodedFormEntity之间的区别
这篇文章主要介绍了StringEntity 和 UrlEncodedFormEntity之间的区别，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
Spring是一个非常流行的Java Web开发框架，它提供了强大的依赖注入、面向切面编程、声明式事务管理等功能，为开发者提供了高效、快速地构建Web应用程序的工具，在这篇文章中，咱们将一步一步地构建一个简单的SpringBean容器,需要的朋友可以参考下2023-06-06
这篇文章主要介绍了Java如何调用Matlab程序的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-09-09
这篇文章主要介绍了Java使用正则表达式去除小数点后面多余的0功能,结合具体实例形式分析了java字符串正则替换相关操作技巧,需要的朋友可以参考下2017-06-06
Springboot项目javax.validation使用方法详解
这篇文章主要介绍了Springboot项目javax.validation使用方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
这篇文章主要为大家介绍了RocketMQ的push消费方式实现示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪<BR>2022-08-08
Java 8 (又称为 jdk 1.8) 是 Java 语言开发的一个主要版本。Lambda 表达式，也可称为闭包，它是推动 Java 8 发布的最重要新特性。本文通过详细的代码示例介绍了Java8新特性感兴趣的朋友可以参考一下2023-04-04
最新评论