fix: handle properly small or zero tasksStealingRatio

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
poolifier · Aug 30, 2024 · ab6e570 · ab6e570
1 parent 74970f2
commit ab6e570
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pools/abstract-pool.ts b/src/pools/abstract-pool.ts
@@ -1848,8 +1848,9 @@ export abstract class AbstractPool<
     if (
       !workerNode.info.continuousStealing &&
       (this.cannotStealTask() ||
+        this.opts.tasksQueueOptions?.tasksStealingRatio === 0 ||
         (this.info.stealingWorkerNodes ?? 0) >
-          Math.round(
+          Math.ceil(
             this.workerNodes.length *
               this.opts.tasksQueueOptions!.tasksStealingRatio!,
           ))
@@ -1913,8 +1914,9 @@ export abstract class AbstractPool<
     if (
       this.cannotStealTask() ||
       this.backPressure ||
+      this.opts.tasksQueueOptions?.tasksStealingRatio === 0 ||
       (this.info.stealingWorkerNodes ?? 0) >
-        Math.round(
+        Math.ceil(
           this.workerNodes.length *
             this.opts.tasksQueueOptions!.tasksStealingRatio!,
         )