feat(queue): add getJobSchedulersCount method (#2945)

taskforcesh · Nov 29, 2024 · 38820dc · 38820dc
1 parent b56c3b4
commit 38820dc
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 6 deletions.
diff --git a/src/classes/job-scheduler.ts b/src/classes/job-scheduler.ts
@@ -278,12 +278,11 @@ export class JobScheduler extends QueueBase {
     return Promise.all(jobs);
   }
 
-  async getSchedulersCount(
-    client: RedisClient,
-    prefix: string,
-    queueName: string,
-  ): Promise<number> {
-    return client.zcard(`${prefix}:${queueName}:repeat`);
+  async getSchedulersCount(): Promise<number> {
+    const jobSchedulersKey = this.keys.repeat;
+    const client = await this.client;
+
+    return client.zcard(jobSchedulersKey);
   }
 
   private getSchedulerNextJobId({

diff --git a/src/classes/queue.ts b/src/classes/queue.ts
@@ -591,6 +591,16 @@ export class Queue<
     return (await this.jobScheduler).getJobSchedulers(start, end, asc);
   }
 
+  /**
+   *
+   * Get the number of job schedulers.
+   *
+   * @returns The number of job schedulers.
+   */
+  async getJobSchedulersCount(): Promise<number> {
+    return (await this.jobScheduler).getSchedulersCount();
+  }
+
   /**
    * Removes a repeatable job.
    *

diff --git a/tests/test_job_scheduler.ts b/tests/test_job_scheduler.ts
@@ -183,6 +183,9 @@ describe('Job Scheduler', function () {
 
       const delayed = await queue.getDelayed();
       expect(delayed).to.have.length(3);
+
+      const jobSchedulersCount = await queue.getJobSchedulersCount();
+      expect(jobSchedulersCount).to.be.eql(3);
     });
   });