+import type { IPoolInternal } from '../pool-internal'
+import type { IPoolWorker } from '../pool-worker'
+
/**
* Enumeration of worker choice strategies.
*/
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
+ */
+ readonly pool: IPoolInternal<Worker, Data, Response>
/**
* Is the pool attached to the strategy dynamic?.
+ * @readonly
*/
readonly isDynamicPool: boolean
/**
* Required pool tasks usage statistics.
+ * @readonly
*/
readonly requiredStatistics: RequiredStatistics
/**
* 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
}