定時任務管理器

• 負責啟動一個定時任務、停止一個定時任務、更新一個定時任務

/*** 定時任務管理器* 1、創建并啟動一個定時任務* 2、停止一個定時任務* 3、更新一個定時任務*/
public class ScheduleManager {private final Logger logger = LoggerFactory.getLogger(ScheduleManager.class);@Autowiredprivate ThreadPoolTaskScheduler taskScheduler;/*** 內部正在執行的定時任務緩存*/private final Map<String, ScheduleTaskHolder> cache = new ConcurrentHashMap<>();public ScheduleManager(ThreadPoolTaskScheduler taskScheduler) {this.taskScheduler = taskScheduler;}/*** 啟動一個定時任務** @param scheduleTask 定時任務實現類* @param cron         定時任務的cron表達式* @return key*/public String startTask(ScheduleTask scheduleTask, String cron) {ScheduledFuture<?> scheduledFuture = taskScheduler.schedule(scheduleTask, new CronTrigger(cron));String key = UUID.randomUUID().toString();ScheduleTaskHolder holder = new ScheduleTaskHolder(scheduleTask, scheduledFuture);cache.put(key, holder);logger.info("{} 定時任務啟動成功！唯一標識為：{}", scheduleTask.getName(), key);return key;}/*** 停止一個定時任務** @param key 定時任務的唯一標識*/public void stopTask(String key) {if (StringUtils.isBlank(key)) {return;}ScheduleTaskHolder holder = cache.get(key);if (Objects.isNull(holder)) {return;}ScheduledFuture scheduledFuture = holder.getScheduledFuture();boolean cancel = scheduledFuture.cancel(true);if (cancel) {logger.info("{} 定時任務停止成功！唯一標識為：{}", holder.getScheduleTask().getName(), key);} else {logger.error("{} 定時任務停止失敗！唯一標識為：{}", holder.getScheduleTask().getName(), key);}}/*** 更新一個定時任務的執行時間** @param key  定時任務的唯一標識* @param cron 新的cron表達式* @return key*/public String changeTask(String key, String cron) {if (StringUtils.isAnyBlank(key, cron)) {throw new RuntimeException("定時任務的唯一標識以及新的執行表達式不能為空");}ScheduleTaskHolder holder = cache.get(key);if (Objects.isNull(holder)) {throw new RuntimeException(key + "唯一標識不存在");}stopTask(key);return startTask(holder.getScheduleTask(), cron);}}

定時任務的任務接口

• 定時任務的運行邏輯在重寫的 run 方法中實現

/*** 定時任務的任務接口*/
public interface ScheduleTask extends Runnable {/*** 獲取定時任務的名稱** @return*/String getName();}

定時任務和定時任務結果的緩存對象

• 負責緩存定時任務和控制該定時任務

/*** 定時任務和定時任務結果的緩存對象*/
public class ScheduleTaskHolder implements Serializable {/*** 執行任務實體*/private ScheduleTask scheduleTask;/*** 執行任務的結果實體*/private ScheduledFuture scheduledFuture;public ScheduleTaskHolder() {}public ScheduleTaskHolder(ScheduleTask scheduleTask, ScheduledFuture scheduledFuture) {this.scheduleTask = scheduleTask;this.scheduledFuture = scheduledFuture;}public ScheduleTask getScheduleTask() {return scheduleTask;}public void setScheduleTask(ScheduleTask scheduleTask) {this.scheduleTask = scheduleTask;}public ScheduledFuture getScheduledFuture() {return scheduledFuture;}public void setScheduledFuture(ScheduledFuture scheduledFuture) {this.scheduledFuture = scheduledFuture;}
}

定時任務

• 具體實現的定時任務

/*** 定時任務*/
public class ThreadPoolWarnTask implements ScheduleTask {@Overridepublic String getName() {return "threadPoolWarnTask";}@Overridepublic void run() {}
}

使用姿勢

• 引入 ScheduleManager 對象
• 開啟定時任務、關閉定時任務、更新定時任務

// 傳入一個定時任務和CRON表達式開啟定時任務，返回該定時任務的唯一標識
String taskKey = scheduleManager.startTask(threadPoolWarnTask, CRON);
// 停止定時任務
scheduleManager.stopTask(taskKey);
// 更新定時任務的CRON
scheduleManager.changeTask(taskKey, newCRON);