+import { DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS } from '../../utils'
+import type { IPool } from '../pool'
import type { IWorker } from '../worker'
import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy'
import type {
IWorkerChoiceStrategy,
- RequiredStatistics
+ RequiredStatistics,
+ WorkerChoiceStrategyOptions
} from './selection-strategies-types'
/**
*
* @typeParam Worker - Type of worker which manages the strategy.
* @typeParam Data - Type of data sent to the worker. This can only be serializable data.
- * @typeParam Response - Type of response of execution. This can only be serializable data.
+ * @typeParam Response - Type of execution response. This can only be serializable data.
*/
export class FairShareWorkerChoiceStrategy<
Worker extends IWorker,
WorkerVirtualTaskTimestamp
> = new Map<number, WorkerVirtualTaskTimestamp>()
+ /** @inheritDoc */
+ public constructor (
+ pool: IPool<Worker, Data, Response>,
+ opts: WorkerChoiceStrategyOptions = DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
+ ) {
+ super(pool, opts)
+ this.checkOptions(this.opts)
+ }
+
/** @inheritDoc */
public reset (): boolean {
this.workerLastVirtualTaskTimestamp.clear()
/** @inheritDoc */
public remove (workerNodeKey: number): boolean {
const deleted = this.workerLastVirtualTaskTimestamp.delete(workerNodeKey)
- for (const [key, value] of this.workerLastVirtualTaskTimestamp.entries()) {
+ for (const [key, value] of this.workerLastVirtualTaskTimestamp) {
if (key > workerNodeKey) {
this.workerLastVirtualTaskTimestamp.set(key - 1, value)
}
performance.now(),
this.workerLastVirtualTaskTimestamp.get(workerNodeKey)?.end ?? -Infinity
)
+ const workerVirtualTaskTRunTime = this.requiredStatistics.medRunTime
+ ? this.pool.workerNodes[workerNodeKey].tasksUsage.medRunTime
+ : this.pool.workerNodes[workerNodeKey].tasksUsage.avgRunTime
this.workerLastVirtualTaskTimestamp.set(workerNodeKey, {
start: workerVirtualTaskStartTimestamp,
- end:
- workerVirtualTaskStartTimestamp +
- (this.pool.workerNodes[workerNodeKey].tasksUsage.avgRunTime ?? 0)
+ end: workerVirtualTaskStartTimestamp + (workerVirtualTaskTRunTime ?? 0)
})
}
}