perf: use a single map to store pool workers and their related data
[poolifier.git] / src / pools / selection-strategies / round-robin-worker-choice-strategy.ts
CommitLineData
ea7a90d3 1import type { IPoolWorker } from '../pool-worker'
bdaf31cd
JB
2import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy'
3
4/**
5 * Selects the next worker in a round robin fashion.
6 *
38e795c1
JB
7 * @typeParam Worker - Type of worker which manages the strategy.
8 * @typeParam Data - Type of data sent to the worker. This can only be serializable data.
9 * @typeParam Response - Type of response of execution. This can only be serializable data.
bdaf31cd
JB
10 */
11export class RoundRobinWorkerChoiceStrategy<
ea7a90d3 12 Worker extends IPoolWorker,
bdaf31cd
JB
13 Data,
14 Response
15> extends AbstractWorkerChoiceStrategy<Worker, Data, Response> {
16 /**
ffcbbad8 17 * Id of the next worker.
bdaf31cd 18 */
ffcbbad8 19 private nextWorkerId: number = 0
bdaf31cd 20
38e795c1 21 /** {@inheritDoc} */
a6f7f1b4 22 public reset (): boolean {
ffcbbad8 23 this.nextWorkerId = 0
ea7a90d3
JB
24 return true
25 }
26
38e795c1 27 /** {@inheritDoc} */
bdaf31cd 28 public choose (): Worker {
ffcbbad8
JB
29 const chosenWorker = this.pool.workers.get(this.nextWorkerId)
30 ?.worker as Worker
31 this.nextWorkerId =
32 this.nextWorkerId === this.pool.workers.size - 1
bdaf31cd 33 ? 0
ffcbbad8 34 : this.nextWorkerId + 1
bdaf31cd
JB
35 return chosenWorker
36 }
37}