在Spring Boot中,可以使用ThreadPoolTaskScheduler来动态地添加和移除定时任务。以下是一个简单的例子:
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.util.concurrent.ScheduledFuture;
@Configuration
public class DynamicScheduleConfig {
private final ThreadPoolTaskScheduler taskScheduler;
public DynamicScheduleConfig(ThreadPoolTaskScheduler taskScheduler) {
this.taskScheduler = taskScheduler;
}
@Bean
public ScheduledTaskRegistrar taskRegistrar() {
return new ScheduledTaskRegistrar();
}
public ScheduledFuture<?> addTask(Runnable task) {
return taskScheduler.schedule(task, new CronTrigger("0/5 * * * * ?")); // 例子使用了Cron表达式,每5秒执行一次
}
public void removeTask(ScheduledFuture<?> task) {
task.cancel(true);
}
}
使用时,你可以注入DynamicScheduleConfig并调用addTask方法来添加任务,调用removeTask方法来移除任务。
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.concurrent.ScheduledFuture;
@Service
public class ScheduleService {
private final DynamicScheduleConfig scheduleConfig;
private ScheduledFuture<?> myTask;
@Autowired
public ScheduleService(DynamicScheduleConfig scheduleConfig) {
this.scheduleConfig = scheduleConfig;
}
public void addTask() {
myTask = scheduleConfig.addTask(() -> {
// 定时任务的逻辑
System.out.println("Task executed");
});
}
public void removeTask() {
if (myTask != null) {
scheduleConfig.removeTask(myTask);
}
}
}
确保ThreadPoolTaskScheduler已经配置好,通常在一个配置类中配置它:
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
@Configuration
public class AsyncConfig {
@Bean(destroyMethod="shutdown")
public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
scheduler.setPoolSize(10);
scheduler.setThreadNamePrefix("my-scheduler-");
scheduler.initialize();
return scheduler;
}
}
这样,你就可以动态地添加和移除定时任务了。