-import type { AbstractPoolWorker } from '../abstract-pool-worker'
-
/**
* Enumeration of worker choice strategies.
*/
*/
ROUND_ROBIN: 'ROUND_ROBIN',
/**
- * Less recently used worker selection strategy.
+ * Less used worker selection strategy.
+ */
+ LESS_USED: 'LESS_USED',
+ /**
+ * Less busy worker selection strategy.
+ */
+ LESS_BUSY: 'LESS_BUSY',
+ /**
+ * Fair share worker selection strategy.
+ */
+ FAIR_SHARE: 'FAIR_SHARE',
+ /**
+ * Weighted round robin worker selection strategy.
*/
- LESS_RECENTLY_USED: 'LESS_RECENTLY_USED'
+ WEIGHTED_ROUND_ROBIN: 'WEIGHTED_ROUND_ROBIN'
} as const)
/**
export type WorkerChoiceStrategy = keyof typeof WorkerChoiceStrategies
/**
- * Worker choice strategy interface.
+ * Worker choice strategy options.
+ */
+export interface WorkerChoiceStrategyOptions {
+ /**
+ * Use tasks median run time instead of average run time.
+ *
+ * @defaultValue false
+ */
+ medRunTime?: boolean
+}
+
+/**
+ * Pool worker tasks usage statistics requirements.
*
- * @template Worker Type of worker which manages the strategy.
+ * @internal
*/
-export interface IWorkerChoiceStrategy<Worker extends AbstractPoolWorker> {
+export interface RequiredStatistics {
+ /**
+ * Require tasks run time.
+ */
+ runTime: boolean
+ /**
+ * Require tasks average run time.
+ */
+ avgRunTime: boolean
+ /**
+ * Require tasks median run time.
+ */
+ medRunTime: boolean
+}
+
+/**
+ * Worker choice strategy interface.
+ */
+export interface IWorkerChoiceStrategy {
+ /**
+ * Required tasks usage statistics.
+ */
+ readonly requiredStatistics: RequiredStatistics
+ /**
+ * Resets strategy internals (counters, statistics, etc.).
+ */
+ reset: () => boolean
+ /**
+ * Chooses a worker node in the pool and returns its key.
+ */
+ choose: () => number
/**
- * Is the pool attached to the strategy dynamic?.
+ * Removes a worker node key from strategy internals.
+ *
+ * @param workerNodeKey - The worker node key.
*/
- isDynamicPool: boolean
+ remove: (workerNodeKey: number) => boolean
/**
- * Choose a worker in the pool.
+ * Sets the worker choice strategy options.
+ *
+ * @param opts - The worker choice strategy options.
*/
- choose(): Worker
+ setOptions: (opts: WorkerChoiceStrategyOptions) => void
}