Spring动态配置定时任务(启用、停用)
程序员文章站
2022-05-30 23:41:48
...
package com.csdn.taskk;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import java.util.HashMap;
import java.util.concurrent.ScheduledFuture;
/**
* 定时计费工具类
* @version 1.0.0
*/
@Configuration
public class ScheduleConfig {
private static final Logger logger = LoggerFactory.getLogger(ScheduleConfig.class);
@Autowired
private ThreadPoolTaskScheduler threadPoolTaskScheduler;
/**
* 存放所有启动定时任务对象
*/
private static HashMap<String, ScheduledFuture<?>> scheduleMap = new HashMap<>();
@Bean
public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
executor.initialize();
executor.setPoolSize(20);
executor.setThreadNamePrefix("taskExecutor-");
executor.setWaitForTasksToCompleteOnShutdown(true);
executor.setAwaitTerminationSeconds(60);
return executor;
}
/**
* 手动启动定时器
* <p><b>修改人:</b><br> 2019年11月28日 19:52:56<br>
* <p><b>修改说明:</b><br> <br>
* @param crons 定时表达式
* @return
*/
public void startCron(String crons) {
try {
//每次都会把之前所有的定时任务都关闭,根据新的状态重新启用一次,达到最新配置
ScheduledFuture<?> scheduledFuture =
scheduleMap.get(this.getClass().getName());
//此处只需要判断对象是否为空即可
if (scheduledFuture != null) {
scheduledFuture.cancel(true);
}
//重新开启定时任务
String name = Thread.currentThread().getName();
threadPoolTaskScheduler = threadPoolTaskScheduler();
ScheduledFuture<?> future =this.threadPoolTaskScheduler.schedule(new myTask(), new CronTrigger(crons));
//将配置存入到map中 以便下次停用时可以停用
scheduleMap.put(this.getClass().getName(), future);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 执行计费
* <p><b>修改人:</b><br> 2019年11月28日 19:02:10<br>
* <p><b>修改说明:</b><br> <br>
* @return
*/
private class myTask implements Runnable {
@Override
public void run() {
System.out.println("写业务处理的内容");
}
}
}
备注:在注入ThreadPoolTaskScheduler 的时候可能会报错,需要在spring配置文件进行配置