[ server ] [ scheduler ]- script scheduler - 5

server/src/main/java/io/wdd/rpc/controller/SchedulerController.java

@@ -26,18 +26,18 @@ public class SchedulerController {
     /**
      * --------------------------------------------------------------
      * 页面定时任务部分
-     *  应该只有脚本功能才可以定时，目前一阶段的功能
-     * */
+     * 应该只有脚本功能才可以定时，目前一阶段的功能
+     */
 
     @ApiOperation(value = "新增一个定时脚本任务")
     @PostMapping(value = "/script/create")
-    public R<String> createScriptScheduler(
+    public R<List<String>> createScriptScheduler(
             @ApiParam(name = "scheduleScript") @RequestBody() ScriptSchedulerVO scriptSchedulerVO
     ) {
 
-        octopusQuartzService.createScriptScheduledMission(scriptSchedulerVO);
+        List<String> resultList = octopusQuartzService.createScriptScheduledMission(scriptSchedulerVO);
 
-        return R.ok("ok");
+        return R.ok(resultList);
 
     }
 

server/src/main/java/io/wdd/rpc/scheduler/job/AgentScriptSchedulerJob.java

@@ -25,7 +25,7 @@ public class AgentScriptSchedulerJob extends QuartzJobBean {
                 .getJobDataMap();
 
             // ScriptScheduleDTO
-        ScriptSchedulerDTO scriptSchedulerDTO = (ScriptSchedulerDTO) jobDataMap.get("scriptSchedulerPO");
+        ScriptSchedulerDTO scriptSchedulerDTO = (ScriptSchedulerDTO) jobDataMap.get("scriptSchedulerDTO");
 
         // 调用实际任务执行器
         agentApplyScheduledScript.apply(scriptSchedulerDTO);

server/src/main/java/io/wdd/rpc/scheduler/service/QuartzSchedulerService.java

@@ -98,9 +98,11 @@ public interface QuartzSchedulerService {
     /**
      * --------------------------------------------------------------
      * 页面定时任务部分
-     *  应该只有脚本功能才可以定时，目前一阶段的功能
-     * */
-    void createScriptScheduledMission(ScriptSchedulerVO scriptSchedulerVO);
+     * 应该只有脚本功能才可以定时，目前一阶段的功能
+     *
+     * @return
+     */
+    List<String> createScriptScheduledMission(ScriptSchedulerVO scriptSchedulerVO);
 
 
 }

server/src/main/java/io/wdd/rpc/scheduler/service/QuartzSchedulerServiceImpl.java

@@ -2,6 +2,7 @@ package io.wdd.rpc.scheduler.service;
 
 import io.wdd.common.handler.MyRuntimeException;
 import io.wdd.common.utils.FunctionReader;
+import io.wdd.common.utils.TimeUtils;
 import io.wdd.rpc.scheduler.beans.OctopusQuartzJob;
 import io.wdd.rpc.scheduler.beans.ScriptSchedulerDTO;
 import io.wdd.rpc.scheduler.beans.ScriptSchedulerVO;
@@ -13,7 +14,6 @@ import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.beanutils.BeanUtils;
 import org.apache.commons.lang3.ObjectUtils;
 import org.apache.commons.lang3.RandomStringUtils;
-import org.apache.commons.lang3.StringUtils;
 import org.quartz.*;
 import org.quartz.DateBuilder.IntervalUnit;
 import org.quartz.impl.matchers.GroupMatcher;
@@ -23,6 +23,7 @@ import org.springframework.stereotype.Service;
 
 import javax.annotation.PostConstruct;
 import javax.annotation.Resource;
+import java.time.LocalDateTime;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -38,12 +39,10 @@ import static org.quartz.TriggerBuilder.newTrigger;
 @Service
 public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
 
-    @Autowired
-    private Scheduler scheduler;
-
     @Resource
     ScriptSchedulerService scriptSchedulerService;
-
+    @Autowired
+    private Scheduler scheduler;
 
 
     /**
@@ -52,14 +51,11 @@ public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
      *  应该只有脚本功能才可以定时，目前一阶段的功能
      * */
 
-
-
     /**
-     *
-     *
-     * */
+     * @return
+     */
     @Override
-    public void createScriptScheduledMission(ScriptSchedulerVO scriptSchedulerVO) {
+    public List<String> createScriptScheduledMission(ScriptSchedulerVO scriptSchedulerVO) {
 
         // validate important value
         ScriptSchedulerDTO scriptSchedulerDTO = validateAndConvertToScriptSchedulerDTO(scriptSchedulerVO);
@@ -68,10 +64,13 @@ public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
         // build the trigger
         // bind job and trigger
         HashMap<String, Object> dataMap = new HashMap<>();
-        dataMap.put("scriptSchedulerDTO", scriptSchedulerDTO);
+        dataMap.put(
+                "scriptSchedulerDTO",
+                scriptSchedulerDTO
+        );
         this.addMission(
                 AgentScriptSchedulerJob.class,
-                scriptSchedulerVO.getName(),
+                scriptSchedulerVO.getSchedulerUuid(),
                 SCHEDULE_MISSION_GROUP_NAME,
                 1,
                 scriptSchedulerVO.getScriptContent(),
@@ -79,32 +78,58 @@ public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
         );
 
         // persistent the script scheduled mission
-        // todo dto should store more info
+        // dto should store more info
         ScriptSchedulerPO scriptSchedulerPO = convertToScriptSchedulerPO(scriptSchedulerDTO);
-        scriptSchedulerService.save(scriptSchedulerPO);
+        log.info(String.valueOf(scriptSchedulerPO));
+
+//        scriptSchedulerService.save(scriptSchedulerPO);
+
+        return scriptSchedulerDTO.getResultKeyList();
     }
 
     /**
-     *  转换 中间层 --> 持久层
+     * 转换 中间层 --> 持久层
      *
      * @param dto 定时脚本任务-中间转换状态-实体类
      * @return 定时脚本任务-持久化-实体类
      */
     private ScriptSchedulerPO convertToScriptSchedulerPO(ScriptSchedulerDTO dto) {
-        // todo should be a static method
 
-
-        return ScriptSchedulerPO
+        ScriptSchedulerPO schedulerPO = ScriptSchedulerPO
                 .builder()
                 .cronExpress(dto.getCronExpress())
                 .schedulerUuid(dto.getSchedulerUuid())
                 .scriptContent(String.valueOf(dto.getCommandList()))
                 .targetMachine(String.valueOf(dto.getTargetMachineList()))
                 .build();
+
+        // 获取JobDetail存储上次调度时间和下次执行时间
+        try {
+            Trigger schedulerTrigger = scheduler.getTrigger(
+                    new TriggerKey(
+                            dto.getSchedulerUuid(),
+                            SCHEDULE_MISSION_GROUP_NAME
+                    )
+            );
+
+
+            LocalDateTime last_schedule_time = TimeUtils.cvFromDate(schedulerTrigger.getFinalFireTime());
+            LocalDateTime next_schedule_time = TimeUtils.cvFromDate(schedulerTrigger.getNextFireTime());
+
+            schedulerPO.setLastScheduleTime(last_schedule_time);
+            schedulerPO.setNextScheduleTime(next_schedule_time);
+
+        } catch (SchedulerException e) {
+            throw new RuntimeException(e);
+        }
+
+
+        return schedulerPO;
     }
 
     /**
-     *  转换 视图层 --> 中间层
+     * 转换 视图层 --> 中间层
+     *
      * @param scriptSchedulerVO 定时脚本任务-前端页面-实体类
      * @return 定时脚本任务-中间转换状态-实体类
      */
@@ -130,13 +155,16 @@ public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
             throw new MyRuntimeException("target machine wrong !");
         }
         ArrayList<String> targetMachineList = new ArrayList<>();
-        targetMachineList.stream().forEach(
-                targetMachineList::add
-        );
+        targetMachineList
+                .stream()
+                .forEach(
+                        targetMachineList::add
+                );
 
         // 生成DTO对象
         ScriptSchedulerDTO dto = new ScriptSchedulerDTO();
-        BeanUtils.copyProperties(dto,scriptSchedulerVO);
+        BeanUtils.copyProperties(dto,
+                                 scriptSchedulerVO);
 
         // 设置属性值
         dto.setCommandList(null);
@@ -555,6 +583,4 @@ public class QuartzSchedulerServiceImpl implements QuartzSchedulerService {
     }
 
 
-
-
 }