X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Fselection-strategies-types.ts;h=137c1be7562e50309818763f556651aed2034b04;hb=6533c3e6b6a95daade8d20577d6b816512e2aeeb;hp=a05fd9b784c9da682cd0e270f09b81ce41325d70;hpb=c923ce5670eeae4194aa996d44a1071e88cb21ad;p=poolifier.git

diff --git a/src/pools/selection-strategies/selection-strategies-types.ts b/src/pools/selection-strategies/selection-strategies-types.ts
index a05fd9b7..137c1be7 100644
--- a/src/pools/selection-strategies/selection-strategies-types.ts
+++ b/src/pools/selection-strategies/selection-strategies-types.ts
@@ -1,3 +1,6 @@
+import type { IPoolInternal } from '../pool-internal'
+import type { IPoolWorker } from '../pool-worker'
+
 /**
  * Enumeration of worker choice strategies.
  */
@@ -30,18 +33,27 @@ export const WorkerChoiceStrategies = Object.freeze({
 export type WorkerChoiceStrategy = keyof typeof WorkerChoiceStrategies
 
 /**
- * Pool tasks usage statistics requirements.
+ * Pool worker tasks usage statistics requirements.
  */
 export interface RequiredStatistics {
   runTime: boolean
+  avgRunTime: boolean
 }
 
 /**
  * Worker choice strategy interface.
  */
-export interface IWorkerChoiceStrategy {
+export interface IWorkerChoiceStrategy<
+  Worker extends IPoolWorker,
+  Data = unknown,
+  Response = unknown
+> {
+  /**
+   * The pool instance.
+   */
+  readonly pool: IPoolInternal<Worker, Data, Response>
   /**
-   * Is the pool attached to the strategy dynamic?.
+   * Is the pool bound to the strategy dynamic?.
    */
   readonly isDynamicPool: boolean
   /**
@@ -56,4 +68,10 @@ export interface IWorkerChoiceStrategy {
    * Chooses a worker in the pool and returns its key.
    */
   choose: () => number
+  /**
+   * Removes a worker reference from strategy internals.
+   *
+   * @param workerKey - The worker key.
+   */
+  remove: (workerKey: number) => boolean
 }