QuartzSchedulerService for Dynamic Scheduling with DB Persistence.

Author: zaxxio

pom.xml

@@ -137,12 +137,12 @@
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-validation</artifactId>
         </dependency>
-        <dependency>
-            <groupId>org.springframework.boot</groupId>
-            <artifactId>spring-boot-devtools</artifactId>
-            <scope>runtime</scope>
-            <optional>true</optional>
-        </dependency>
+<!--        <dependency>-->
+<!--            <groupId>org.springframework.boot</groupId>-->
+<!--            <artifactId>spring-boot-devtools</artifactId>-->
+<!--            <scope>runtime</scope>-->
+<!--            <optional>true</optional>-->
+<!--        </dependency>-->
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-logging</artifactId>

src/main/java/org/wsd/app/config/QuartzConfig.java

@@ -25,7 +25,10 @@
 import lombok.extern.log4j.Log4j2;
 import org.springframework.context.annotation.Configuration;
 
+import javax.sql.DataSource;
+
 @Log4j2
 @Configuration
 public class QuartzConfig {
+
 }

src/main/java/org/wsd/app/controller/QuartzController.java

@@ -33,6 +33,21 @@ public List<JobTimer> getRunningTimers() {
         return quartzSchedulerService.getRunningTimers();
     }
 
