feat: add support for getWorkersCount

taskforcesh · Apr 7, 2024 · f07db90 · f07db90
1 parent 9c550c1
commit f07db90
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lib/responders/bull-responders.ts b/lib/responders/bull-responders.ts
@@ -17,7 +17,7 @@ function paginate(
   start = start || 0;
   end = end || -1;
   return (<any>queue)
-    [method](start, end, opts)
+  [method](start, end, opts)
     .then(function (jobs: Bull.Job[]) {
       respond(ws, messageId, jobs);
     });
@@ -100,10 +100,13 @@ async function respondQueueCommand(
     case "getCompletedCount":
     case "getFailedCount":
     case "getRepeatableCount":
-    case "getWorkersCount":
       const count = await (<any>queue)[data.cmd]();
       respond(ws, msg.id, count);
       break;
+    case "getWorkersCount":
+      const workers = await queue.getWorkers();
+      respond(ws, msg.id, workers.length);
+      break;
     case "removeRepeatableByKey":
       await queue.removeRepeatableByKey(data.key);
       respond(ws, msg.id);

diff --git a/lib/responders/bullmq-responders.ts b/lib/responders/bullmq-responders.ts
@@ -101,10 +101,13 @@ async function respondQueueCommand(
     case "getCompletedCount":
     case "getFailedCount":
     case "getRepeatableCount":
-    case "getWorkersCount":
       const count = await (<any>queue)[data.cmd]();
       respond(ws, msg.id, count);
       break;
+    case "getWorkersCount":
+      const workers = await queue.getWorkers();
+      respond(ws, msg.id, workers.length);
+      break;
     case "removeRepeatableByKey":
       await queue.removeRepeatableByKey(data.key);
       respond(ws, msg.id);