Spring Boot 实现定时任务动态管理

在Spring Boot中,可以使用@Scheduled注解来创建定时任务,但这种方式不支持动态管理定时任务。要实现定时任务的动态管理,你可以使用ScheduledTaskRegistrar来注册和注销定时任务。

以下是一个简单的例子,展示如何在Spring Boot中动态添加和移除定时任务:




import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.support.CronTrigger;
 
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
 
@Configuration
@EnableScheduling
public class DynamicScheduleConfig implements SchedulingConfigurer {
 
    private final ConcurrentMap<String, Runnable> schedules = new ConcurrentHashMap<>();
 
    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
        taskRegistrar.setScheduler(Executors.newScheduledThreadPool(5));
        schedules.forEach(taskRegistrar::schedule);
    }
 
    public void addTask(String taskId, Runnable task, String cronExpression) {
        schedules.put(taskId, task);
        triggerTask(taskId, cronExpression);
    }
 
    public void removeTask(String taskId) {
        Runnable task = schedules.remove(taskId);
        if (task != null) {
            // 停止任务的执行
        }
    }
 
    public void triggerTask(String taskId, String cronExpression) {
        ScheduledTaskRegistrar registrar = new ScheduledTaskRegistrar();
        Runnable task = schedules.get(taskId);
        if (task != null) {
            CronTrigger trigger = new CronTrigger(cronExpression);
            registrar.addTriggerTask(task, trigger);
        }
        // 重新配置ScheduledTaskRegistrar以应用更改
    }
}

在这个例子中,我们使用了ConcurrentHashMap来存储定时任务,并且提供了添加、移除和触发定时任务的方法。configureTasks方法会在应用启动时加载所有任务,而addTask方法可以动态添加新的任务,removeTask方法可以移除任务,triggerTask方法可以改变任务的触发规则。

请注意,这个例子没有完全实现停止已经执行的任务的逻辑,实际应用中你需要管理正在执行的任务并相应地停止它们。同时,这个例子使用了Executors.newScheduledThreadPool来创建一个新的线程池,你可能需要根据实际情况来管理线程池的生命周期。

最后修改于:2024年09月06日 09:31

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日