WorkerChoiceStrategy
} from './selection-strategies-types'
import { WorkerChoiceStrategies } from './selection-strategies-types'
-import { SelectionStrategiesUtils } from './selection-strategies-utils'
+import { getWorkerChoiceStrategy } from './selection-strategies-utils'
/**
* Selects the next worker for dynamic pool.
*
- * @template Worker Type of worker which manages the strategy.
- * @template Data Type of data sent to the worker. This can only be serializable data.
- * @template Response Type of response of execution. This can only be serializable data.
+ * @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.
*/
export class DynamicPoolWorkerChoiceStrategy<
- Worker extends IPoolWorker,
- Data,
- Response
-> extends AbstractWorkerChoiceStrategy<Worker, Data, Response> {
- private workerChoiceStrategy: IWorkerChoiceStrategy<Worker>
+ Worker extends IPoolWorker,
+ Data,
+ Response
+ >
+ extends AbstractWorkerChoiceStrategy<Worker, Data, Response>
+ implements IWorkerChoiceStrategy {
+ private readonly workerChoiceStrategy: IWorkerChoiceStrategy
/**
* Constructs a worker choice strategy for dynamic pool.
*
- * @param pool The pool instance.
- * @param createDynamicallyWorkerCallback The worker creation callback for dynamic pool.
- * @param workerChoiceStrategy The worker choice strategy when the pull is busy.
+ * @param pool - The pool instance.
+ * @param createWorkerCallback - The worker creation callback for dynamic pool.
+ * @param workerChoiceStrategy - The worker choice strategy when the pool is busy.
*/
public constructor (
pool: IPoolInternal<Worker, Data, Response>,
- private createDynamicallyWorkerCallback: () => Worker,
+ private readonly createWorkerCallback: () => number,
workerChoiceStrategy: WorkerChoiceStrategy = WorkerChoiceStrategies.ROUND_ROBIN
) {
super(pool)
- this.workerChoiceStrategy = SelectionStrategiesUtils.getWorkerChoiceStrategy(
+ this.workerChoiceStrategy = getWorkerChoiceStrategy(
this.pool,
workerChoiceStrategy
)
this.requiredStatistics = this.workerChoiceStrategy.requiredStatistics
}
- /** @inheritDoc */
+ /** {@inheritDoc} */
public reset (): boolean {
return this.workerChoiceStrategy.reset()
}
- /** @inheritDoc */
- public choose (): Worker {
- const freeWorker = this.pool.findFreeWorker()
- if (freeWorker) {
- return freeWorker
+ /** {@inheritDoc} */
+ public choose (): number {
+ const freeWorkerKey = this.pool.findFreeWorkerKey()
+ if (freeWorkerKey !== -1) {
+ return freeWorkerKey
}
- if (this.pool.busy === true) {
+ if (this.pool.busy) {
return this.workerChoiceStrategy.choose()
}
// All workers are busy, create a new worker
- return this.createDynamicallyWorkerCallback()
+ return this.createWorkerCallback()
+ }
+
+ /** {@inheritDoc} */
+ public remove (workerKey: number): boolean {
+ return this.workerChoiceStrategy.remove(workerKey)
}
}