一、背景

之前公司經(jīng)常會(huì)遇到配置定時(shí)任務(wù)，簡(jiǎn)單的任務(wù)可以直接依賴spring。簡(jiǎn)單任務(wù)直接使用 @scheduled 注解配合@EnableScheduling。但是如何實(shí)現(xiàn)簡(jiǎn)單的動(dòng)態(tài)cron呢？

開發(fā)原則：盡可能在項(xiàng)目本身去實(shí)現(xiàn)，少依賴第三方框架，避免項(xiàng)目過于臃腫和復(fù)雜。

倆種任務(wù)調(diào)度方式：

二、本篇說明

springBoot 基礎(chǔ)模塊 spring-boot-starter-web 已經(jīng)內(nèi)置 schedule ，無需引入額外依賴。先思考幾個(gè)問題：

1、動(dòng)態(tài) cron 實(shí)現(xiàn)的原理

任務(wù)的 【 停止】是基于 future接口 的cancel() 方法。任務(wù)的 【增加、刪除、啟動(dòng)】是基于 注冊(cè)到 類ScheduledTaskRegistrar 的 ScheduledFuture的數(shù)量。涉及核心類：

ScheduledFuture SchedulingConfigurer ScheduledTaskRegistrar

2、多任務(wù)并行執(zhí)行配置spring默認(rèn)機(jī)制對(duì)schedule是單線程，需要配置多線程并行執(zhí)行。

3、如何配置多個(gè)任務(wù)好多博文，都是配置一個(gè)cron，這讓初學(xué)者很難受。

4、如何配置任務(wù)分組根據(jù)自己業(yè)務(wù)背景，可根據(jù)步驟三，進(jìn)行改造。

5、如何配置服務(wù)啟動(dòng)自啟任務(wù)。想要程序啟動(dòng)時(shí)首次去加我們?cè)O(shè)置的task，只需實(shí)現(xiàn) CommandLineRunner 即可。

6、如何從數(shù)據(jù)庫讀取配置這個(gè)其實(shí)很簡(jiǎn)單，在實(shí)現(xiàn) ScheduledTaskRegistrar 時(shí)，先直接查詢我們需要的數(shù)據(jù)即可。

7、如何優(yōu)雅的實(shí)現(xiàn)我們的代碼這里為了我們多個(gè)task實(shí)現(xiàn)時(shí)，去除臃腫的if else ，使用策略模式去實(shí)現(xiàn)我們的task，這里代碼里面會(huì)具體介紹。

參考類圖：

8、如何去觸發(fā)我們的schedule 【增刪啟?！颗渲煤?task任務(wù)類，注入到 controller ，通過接口直接調(diào)用即可。

三、代碼實(shí)現(xiàn)

先貼出我的github 代碼，下面代碼描述不全。

普通多任務(wù)動(dòng)態(tài)cron 分組多任務(wù)動(dòng)態(tài)cron1. 普通多任務(wù)動(dòng)態(tài)cron 實(shí)現(xiàn)

1.1 對(duì)應(yīng)數(shù)據(jù)庫的實(shí)體類 TaskEntity

@Data@AllArgsConstructor@NoArgsConstructorpublic class TaskEntity { /** * 任務(wù)id */ private int taskId; /** * 任務(wù)說明 */ private String desc; /** * cron 表達(dá)式 */ private String expression;}

1.2 配置每個(gè)任務(wù)實(shí)現(xiàn)

配置任務(wù)接口 TaskService

public interface TaskService { void HandlerJob(); Integer jobId();}

配置任務(wù)接口實(shí)現(xiàn) TaskServiceJob1Impl、TaskServiceJob2Impl …

