feat(job-scheduler): save iteration count (#3018)

taskforcesh · Jan 24, 2025 · ad5c07c · ad5c07c
1 parent 824fc4c
commit ad5c07c
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 1 deletion.
diff --git a/src/classes/job-scheduler.ts b/src/classes/job-scheduler.ts
@@ -275,6 +275,7 @@ export class JobScheduler extends QueueBase {
     if (jobData) {
       return {
         key,
+        iterationCount: parseInt(jobData.ic) || null,
         name: jobData.name,
         endDate: parseInt(jobData.endDate) || null,
         tz: jobData.tz || null,

diff --git a/src/commands/addJobScheduler-6.lua b/src/commands/addJobScheduler-6.lua
@@ -76,7 +76,7 @@ local function storeRepeatableJob(schedulerId, schedulerKey, repeatKey, nextMill
     table.insert(optionalValues, templateData)
   end
 
-  rcall("HMSET", schedulerKey, "name", opts['name'],
+  rcall("HMSET", schedulerKey, "name", opts['name'], "ic", 1,
     unpack(optionalValues))
 end
 

diff --git a/src/commands/updateJobScheduler-7.lua b/src/commands/updateJobScheduler-7.lua
@@ -47,6 +47,7 @@ if prevMillis ~= false then
     local schedulerAttributes = rcall("HMGET", schedulerKey, "name", "data")
 
     rcall("ZADD", repeatKey, nextMillis, jobSchedulerId)
+    rcall("HINCRBY", schedulerKey, "ic", 1)
 
     local eventsKey = KEYS[5]
     local metaKey = KEYS[2]

diff --git a/src/interfaces/job-scheduler-json.ts b/src/interfaces/job-scheduler-json.ts
@@ -9,6 +9,7 @@ export interface JobSchedulerJson<D = any> {
   key: string; // key is actually the job scheduler id
   name: string;
   id?: string | null;
+  iterationCount?: number;
   endDate: number | null;
   tz: string | null;
   pattern: string | null;

diff --git a/tests/test_job_scheduler.ts b/tests/test_job_scheduler.ts
@@ -408,6 +408,7 @@ describe('Job Scheduler', function () {
 
     expect(scheduler).to.deep.equal({
       key: 'test',
+      iterationCount: 1,
       name: 'test',
       endDate: null,
       tz: null,
@@ -750,6 +751,7 @@ describe('Job Scheduler', function () {
 
         expect(scheduler).to.deep.equal({
           key: 'rrule',
+          iterationCount: 1,
           name: 'rrule',
           endDate: null,
           next: 1486481042000,
@@ -1719,6 +1721,7 @@ describe('Job Scheduler', function () {
 
       expect(jobSchedulers[0]).to.deep.equal({
         key: 'test',
+        iterationCount: 2,
         name: 'a',
         endDate: null,
         tz: 'Asia/Calcutta',