Springboot动态添加、移除定时任务

Springboot

发布日期: 2023-06-10

更新日期: 2024-09-27

文章字数: 547

阅读时长: 2 分

阅读次数:

在Spring Boot中，可以使用Spring的任务调度框架来实现动态添加和移除定时任务。Spring的任务调度框架提供了TaskScheduler接口和ScheduledFuture接口，可以用于创建、管理和控制定时任务。

下面是一个简单的示例，演示如何在运行时动态添加和移除定时任务：

首先，你需要创建一个定时任务的执行类，实现Runnable接口，并在run()方法中编写具体的任务逻辑。例如：

public class MyTask implements Runnable {
    @Override
    public void run() {
        // 编写定时任务逻辑
        System.out.println("执行定时任务：" + new Date());
    }
}

然后，在你的Spring Boot应用程序中，你可以创建一个定时任务管理类，用于动态添加和移除定时任务。可以使用TaskScheduler来创建和管理定时任务。例如：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;

@Component
public class TaskManager {
    private final Map<String, ScheduledFuture<?>> tasks = new ConcurrentHashMap<>();

    @Autowired
    private TaskScheduler taskScheduler;

    public void addTask(String taskId, String cronExpression) {
        Runnable task = new MyTask(); // 创建定时任务的执行类
        ScheduledFuture<?> future = taskScheduler.schedule(task, new CronTrigger(cronExpression));
        tasks.put(taskId, future);
        System.out.println("添加定时任务：" + taskId);
    }

    public void removeTask(String taskId) {
        ScheduledFuture<?> future = tasks.get(taskId);
        if (future != null) {
            future.cancel(true);
            tasks.remove(taskId);
            System.out.println("移除定时任务：" + taskId);
        }
    }
}

在上面的示例中，TaskManager类使用了ConcurrentHashMap来保存定时任务的标识符和ScheduledFuture对象之间的映射关系。

最后，你可以在其他的Spring组件中注入TaskManager，并在需要的时候调用addTask()方法来动态添加定时任务，调用removeTask()方法来移除定时任务。例如：

@Autowired
private TaskManager taskManager;

// 添加定时任务
taskManager.addTask("task1", "0/5 * * * * ?"); // 每5秒执行一次

// 移除定时任务
taskManager.removeTask("task1");