代码补全

2023-03-11 10:59:40 +08:00
parent 79c73438da
commit 27c06ce748
4 changed files with 132 additions and 82 deletions
--- a/zyplayer-doc-data/src/main/java/com/zyplayer/doc/data/repository/manage/mapper/BackupTaskMapper.java
+++ b/zyplayer-doc-data/src/main/java/com/zyplayer/doc/data/repository/manage/mapper/BackupTaskMapper.java
@@ -4,9 +4,7 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.zyplayer.doc.data.repository.manage.entity.BackupTask;

 /**
- * <p>
- *  Mapper 接口
- * </p>
+ *  备份任务Mapper 接口
 *
 * @author diantu
 * @since 2023-03-03
--- a/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/framework/utils/QuartzManagerUtils.java
+++ b/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/framework/utils/QuartzManagerUtils.java
@@ -1,9 +1,8 @@
 package com.zyplayer.doc.db.framework.utils;

-import lombok.extern.slf4j.Slf4j;
 import org.quartz.*;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Component;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;


 /**
@@ -12,10 +11,10 @@ import org.springframework.stereotype.Component;
 * @author diantu
 * @since 2023年3月8日
 */
-@Slf4j
-@Component
 public class QuartzManagerUtils {

+    private static Logger logger = LoggerFactory.getLogger(QuartzManagerUtils.class);
+
    /**
     * 参数传递key
     */
@@ -26,109 +25,103 @@ public class QuartzManagerUtils {
     */
    public static final String CLASS_NAME = "com.zyplayer.doc.db.framework.db.job.BackupJob";

-    /**
-     * 程序调度器
-     */
-//    @Autowired
-//    private Scheduler scheduler;
-



    /**
     * 添加定时任务
     */
-//    public void add(Integer id, String cronExpression, String param, Boolean status) {
-//        try {
-//            // 构建job信息
-//            JobDetail jobDetail = JobBuilder.newJob(getClass(CLASS_NAME).getClass()).withIdentity(getKey(id)).usingJobData(PARAM_KEY, param).build();
-//            // 表达式调度构建器(即任务执行的时间)
-//            CronScheduleBuilder scheduleBuilder = CronScheduleBuilder.cronSchedule(cronExpression);
-//            // 按新的cronExpression表达式构建一个新的trigger
-//            CronTrigger trigger = TriggerBuilder.newTrigger().withIdentity(getKey(id)).withSchedule(scheduleBuilder).build();
-//            // 创建定时任务
-//            scheduler.scheduleJob(jobDetail, trigger);
-//            // 停止
-//            if (!status) {
-//                stop(id);
-//            }
-//        } catch (Exception e) {
-//            log.error("添加定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public static void add(Scheduler scheduler,Long id, String cronExpression, String param, Boolean status) {
+        try {
+            // 构建job信息
+            JobDetail jobDetail = JobBuilder.newJob(getClass(CLASS_NAME).getClass()).withIdentity(getKey(id)).usingJobData(PARAM_KEY, param).build();
+            // 表达式调度构建器(即任务执行的时间)
+            CronScheduleBuilder scheduleBuilder = CronScheduleBuilder.cronSchedule(cronExpression);
+            // 按新的cronExpression表达式构建一个新的trigger
+            CronTrigger trigger = TriggerBuilder.newTrigger().withIdentity(getKey(id)).withSchedule(scheduleBuilder).build();
+            // 创建定时任务
+            scheduler.scheduleJob(jobDetail, trigger);
+            // 停止
+            if (!status) {
+                stop(scheduler,id);
+            }
+        } catch (Exception e) {
+            logger.error("添加定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 编辑定时任务
     */
-//    public void update(Integer id, String cronExpression, String param, Boolean status) {
-//        try {
-//            // 判断是否存在，存在先删除
-//            if (scheduler.checkExists(JobKey.jobKey(getKey(id)))) {
-//                scheduler.deleteJob(JobKey.jobKey(getKey(id)));
-//            }
-//            // 再创建
-//            add(id, cronExpression, param, status);
-//        } catch (Exception e) {
-//            log.error("修改定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public void update(Scheduler scheduler,Long id, String cronExpression, String param, Boolean status) {
+        try {
+            // 判断是否存在，存在先删除
+            if (scheduler.checkExists(JobKey.jobKey(getKey(id)))) {
+                scheduler.deleteJob(JobKey.jobKey(getKey(id)));
+            }
+            // 再创建
+            add(scheduler,id, cronExpression, param, status);
+        } catch (Exception e) {
+            logger.error("修改定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 暂停任务
     */
-//    public void stop(Integer id) {
-//        try {
-//            scheduler.pauseJob(JobKey.jobKey(getKey(id)));
-//        } catch (SchedulerException e) {
-//            // 暂停定时任务失败
-//            log.error("暂停定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public static void stop(Scheduler scheduler, Long id) {
+        try {
+            scheduler.pauseJob(JobKey.jobKey(getKey(id)));
+        } catch (SchedulerException e) {
+            // 暂停定时任务失败
+            logger.error("暂停定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 恢复任务
     */
-//    public void start(Integer id) {
-//        try {
-//            scheduler.resumeJob(JobKey.jobKey(getKey(id)));
-//        } catch (SchedulerException e) {
-//            // 暂停定时任务失败
-//            log.error("启动定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public void start(Scheduler scheduler,Long id) {
+        try {
+            scheduler.resumeJob(JobKey.jobKey(getKey(id)));
+        } catch (SchedulerException e) {
+            // 暂停定时任务失败
+            logger.error("启动定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 立即执行一次
     */
-//    public void run(Integer id) {
-//        try {
-//            scheduler.triggerJob(JobKey.jobKey(getKey(id)));
-//        } catch (SchedulerException e) {
-//            // 暂停定时任务失败
-//            log.error("执行定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public void run(Scheduler scheduler,Long id) {
+        try {
+            scheduler.triggerJob(JobKey.jobKey(getKey(id)));
+        } catch (SchedulerException e) {
+            // 暂停定时任务失败
+            logger.error("执行定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 删除定时任务
     */
-//    public void delete(Integer id) {
-//        try {
-//            // 停止触发器
-//            scheduler.pauseTrigger(TriggerKey.triggerKey(getKey(id)));
-//            // 移除触发器
-//            scheduler.unscheduleJob(TriggerKey.triggerKey(getKey(id)));
-//            // 删除任务
-//            scheduler.deleteJob(JobKey.jobKey(getKey(id)));
-//        } catch (Exception e) {
-//            log.error("删除定时任务失败：{}", e.getMessage());
-//        }
-//    }
+    public void delete(Scheduler scheduler,Long id) {
+        try {
+            // 停止触发器
+            scheduler.pauseTrigger(TriggerKey.triggerKey(getKey(id)));
+            // 移除触发器
+            scheduler.unscheduleJob(TriggerKey.triggerKey(getKey(id)));
+            // 删除任务
+            scheduler.deleteJob(JobKey.jobKey(getKey(id)));
+        } catch (Exception e) {
+            logger.error("删除定时任务失败：{}", e.getMessage());
+        }
+    }

    /**
     * 根据类名获取类
     */
-    private Job getClass(String className) throws Exception {
+    private static Job getClass(String className) throws Exception {
        Class<?> class1 = Class.forName(className);
        return (Job) class1.newInstance();
    }
@@ -136,7 +129,7 @@ public class QuartzManagerUtils {
    /**
     * 拼接key
     */
-    public String getKey(Integer id) {
+    public static String getKey(Long id) {
        return "dbBackUp-" + id;
    }
 }
--- a/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/backup/BackupTaskService.java
+++ b/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/backup/BackupTaskService.java
@@ -0,0 +1,10 @@
+package com.zyplayer.doc.db.service.backup;
+
+/**
+ * 备份任务
+ *
+ * @author diantu
+ * @since 2023年3月8日
+ */
+public interface BackupTaskService {
+}
--- a/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/backup/impl/BackupTaskServiceImpl.java
+++ b/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/backup/impl/BackupTaskServiceImpl.java
@@ -0,0 +1,49 @@
+package com.zyplayer.doc.db.service.backup.impl;
+
+import cn.hutool.core.collection.CollectionUtil;
+import com.alibaba.fastjson.JSON;
+import com.zyplayer.doc.data.repository.manage.entity.BackupTask;
+import com.zyplayer.doc.data.repository.manage.mapper.BackupTaskMapper;
+import com.zyplayer.doc.db.framework.utils.QuartzManagerUtils;
+import com.zyplayer.doc.db.service.backup.BackupTaskService;
+import org.quartz.Scheduler;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import javax.annotation.PostConstruct;
+import java.util.List;
+
+/**
+ * 备份任务实现
+ *
+ * @author diantu
+ * @since 2023年3月8日
+ */
+public class BackupTaskServiceImpl implements BackupTaskService {
+
+    private static Logger logger = LoggerFactory.getLogger(BackupTaskServiceImpl.class);
+
+    @Autowired
+    private Scheduler scheduler;
+
+    @Autowired
+    private BackupTaskMapper backupTaskMapper;
+
+    /**
+     * 系统启动执行
+     */
+    @PostConstruct
+    public void init() {
+        List<BackupTask> list = backupTaskMapper.selectList(null);
+        if (CollectionUtil.isNotEmpty(list)) {
+            for (BackupTask item : list) {
+                try {
+                    QuartzManagerUtils.add(scheduler,item.getId(), item.getCron(), JSON.toJSONString(item.getParam()), item.getStatus());
+                } catch (Exception e) {
+                    logger.error(e.getMessage());
+                }
+            }
+        }
+    }
+
+}