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动态控制定时任务内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
Ribbon是Netflix的组件之一，负责注册中心的负载均衡，有助于控制HTTP和TCP客户端行为。Spring Cloud Netflix Ribbon一般配合Ribbon进行使用，利用在Eureka中读取的服务信息，在调用服务节点时合理进行负载2023-02-02
Spring的BeanFactoryPostProcessor接口示例代码详解
这篇文章主要介绍了Spring的BeanFactoryPostProcessor接口,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-02-02
这篇文章主要介绍了如何使用ModelMapper库进行对象之间的属性映射实现详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-07-07
Spring中的singleton和prototype的实现
这篇文章主要介绍了Spring中的singleton和prototype的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-07-07
这篇文章主要为大家详细介绍了Java动态生成doc文档的多种方式,具有一定的参考价值，感兴趣的小伙伴们可以参考一下2016-09-09
本文主要介绍java selenium 操作浏览器,这里整理了相关资料，并附示例代码，有需要的小伙伴可以参考下2016-08-08
很多springboot项目使用的是yml格式，主要目的是方便对读懂其他人的项目，下面小编通过本文给大家分享Spring Boot使用yml格式进行配置的方法，需要的朋友参考下吧2018-04-04
桥接模式是一种很实用的结构型模式，如果系统中某个类存在两个独立变化的维度，通过桥接模式将这两个维度分离出来，使两者可以独立扩展2023-02-02
今天小编就为大家分享一篇关于JavaTCP上传文本文件代码，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2019-02-02
这篇文章主要介绍了Spring Cloud如何使用Feign构造多参数的请求,以GET以及POST方法的请求为例进行讲解,具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
最新评论