X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Fworker-choice-strategy-context.ts;h=2b3c9fb68b33b1d4925f9dab7be3cf14b123bf88;hb=refs%2Ftags%2Fv2.4.12;hp=2a487acbdfde0f2d07c2f9eae9233d26da387604;hpb=7254e41972596ec68a8790747b9fd4ef0def5afc;p=poolifier.git

diff --git a/src/pools/selection-strategies/worker-choice-strategy-context.ts b/src/pools/selection-strategies/worker-choice-strategy-context.ts
index 2a487acb..2b3c9fb6 100644
--- a/src/pools/selection-strategies/worker-choice-strategy-context.ts
+++ b/src/pools/selection-strategies/worker-choice-strategy-context.ts
@@ -114,6 +114,19 @@ export class WorkerChoiceStrategyContext<
     this.workerChoiceStrategies.get(this.workerChoiceStrategy)?.reset()
   }
 
+  /**
+   * Updates the worker node key in the worker choice strategy internals in the context.
+   *
+   * @returns `true` if the update is successful, `false` otherwise.
+   */
+  public update (workerNodeKey: number): boolean {
+    return (
+      this.workerChoiceStrategies.get(
+        this.workerChoiceStrategy
+      ) as IWorkerChoiceStrategy
+    ).update(workerNodeKey)
+  }
+
   /**
    * Executes the worker choice strategy algorithm in the context.
    *
@@ -128,7 +141,7 @@ export class WorkerChoiceStrategyContext<
   }
 
   /**
-   * Removes a worker node key from the worker choice strategy in the context.
+   * Removes the worker node key from the worker choice strategy in the context.
    *
    * @param workerNodeKey - The key of the worker node.
    * @returns `true` if the removal is successful, `false` otherwise.