@Servicepublic class TaskServiceJob1Impl implements TaskService { @Override public void HandlerJob() { System.out.println('------job1 開始執(zhí)行---------：'+new Date()); System.out.println(new SimpleDateFormat('yyyy-MM-dd HH:mm:ss').format(new Date()) + ' ' + Thread.currentThread().getName() + ' 任務(wù)一啟動(dòng)'); try { Thread.sleep(10000);//任務(wù)耗時(shí)10秒 } catch (InterruptedException e) { e.printStackTrace(); } System.out.println(new SimpleDateFormat('yyyy-MM-dd HH:mm:ss').format(new Date()) + ' ' + Thread.currentThread().getName() + ' 結(jié)束'); } @Override public Integer jobId() { return 1; }}

1.3 配置任務(wù)解析器 TaskSolverChooser

注：這里引入策略模式為啥要配置 任務(wù)解析器選擇器:因?yàn)槲覀儗?shí)現(xiàn)多個(gè)任務(wù)時(shí)，一個(gè)任務(wù)對(duì)應(yīng)一個(gè) CronTask，需要在 MyScheduledTask 里面去實(shí)現(xiàn)我們每一個(gè)方法。譬如，我們有100個(gè)任務(wù)就要自定義100個(gè)任務(wù)實(shí)現(xiàn)方法，代碼會(huì)很臃腫，明顯不符合，【開閉原則】，于是這里采用策略模式，解耦我們多個(gè)任務(wù)業(yè)務(wù)實(shí)現(xiàn)邏輯。

@Slf4j@Componentpublic class TaskSolverChooser implements ApplicationContextAware { private ApplicationContext applicationContext; private Map<Integer, TaskService> chooseMap = new HashMap<>(16); /** * 拿到spring context 上下文 */ @Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { this.applicationContext = applicationContext; } @PostConstruct private void registerToTaskSolver(){ Map<String, TaskService> taskServiceMap = applicationContext.getBeansOfType(TaskService.class); for (TaskService value : taskServiceMap.values()) { chooseMap.put(value.jobId(), value); log.info('task {} 處理器: {} 注冊(cè)成功',new Object[]{value.jobId(),value}); } } /** * 獲取需要的job */ public TaskService getTask(Integer jobId){ return chooseMap.get(jobId); }}

1.4 配置MyScheduledTask （動(dòng)態(tài)cron核心配置）

說明：1、配置多線程執(zhí)行任務(wù)2、配置 刷新 task3、配置 停止 task4、配置 執(zhí)行task 業(yè)務(wù)邏輯

@Componentpublic class MyScheduledTask implements SchedulingConfigurer { private volatile ScheduledTaskRegistrar registrar; private final ConcurrentHashMap<Integer, ScheduledFuture<?>> scheduledFutures = new ConcurrentHashMap<>(); private final ConcurrentHashMap<Integer, CronTask> cronTasks = new ConcurrentHashMap<>(); @Autowired private TaskSolverChooser taskSolverChooser; @Override public void configureTasks(ScheduledTaskRegistrar registrar) { //設(shè)置20個(gè)線程,默認(rèn)單線程,如果不設(shè)置的話，不能同時(shí)并發(fā)執(zhí)行任務(wù) registrar.setScheduler(Executors.newScheduledThreadPool(10)); this.registrar = registrar; } /** * 修改 cron 需要 調(diào)用該方法 */ public void refresh(List<TaskEntity> tasks){ //取消已經(jīng)刪除的策略任務(wù) Set<Integer> sids = scheduledFutures.keySet(); for (Integer sid : sids) { if(!exists(tasks, sid)){scheduledFutures.get(sid).cancel(false); } } for (TaskEntity TaskEntity : tasks) { String expression = TaskEntity.getExpression(); //計(jì)劃任務(wù)表達(dá)式為空則跳過 if(!StringUtils.hasLength(expression)){continue; } //計(jì)劃任務(wù)已存在并且表達(dá)式未發(fā)生變化則跳過 if (scheduledFutures.containsKey(TaskEntity.getTaskId()) && cronTasks.get(TaskEntity.getTaskId()).getExpression().equals(expression)) {continue; } //如果策略執(zhí)行時(shí)間發(fā)生了變化，則取消當(dāng)前策略的任務(wù) if(scheduledFutures.containsKey(TaskEntity.getTaskId())){scheduledFutures.get(TaskEntity.getTaskId()).cancel(false);scheduledFutures.remove(TaskEntity.getTaskId());cronTasks.remove(TaskEntity.getTaskId()); } //業(yè)務(wù)邏輯處理 CronTask task = cronTask(TaskEntity, expression); //執(zhí)行業(yè)務(wù) ScheduledFuture<?> future = registrar.getScheduler().schedule(task.getRunnable(), task.getTrigger()); cronTasks.put(TaskEntity.getTaskId(), task); scheduledFutures.put(TaskEntity.getTaskId(), future); } } /** * 停止 cron 運(yùn)行 */ public void stop(List<TaskEntity> tasks){ tasks.forEach(item->{ if (scheduledFutures.containsKey(item.getTaskId())) {// mayInterruptIfRunning設(shè)成false話，不允許在線程運(yùn)行時(shí)中斷，設(shè)成true的話就允許。scheduledFutures.get(item.getTaskId()).cancel(false);scheduledFutures.remove(item.getTaskId()); } }); } /** * 業(yè)務(wù)邏輯處理 */ public CronTask cronTask(TaskEntity TaskEntity, String expression) { return new CronTask(() -> { //每個(gè)計(jì)劃任務(wù)實(shí)際需要執(zhí)行的具體業(yè)務(wù)邏輯 //采用策略，模式 ，執(zhí)行我們的job taskSolverChooser.getTask(TaskEntity.getTaskId()).HandlerJob();}, expression); } private boolean exists(List<TaskEntity> tasks, Integer tid){ for(TaskEntity TaskEntity:tasks){ if(TaskEntity.getTaskId() == tid){return true; } } return false; } @PreDestroy public void destroy() { this.registrar.destroy(); }}

1.5 配置程序啟動(dòng)時(shí)首次去加我們?cè)O(shè)置的task

@Componentpublic class StartInitTask implements CommandLineRunner { @Autowired private MyScheduledTask myScheduledTask; @Override public void run(String... args) throws Exception { List<TaskEntity> list = Arrays.asList(new TaskEntity(1, '測(cè)試1', '0/1 * * * * ?'),new TaskEntity(2, '測(cè)試2', '0/1 * * * * ?') ); myScheduledTask.refresh(list); }}

1.6 配置web接口去觸發(fā)，增刪啟停

@RestControllerpublic class StartController { @Autowired private MyScheduledTask scheduledTask; @PostMapping(value = '/startOrChangeCron') public String changeCron(@RequestBody List<TaskEntity> list){ if (CollectionUtils.isEmpty(list)) { // 這里模擬存在數(shù)據(jù)庫的數(shù)據(jù) list = Arrays.asList( new TaskEntity(1, '測(cè)試1','0/1 * * * * ?') , new TaskEntity(2, '測(cè)試2','0/1 * * * * ?') ); } scheduledTask.refresh(list); return 'task任務(wù):' + list.toString() + '已經(jīng)開始運(yùn)行'; } @PostMapping(value = '/stopCron') public String stopCron(@RequestBody List<TaskEntity> list){ if (CollectionUtils.isEmpty(list)) { // 這里模擬將要停止的cron可通過前端傳來 list = Arrays.asList( new TaskEntity(1, '測(cè)試1','0/1 * * * * ?') , new TaskEntity(2, '測(cè)試2','0/1 * * * * ?') ); } scheduledTask.stop(list); List<Integer> collect = list.stream().map(TaskEntity::getTaskId).collect(Collectors.toList()); return 'task任務(wù):' + collect.toString() + '已經(jīng)停止啟動(dòng)'; }}2. 分組多任務(wù)動(dòng)態(tài)cron 實(shí)現(xiàn)

實(shí)現(xiàn)原理：基于反射實(shí)現(xiàn)，根據(jù)方法全類名，去動(dòng)態(tài)執(zhí)行方法。多任務(wù)分組配置，根據(jù)任務(wù)類型進(jìn)行分組。eg：定時(shí)任務(wù)人員的相關(guān)操作，有檢測(cè)人員離職狀態(tài)，人員業(yè)績(jī)達(dá)標(biāo)，人員考勤…等，作用：對(duì)人員定時(shí)任務(wù)做一個(gè)分類，在同一個(gè)類里面去實(shí)現(xiàn)不同的task，比較《1. 普通多任務(wù)動(dòng)態(tài)cron 實(shí)現(xiàn)》，是一個(gè)類可以實(shí)現(xiàn)一個(gè)task《2. 分組多任務(wù)動(dòng)態(tài)cron 實(shí)現(xiàn)》，是一個(gè)類可以實(shí)現(xiàn)多個(gè)task詳細(xì)可參考： 分組多任務(wù)動(dòng)態(tài)cron

3 測(cè)試記錄

測(cè)試1 項(xiàng)目啟動(dòng)自啟TaskServiceJob1Impl和TaskServiceJob1Impl … 設(shè)置 阻塞10s觀察日志時(shí)間可發(fā)現(xiàn)，已經(jīng)同時(shí)并發(fā)執(zhí)行倆個(gè)任務(wù)。

測(cè)試2 觸發(fā) 刷新【增、刪、啟】我們的task，。其實(shí)這里沒這么智能，如果需要觸發(fā)刷新接口，實(shí)際上是重新加載我們的task，就是對(duì)應(yīng)觸發(fā)我們，增加任務(wù)任務(wù)，刪除任務(wù)，啟動(dòng)任務(wù)。使用idea插件測(cè)試接口

觀察日志

測(cè)試3 觸發(fā) 停止接口，停止一個(gè)接口。這里測(cè)試略過…

四、總結(jié)

其實(shí)實(shí)現(xiàn)簡(jiǎn)單的動(dòng)態(tài)配置，以上代碼可用，比較簡(jiǎn)單。

到此這篇關(guān)于spring schedule配置多任務(wù)動(dòng)態(tài)cron(增刪啟停)的文章就介紹到這了,更多相關(guān)spring schedule 多任務(wù)動(dòng)態(tài)cron內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！