Allow worker choice strategy to specify their statistics requirements

[poolifier.git] / src / pools / selection-strategies / worker-choice-strategy-context.ts

diff --git a/src/pools/selection-strategies/worker-choice-strategy-context.ts b/src/pools/selection-strategies/worker-choice-strategy-context.ts
index b84a8632ffc35a78a71742aa206274997b45eb6d..8290f89ed9a143a8420a45cb1c5965454bebe35e 100644 (file)
--- a/src/pools/selection-strategies/worker-choice-strategy-context.ts
+++ b/src/pools/selection-strategies/worker-choice-strategy-context.ts
@@ -60,6 +60,15 @@ export class WorkerChoiceStrategyContext<
     )
   }
 
+  /**
+   * Get the worker choice strategy used in the context.
+   *
+   * @returns The worker choice strategy.
+   */
+  public getWorkerChoiceStrategy (): IWorkerChoiceStrategy<Worker> {
+    return this.workerChoiceStrategy
+  }
+
   /**
    * Set the worker choice strategy to use in the context.
    *