+    @DeleteMapping(path = "/deleteJob")
+    public boolean deleteJob(@RequestParam("jobId") String jobId) {
+        return quartzSchedulerService.deleteJob(jobId);
+    }
+
+    @PostMapping(path = "/pauseJob")
+    public boolean pauseJob(@RequestParam("jobId") String jobId) {
+        return quartzSchedulerService.pauseJob(jobId);
+    }
+
+    @PostMapping(path = "/resumeJob")
+    public boolean resumeJob(@RequestParam("jobId") String jobId) {
+        return quartzSchedulerService.resumeJob(jobId);
+    }
+
     @GetMapping(path = "/getRunningTimer")
     public JobTimer getRunningTimer(@RequestParam("timerId") String timerId, @RequestParam("group") String group) {
         return quartzSchedulerService.getRunningTimer(timerId, group);

src/main/java/org/wsd/app/quartz/JobTimer.java

@@ -3,7 +3,6 @@
 import lombok.Data;
 
 import java.io.Serializable;
-import java.util.UUID;
 
 @Data
 public class JobTimer implements Serializable {

src/main/java/org/wsd/app/quartz/QuartzSchedulerService.java

@@ -4,6 +4,7 @@
 import jakarta.annotation.PreDestroy;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.log4j.Log4j2;
+import org.apache.commons.lang3.StringUtils;
 import org.quartz.*;
 import org.quartz.impl.matchers.GroupMatcher;
 import org.springframework.stereotype.Service;
@@ -31,9 +32,9 @@ public void onInit() {
     public void scheduleJob(Class<? extends Job> clazz, JobTimer jobTimer) {
         final JobDetail job = buildJob(clazz, jobTimer);
         final Trigger trigger = buildTrigger(jobTimer);
-
         try {
             scheduler.scheduleJob(job, trigger);
+            log.info("Quartz job has been scheduled with id : " + jobTimer.getJobId());
         } catch (SchedulerException e) {
             log.error(e.getMessage(), e);
             throw new RuntimeException(e);
@@ -53,10 +54,11 @@ public JobTimer getRunningTimer(String timerId, String group) {
 
 
     public void updateTimer(String timerId, String group, JobTimer jobTimer) {
-        final JobKey jobKey = new JobKey(timerId, group);
+        final JobKey jobKey = new JobKey(timerId);
         try {
             final JobDetail job = scheduler.getJobDetail(jobKey);
             job.getJobDataMap().put(jobKey.getName(), jobTimer);
+            scheduler.addJob(job, true, true);
         } catch (SchedulerException e) {
             log.error(e.getMessage(), e);
         }
@@ -84,29 +86,39 @@ public List<JobTimer> getRunningTimers() {
     }
 
     private Trigger buildTrigger(JobTimer jobTimer) {
-        SimpleScheduleBuilder builder = SimpleScheduleBuilder.simpleSchedule()
-                .withIntervalInMilliseconds(jobTimer.getRepeatIntervalMs());
-
-        if (jobTimer.isRunForever()) {
-            builder = builder.repeatForever();
+        if (StringUtils.isNotEmpty(jobTimer.getCronExpression())) {
+            return TriggerBuilder.newTrigger()
+                    .withIdentity(jobTimer.getJobId())
+                    .withSchedule(CronScheduleBuilder.cronSchedule(jobTimer.getCronExpression()))
+                    .startAt(new Date(System.currentTimeMillis() + jobTimer.getInitialOffsetMs()))
+                    .build();
         } else {
-            builder = builder.withRepeatCount(jobTimer.getTotalFireCount() - 1);
+            SimpleScheduleBuilder builder = SimpleScheduleBuilder.simpleSchedule()
+                    .withIntervalInMilliseconds(jobTimer.getRepeatIntervalMs());
+
+            if (jobTimer.isRunForever()) {
+                builder = builder.repeatForever();
+            } else {
+                builder = builder.withRepeatCount(jobTimer.getTotalFireCount() - 1);
+            }
+
+            return TriggerBuilder
+                    .newTrigger()
+                    .withIdentity(jobTimer.getJobId())
+                    // .withIdentity(jobTimer.getJobId(), jobTimer.getGroupName())
+                    .withSchedule(builder)
+                    .startAt(new Date(System.currentTimeMillis() + jobTimer.getInitialOffsetMs()))
+                    .build();
         }
-
-        return TriggerBuilder
-                .newTrigger()
-                .withIdentity(jobTimer.getJobId(), jobTimer.getGroupName())
-                .withSchedule(builder)
-                .startAt(new Date(System.currentTimeMillis() + jobTimer.getInitialOffsetMs()))
-                .build();
     }
 
     private JobDetail buildJob(Class<? extends Job> clazz, JobTimer jobTimer) {
         final JobDataMap dataMap = new JobDataMap();
         dataMap.put(jobTimer.getJobId(), jobTimer);
         return JobBuilder
                 .newJob(clazz)
-                .withIdentity(jobTimer.getJobId(), jobTimer.getGroupName())
+                .withIdentity(jobTimer.getJobId())
+                //.withIdentity(jobTimer.getJobId(), jobTimer.getGroupName())
                 .setJobData(dataMap)
                 .build();
     }
@@ -121,4 +133,33 @@ public void onDestroy() {
         }
     }
 
+    public boolean deleteJob(String jobId) {
+        try {
+            boolean deletedJob = scheduler.deleteJob(new JobKey(jobId));
+            log.info("Quartz job has been deleted with id : " + jobId);
+            return deletedJob;
+        } catch (SchedulerException e) {
+            return false;
+        }
+    }
+
+    public boolean pauseJob(String jobId) {
+        try {
+            scheduler.pauseJob(new JobKey(jobId));
+            log.info("Quartz job has been pause with id : " + jobId);
+            return true;
+        } catch (SchedulerException e) {
+            return false;
+        }
+    }
+
+    public boolean resumeJob(String jobId) {
+        try {
+            log.info("Quartz job has been resumed with id : " + jobId);
+            scheduler.resumeJob(new JobKey(jobId));
+            return true;
+        } catch (SchedulerException e) {
+            return false;
+        }
+    }
 }

src/main/java/org/wsd/app/quartz/SampleJob.java

@@ -1,17 +1,19 @@
 package org.wsd.app.quartz;
 
+import lombok.extern.log4j.Log4j2;
 import org.quartz.Job;
 import org.quartz.JobDataMap;
 import org.quartz.JobExecutionContext;
 import org.quartz.JobExecutionException;
 
 import java.util.Date;
 
+@Log4j2
 public class SampleJob implements Job {
     @Override
     public void execute(JobExecutionContext context) throws JobExecutionException {
         JobDataMap dataMap = context.getJobDetail().getJobDataMap();
-        JobTimer jobTimer = (JobTimer) dataMap.get("SampleJob");
-        System.out.println("Executing job..." + new Date() + " " + jobTimer);
+        JobTimer jobTimer = (JobTimer) dataMap.get("1");
+        log.info("Remaining Job Count : " + jobTimer.getRemainingCount());
     }
 }

src/main/java/org/wsd/app/quartz/listener/SimpleTriggerListener.java

@@ -1,6 +1,7 @@
 package org.wsd.app.quartz.listener;
 
 import lombok.RequiredArgsConstructor;
+import org.apache.commons.lang3.StringUtils;
 import org.quartz.*;
 import org.springframework.stereotype.Service;
 import org.wsd.app.quartz.JobTimer;
@@ -23,7 +24,9 @@ public void triggerFired(Trigger trigger, JobExecutionContext jobExecutionContex
         JobTimer jobTimer = (JobTimer) dataMap.get(timerId);
         if (!jobTimer.isRunForever()) {
             int remainingCount = jobTimer.getRemainingCount();
-            jobTimer.setRemainingCount(remainingCount - 1);
+            if (!StringUtils.isNotEmpty(jobTimer.getCronExpression())) {
+                jobTimer.setRemainingCount(remainingCount - 1);
+            }
         }
         this.quartzSchedulerService.updateTimer(timerId, null, jobTimer);
     }

src/main/resources/application.properties

@@ -7,13 +7,13 @@ spring.jpa.properties.hibernate.javax.cache.provider=org.ehcache.jsr107.EhcacheC
 spring.jpa.properties.hibernate.generate_statistics=false
 
 # Quartz Configuration
-#spring.quartz.job-store-type=jdbc
-#spring.quartz.jdbc.initialize-schema=always
-#spring.quartz.properties.org.quartz.jobStore.useProperties=false
-#spring.quartz.properties.org.quartz.jobStore.driverDelegateClass=org.quartz.impl.jdbcjobstore.PostgreSQLDelegate
-#spring.quartz.properties.org.quartz.jobStore.isClustered=true
-#spring.quartz.auto-startup=true
-#spring.quartz.properties.org.quartz.scheduler.instanceId=AUTO
+spring.quartz.job-store-type=jdbc
+spring.quartz.jdbc.initialize-schema=never
+spring.quartz.properties.org.quartz.jobStore.useProperties=false
+spring.quartz.properties.org.quartz.jobStore.driverDelegateClass=org.quartz.impl.jdbcjobstore.PostgreSQLDelegate
+spring.quartz.properties.org.quartz.jobStore.isClustered=true
+spring.quartz.auto-startup=true
+spring.quartz.properties.org.quartz.scheduler.instanceId=AUTO
 
 # Log Configuration
 logging.file.name=logs/app.log