Merge branch 'master' into elu-strategy
[poolifier.git] / src / pools / abstract-pool.ts
1 import crypto from 'node:crypto'
2 import { performance } from 'node:perf_hooks'
3 import type { MessageValue, PromiseResponseWrapper } from '../utility-types'
4 import {
5 DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS,
6 EMPTY_FUNCTION,
7 isPlainObject,
8 median
9 } from '../utils'
10 import { KillBehaviors, isKillBehavior } from '../worker/worker-options'
11 import { CircularArray } from '../circular-array'
12 import { Queue } from '../queue'
13 import {
14 type IPool,
15 PoolEmitter,
16 PoolEvents,
17 type PoolInfo,
18 type PoolOptions,
19 type PoolType,
20 PoolTypes,
21 type TasksQueueOptions,
22 type WorkerType
23 } from './pool'
24 import type {
25 IWorker,
26 Task,
27 TaskStatistics,
28 WorkerNode,
29 WorkerUsage
30 } from './worker'
31 import {
32 WorkerChoiceStrategies,
33 type WorkerChoiceStrategy,
34 type WorkerChoiceStrategyOptions
35 } from './selection-strategies/selection-strategies-types'
36 import { WorkerChoiceStrategyContext } from './selection-strategies/worker-choice-strategy-context'
37
38 /**
39 * Base class that implements some shared logic for all poolifier pools.
40 *
41 * @typeParam Worker - Type of worker which manages this pool.
42 * @typeParam Data - Type of data sent to the worker. This can only be serializable data.
43 * @typeParam Response - Type of execution response. This can only be serializable data.
44 */
45 export abstract class AbstractPool<
46 Worker extends IWorker,
47 Data = unknown,
48 Response = unknown
49 > implements IPool<Worker, Data, Response> {
50 /** @inheritDoc */
51 public readonly workerNodes: Array<WorkerNode<Worker, Data>> = []
52
53 /** @inheritDoc */
54 public readonly emitter?: PoolEmitter
55
56 /**
57 * The execution response promise map.
58 *
59 * - `key`: The message id of each submitted task.
60 * - `value`: An object that contains the worker, the execution response promise resolve and reject callbacks.
61 *
62 * When we receive a message from the worker, we get a map entry with the promise resolve/reject bound to the message id.
63 */
64 protected promiseResponseMap: Map<
65 string,
66 PromiseResponseWrapper<Worker, Response>
67 > = new Map<string, PromiseResponseWrapper<Worker, Response>>()
68
69 /**
70 * Worker choice strategy context referencing a worker choice algorithm implementation.
71 *
72 * Default to a round robin algorithm.
73 */
74 protected workerChoiceStrategyContext: WorkerChoiceStrategyContext<
75 Worker,
76 Data,
77 Response
78 >
79
80 /**
81 * Constructs a new poolifier pool.
82 *
83 * @param numberOfWorkers - Number of workers that this pool should manage.
84 * @param filePath - Path to the worker file.
85 * @param opts - Options for the pool.
86 */
87 public constructor (
88 protected readonly numberOfWorkers: number,
89 protected readonly filePath: string,
90 protected readonly opts: PoolOptions<Worker>
91 ) {
92 if (!this.isMain()) {
93 throw new Error('Cannot start a pool from a worker!')
94 }
95 this.checkNumberOfWorkers(this.numberOfWorkers)
96 this.checkFilePath(this.filePath)
97 this.checkPoolOptions(this.opts)
98
99 this.chooseWorkerNode = this.chooseWorkerNode.bind(this)
100 this.executeTask = this.executeTask.bind(this)
101 this.enqueueTask = this.enqueueTask.bind(this)
102 this.checkAndEmitEvents = this.checkAndEmitEvents.bind(this)
103
104 if (this.opts.enableEvents === true) {
105 this.emitter = new PoolEmitter()
106 }
107 this.workerChoiceStrategyContext = new WorkerChoiceStrategyContext<
108 Worker,
109 Data,
110 Response
111 >(
112 this,
113 this.opts.workerChoiceStrategy,
114 this.opts.workerChoiceStrategyOptions
115 )
116
117 this.setupHook()
118
119 for (let i = 1; i <= this.numberOfWorkers; i++) {
120 this.createAndSetupWorker()
121 }
122 }
123
124 private checkFilePath (filePath: string): void {
125 if (
126 filePath == null ||
127 (typeof filePath === 'string' && filePath.trim().length === 0)
128 ) {
129 throw new Error('Please specify a file with a worker implementation')
130 }
131 }
132
133 private checkNumberOfWorkers (numberOfWorkers: number): void {
134 if (numberOfWorkers == null) {
135 throw new Error(
136 'Cannot instantiate a pool without specifying the number of workers'
137 )
138 } else if (!Number.isSafeInteger(numberOfWorkers)) {
139 throw new TypeError(
140 'Cannot instantiate a pool with a non safe integer number of workers'
141 )
142 } else if (numberOfWorkers < 0) {
143 throw new RangeError(
144 'Cannot instantiate a pool with a negative number of workers'
145 )
146 } else if (this.type === PoolTypes.fixed && numberOfWorkers === 0) {
147 throw new Error('Cannot instantiate a fixed pool with no worker')
148 }
149 }
150
151 private checkPoolOptions (opts: PoolOptions<Worker>): void {
152 if (isPlainObject(opts)) {
153 this.opts.workerChoiceStrategy =
154 opts.workerChoiceStrategy ?? WorkerChoiceStrategies.ROUND_ROBIN
155 this.checkValidWorkerChoiceStrategy(this.opts.workerChoiceStrategy)
156 this.opts.workerChoiceStrategyOptions =
157 opts.workerChoiceStrategyOptions ??
158 DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
159 this.checkValidWorkerChoiceStrategyOptions(
160 this.opts.workerChoiceStrategyOptions
161 )
162 this.opts.restartWorkerOnError = opts.restartWorkerOnError ?? true
163 this.opts.enableEvents = opts.enableEvents ?? true
164 this.opts.enableTasksQueue = opts.enableTasksQueue ?? false
165 if (this.opts.enableTasksQueue) {
166 this.checkValidTasksQueueOptions(
167 opts.tasksQueueOptions as TasksQueueOptions
168 )
169 this.opts.tasksQueueOptions = this.buildTasksQueueOptions(
170 opts.tasksQueueOptions as TasksQueueOptions
171 )
172 }
173 } else {
174 throw new TypeError('Invalid pool options: must be a plain object')
175 }
176 }
177
178 private checkValidWorkerChoiceStrategy (
179 workerChoiceStrategy: WorkerChoiceStrategy
180 ): void {
181 if (!Object.values(WorkerChoiceStrategies).includes(workerChoiceStrategy)) {
182 throw new Error(
183 `Invalid worker choice strategy '${workerChoiceStrategy}'`
184 )
185 }
186 }
187
188 private checkValidWorkerChoiceStrategyOptions (
189 workerChoiceStrategyOptions: WorkerChoiceStrategyOptions
190 ): void {
191 if (!isPlainObject(workerChoiceStrategyOptions)) {
192 throw new TypeError(
193 'Invalid worker choice strategy options: must be a plain object'
194 )
195 }
196 if (
197 workerChoiceStrategyOptions.weights != null &&
198 Object.keys(workerChoiceStrategyOptions.weights).length !== this.maxSize
199 ) {
200 throw new Error(
201 'Invalid worker choice strategy options: must have a weight for each worker node'
202 )
203 }
204 }
205
206 private checkValidTasksQueueOptions (
207 tasksQueueOptions: TasksQueueOptions
208 ): void {
209 if (tasksQueueOptions != null && !isPlainObject(tasksQueueOptions)) {
210 throw new TypeError('Invalid tasks queue options: must be a plain object')
211 }
212 if ((tasksQueueOptions?.concurrency as number) <= 0) {
213 throw new Error(
214 `Invalid worker tasks concurrency '${
215 tasksQueueOptions.concurrency as number
216 }'`
217 )
218 }
219 }
220
221 /** @inheritDoc */
222 public get info (): PoolInfo {
223 return {
224 type: this.type,
225 worker: this.worker,
226 minSize: this.minSize,
227 maxSize: this.maxSize,
228 workerNodes: this.workerNodes.length,
229 idleWorkerNodes: this.workerNodes.reduce(
230 (accumulator, workerNode) =>
231 workerNode.workerUsage.tasks.executing === 0
232 ? accumulator + 1
233 : accumulator,
234 0
235 ),
236 busyWorkerNodes: this.workerNodes.reduce(
237 (accumulator, workerNode) =>
238 workerNode.workerUsage.tasks.executing > 0
239 ? accumulator + 1
240 : accumulator,
241 0
242 ),
243 executedTasks: this.workerNodes.reduce(
244 (accumulator, workerNode) =>
245 accumulator + workerNode.workerUsage.tasks.executed,
246 0
247 ),
248 executingTasks: this.workerNodes.reduce(
249 (accumulator, workerNode) =>
250 accumulator + workerNode.workerUsage.tasks.executing,
251 0
252 ),
253 queuedTasks: this.workerNodes.reduce(
254 (accumulator, workerNode) => accumulator + workerNode.tasksQueue.size,
255 0
256 ),
257 maxQueuedTasks: this.workerNodes.reduce(
258 (accumulator, workerNode) =>
259 accumulator + workerNode.tasksQueue.maxSize,
260 0
261 ),
262 failedTasks: this.workerNodes.reduce(
263 (accumulator, workerNode) =>
264 accumulator + workerNode.workerUsage.tasks.failed,
265 0
266 )
267 }
268 }
269
270 /**
271 * Pool type.
272 *
273 * If it is `'dynamic'`, it provides the `max` property.
274 */
275 protected abstract get type (): PoolType
276
277 /**
278 * Gets the worker type.
279 */
280 protected abstract get worker (): WorkerType
281
282 /**
283 * Pool minimum size.
284 */
285 protected abstract get minSize (): number
286
287 /**
288 * Pool maximum size.
289 */
290 protected abstract get maxSize (): number
291
292 /**
293 * Gets the given worker its worker node key.
294 *
295 * @param worker - The worker.
296 * @returns The worker node key if the worker is found in the pool worker nodes, `-1` otherwise.
297 */
298 private getWorkerNodeKey (worker: Worker): number {
299 return this.workerNodes.findIndex(
300 workerNode => workerNode.worker === worker
301 )
302 }
303
304 /** @inheritDoc */
305 public setWorkerChoiceStrategy (
306 workerChoiceStrategy: WorkerChoiceStrategy,
307 workerChoiceStrategyOptions?: WorkerChoiceStrategyOptions
308 ): void {
309 this.checkValidWorkerChoiceStrategy(workerChoiceStrategy)
310 this.opts.workerChoiceStrategy = workerChoiceStrategy
311 this.workerChoiceStrategyContext.setWorkerChoiceStrategy(
312 this.opts.workerChoiceStrategy
313 )
314 if (workerChoiceStrategyOptions != null) {
315 this.setWorkerChoiceStrategyOptions(workerChoiceStrategyOptions)
316 }
317 for (const workerNode of this.workerNodes) {
318 this.setWorkerNodeTasksUsage(
319 workerNode,
320 this.getWorkerUsage(workerNode.worker)
321 )
322 this.setWorkerStatistics(workerNode.worker)
323 }
324 }
325
326 /** @inheritDoc */
327 public setWorkerChoiceStrategyOptions (
328 workerChoiceStrategyOptions: WorkerChoiceStrategyOptions
329 ): void {
330 this.checkValidWorkerChoiceStrategyOptions(workerChoiceStrategyOptions)
331 this.opts.workerChoiceStrategyOptions = workerChoiceStrategyOptions
332 this.workerChoiceStrategyContext.setOptions(
333 this.opts.workerChoiceStrategyOptions
334 )
335 }
336
337 /** @inheritDoc */
338 public enableTasksQueue (
339 enable: boolean,
340 tasksQueueOptions?: TasksQueueOptions
341 ): void {
342 if (this.opts.enableTasksQueue === true && !enable) {
343 this.flushTasksQueues()
344 }
345 this.opts.enableTasksQueue = enable
346 this.setTasksQueueOptions(tasksQueueOptions as TasksQueueOptions)
347 }
348
349 /** @inheritDoc */
350 public setTasksQueueOptions (tasksQueueOptions: TasksQueueOptions): void {
351 if (this.opts.enableTasksQueue === true) {
352 this.checkValidTasksQueueOptions(tasksQueueOptions)
353 this.opts.tasksQueueOptions =
354 this.buildTasksQueueOptions(tasksQueueOptions)
355 } else if (this.opts.tasksQueueOptions != null) {
356 delete this.opts.tasksQueueOptions
357 }
358 }
359
360 private buildTasksQueueOptions (
361 tasksQueueOptions: TasksQueueOptions
362 ): TasksQueueOptions {
363 return {
364 concurrency: tasksQueueOptions?.concurrency ?? 1
365 }
366 }
367
368 /**
369 * Whether the pool is full or not.
370 *
371 * The pool filling boolean status.
372 */
373 protected get full (): boolean {
374 return this.workerNodes.length >= this.maxSize
375 }
376
377 /**
378 * Whether the pool is busy or not.
379 *
380 * The pool busyness boolean status.
381 */
382 protected abstract get busy (): boolean
383
384 protected internalBusy (): boolean {
385 return (
386 this.workerNodes.findIndex(workerNode => {
387 return workerNode.workerUsage.tasks.executing === 0
388 }) === -1
389 )
390 }
391
392 /** @inheritDoc */
393 public async execute (data?: Data, name?: string): Promise<Response> {
394 const timestamp = performance.now()
395 const workerNodeKey = this.chooseWorkerNode()
396 const submittedTask: Task<Data> = {
397 name,
398 // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
399 data: data ?? ({} as Data),
400 timestamp,
401 id: crypto.randomUUID()
402 }
403 const res = new Promise<Response>((resolve, reject) => {
404 this.promiseResponseMap.set(submittedTask.id as string, {
405 resolve,
406 reject,
407 worker: this.workerNodes[workerNodeKey].worker
408 })
409 })
410 if (
411 this.opts.enableTasksQueue === true &&
412 (this.busy ||
413 this.workerNodes[workerNodeKey].workerUsage.tasks.executing >=
414 ((this.opts.tasksQueueOptions as TasksQueueOptions)
415 .concurrency as number))
416 ) {
417 this.enqueueTask(workerNodeKey, submittedTask)
418 } else {
419 this.executeTask(workerNodeKey, submittedTask)
420 }
421 this.workerChoiceStrategyContext.update(workerNodeKey)
422 this.checkAndEmitEvents()
423 // eslint-disable-next-line @typescript-eslint/return-await
424 return res
425 }
426
427 /** @inheritDoc */
428 public async destroy (): Promise<void> {
429 await Promise.all(
430 this.workerNodes.map(async (workerNode, workerNodeKey) => {
431 this.flushTasksQueue(workerNodeKey)
432 // FIXME: wait for tasks to be finished
433 await this.destroyWorker(workerNode.worker)
434 })
435 )
436 }
437
438 /**
439 * Shutdowns the given worker.
440 *
441 * @param worker - A worker within `workerNodes`.
442 */
443 protected abstract destroyWorker (worker: Worker): void | Promise<void>
444
445 /**
446 * Setup hook to execute code before worker node are created in the abstract constructor.
447 * Can be overridden
448 *
449 * @virtual
450 */
451 protected setupHook (): void {
452 // Intentionally empty
453 }
454
455 /**
456 * Should return whether the worker is the main worker or not.
457 */
458 protected abstract isMain (): boolean
459
460 /**
461 * Hook executed before the worker task execution.
462 * Can be overridden.
463 *
464 * @param workerNodeKey - The worker node key.
465 */
466 protected beforeTaskExecutionHook (workerNodeKey: number): void {
467 ++this.workerNodes[workerNodeKey].workerUsage.tasks.executing
468 }
469
470 /**
471 * Hook executed after the worker task execution.
472 * Can be overridden.
473 *
474 * @param worker - The worker.
475 * @param message - The received message.
476 */
477 protected afterTaskExecutionHook (
478 worker: Worker,
479 message: MessageValue<Response>
480 ): void {
481 const workerUsage =
482 this.workerNodes[this.getWorkerNodeKey(worker)].workerUsage
483 const workerTaskStatistics = workerUsage.tasks
484 --workerTaskStatistics.executing
485 ++workerTaskStatistics.executed
486 if (message.taskError != null) {
487 ++workerTaskStatistics.failed
488 }
489
490 this.updateRunTimeWorkerUsage(workerUsage, message)
491 this.updateWaitTimeWorkerUsage(workerUsage, message)
492 this.updateEluWorkerUsage(workerUsage, message)
493 }
494
495 private updateRunTimeWorkerUsage (
496 workerUsage: WorkerUsage,
497 message: MessageValue<Response>
498 ): void {
499 if (
500 this.workerChoiceStrategyContext.getTaskStatisticsRequirements().runTime
501 ) {
502 workerUsage.runTime.aggregation += message.taskPerformance?.runTime ?? 0
503 if (
504 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
505 .avgRunTime &&
506 workerUsage.tasks.executed !== 0
507 ) {
508 workerUsage.runTime.average =
509 workerUsage.runTime.aggregation / workerUsage.tasks.executed
510 }
511 if (
512 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
513 .medRunTime &&
514 message.taskPerformance?.runTime != null
515 ) {
516 workerUsage.runTime.history.push(message.taskPerformance.runTime)
517 workerUsage.runTime.median = median(workerUsage.runTime.history)
518 }
519 }
520 }
521
522 private updateWaitTimeWorkerUsage (
523 workerUsage: WorkerUsage,
524 message: MessageValue<Response>
525 ): void {
526 if (
527 this.workerChoiceStrategyContext.getTaskStatisticsRequirements().waitTime
528 ) {
529 workerUsage.waitTime.aggregation += message.taskPerformance?.waitTime ?? 0
530 if (
531 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
532 .avgWaitTime &&
533 workerUsage.tasks.executed !== 0
534 ) {
535 workerUsage.waitTime.average =
536 workerUsage.waitTime.aggregation / workerUsage.tasks.executed
537 }
538 if (
539 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
540 .medWaitTime &&
541 message.taskPerformance?.waitTime != null
542 ) {
543 workerUsage.waitTime.history.push(message.taskPerformance.waitTime)
544 workerUsage.waitTime.median = median(workerUsage.waitTime.history)
545 }
546 }
547 }
548
549 private updateEluWorkerUsage (
550 workerTasksUsage: WorkerUsage,
551 message: MessageValue<Response>
552 ): void {
553 if (this.workerChoiceStrategyContext.getTaskStatisticsRequirements().elu) {
554 if (
555 workerTasksUsage.elu != null &&
556 message.taskPerformance?.elu != null
557 ) {
558 workerTasksUsage.elu = {
559 idle: workerTasksUsage.elu.idle + message.taskPerformance.elu.idle,
560 active:
561 workerTasksUsage.elu.active + message.taskPerformance.elu.active,
562 utilization:
563 (workerTasksUsage.elu.utilization +
564 message.taskPerformance.elu.utilization) /
565 2
566 }
567 } else if (message.taskPerformance?.elu != null) {
568 workerTasksUsage.elu = message.taskPerformance.elu
569 }
570 }
571 }
572
573 /**
574 * Chooses a worker node for the next task.
575 *
576 * The default worker choice strategy uses a round robin algorithm to distribute the load.
577 *
578 * @returns The worker node key
579 */
580 protected chooseWorkerNode (): number {
581 let workerNodeKey: number
582 if (this.type === PoolTypes.dynamic && !this.full && this.internalBusy()) {
583 const workerCreated = this.createAndSetupWorker()
584 this.registerWorkerMessageListener(workerCreated, message => {
585 const currentWorkerNodeKey = this.getWorkerNodeKey(workerCreated)
586 if (
587 isKillBehavior(KillBehaviors.HARD, message.kill) ||
588 (message.kill != null &&
589 this.workerNodes[currentWorkerNodeKey].workerUsage.tasks
590 .executing === 0)
591 ) {
592 // Kill message received from the worker: no new tasks are submitted to that worker for a while ( > maxInactiveTime)
593 this.flushTasksQueue(currentWorkerNodeKey)
594 // FIXME: wait for tasks to be finished
595 void (this.destroyWorker(workerCreated) as Promise<void>)
596 }
597 })
598 workerNodeKey = this.getWorkerNodeKey(workerCreated)
599 } else {
600 workerNodeKey = this.workerChoiceStrategyContext.execute()
601 }
602 return workerNodeKey
603 }
604
605 /**
606 * Sends a message to the given worker.
607 *
608 * @param worker - The worker which should receive the message.
609 * @param message - The message.
610 */
611 protected abstract sendToWorker (
612 worker: Worker,
613 message: MessageValue<Data>
614 ): void
615
616 /**
617 * Registers a listener callback on the given worker.
618 *
619 * @param worker - The worker which should register a listener.
620 * @param listener - The message listener callback.
621 */
622 protected abstract registerWorkerMessageListener<
623 Message extends Data | Response
624 >(worker: Worker, listener: (message: MessageValue<Message>) => void): void
625
626 /**
627 * Returns a newly created worker.
628 */
629 protected abstract createWorker (): Worker
630
631 /**
632 * Function that can be hooked up when a worker has been newly created and moved to the pool worker nodes.
633 *
634 * Can be used to update the `maxListeners` or binding the `main-worker`\<-\>`worker` connection if not bind by default.
635 *
636 * @param worker - The newly created worker.
637 */
638 protected abstract afterWorkerSetup (worker: Worker): void
639
640 /**
641 * Creates a new worker and sets it up completely in the pool worker nodes.
642 *
643 * @returns New, completely set up worker.
644 */
645 protected createAndSetupWorker (): Worker {
646 const worker = this.createWorker()
647
648 worker.on('message', this.opts.messageHandler ?? EMPTY_FUNCTION)
649 worker.on('error', this.opts.errorHandler ?? EMPTY_FUNCTION)
650 worker.on('error', error => {
651 if (this.emitter != null) {
652 this.emitter.emit(PoolEvents.error, error)
653 }
654 })
655 worker.on('error', () => {
656 if (this.opts.restartWorkerOnError === true) {
657 this.createAndSetupWorker()
658 }
659 })
660 worker.on('online', this.opts.onlineHandler ?? EMPTY_FUNCTION)
661 worker.on('exit', this.opts.exitHandler ?? EMPTY_FUNCTION)
662 worker.once('exit', () => {
663 this.removeWorkerNode(worker)
664 })
665
666 this.pushWorkerNode(worker)
667
668 this.setWorkerStatistics(worker)
669
670 this.afterWorkerSetup(worker)
671
672 return worker
673 }
674
675 /**
676 * This function is the listener registered for each worker message.
677 *
678 * @returns The listener function to execute when a message is received from a worker.
679 */
680 protected workerListener (): (message: MessageValue<Response>) => void {
681 return message => {
682 if (message.id != null) {
683 // Task execution response received
684 const promiseResponse = this.promiseResponseMap.get(message.id)
685 if (promiseResponse != null) {
686 if (message.taskError != null) {
687 promiseResponse.reject(message.taskError.message)
688 if (this.emitter != null) {
689 this.emitter.emit(PoolEvents.taskError, message.taskError)
690 }
691 } else {
692 promiseResponse.resolve(message.data as Response)
693 }
694 this.afterTaskExecutionHook(promiseResponse.worker, message)
695 this.promiseResponseMap.delete(message.id)
696 const workerNodeKey = this.getWorkerNodeKey(promiseResponse.worker)
697 if (
698 this.opts.enableTasksQueue === true &&
699 this.tasksQueueSize(workerNodeKey) > 0
700 ) {
701 this.executeTask(
702 workerNodeKey,
703 this.dequeueTask(workerNodeKey) as Task<Data>
704 )
705 }
706 }
707 }
708 }
709 }
710
711 private checkAndEmitEvents (): void {
712 if (this.emitter != null) {
713 if (this.busy) {
714 this.emitter?.emit(PoolEvents.busy, this.info)
715 }
716 if (this.type === PoolTypes.dynamic && this.full) {
717 this.emitter?.emit(PoolEvents.full, this.info)
718 }
719 }
720 }
721
722 /**
723 * Sets the given worker node its tasks usage in the pool.
724 *
725 * @param workerNode - The worker node.
726 * @param workerUsage - The worker usage.
727 */
728 private setWorkerNodeTasksUsage (
729 workerNode: WorkerNode<Worker, Data>,
730 workerUsage: WorkerUsage
731 ): void {
732 workerNode.workerUsage = workerUsage
733 }
734
735 /**
736 * Pushes the given worker in the pool worker nodes.
737 *
738 * @param worker - The worker.
739 * @returns The worker nodes length.
740 */
741 private pushWorkerNode (worker: Worker): number {
742 return this.workerNodes.push({
743 worker,
744 workerUsage: this.getWorkerUsage(worker),
745 tasksQueue: new Queue<Task<Data>>()
746 })
747 }
748
749 // /**
750 // * Sets the given worker in the pool worker nodes.
751 // *
752 // * @param workerNodeKey - The worker node key.
753 // * @param worker - The worker.
754 // * @param workerUsage - The worker usage.
755 // * @param tasksQueue - The worker task queue.
756 // */
757 // private setWorkerNode (
758 // workerNodeKey: number,
759 // worker: Worker,
760 // workerUsage: WorkerUsage,
761 // tasksQueue: Queue<Task<Data>>
762 // ): void {
763 // this.workerNodes[workerNodeKey] = {
764 // worker,
765 // workerUsage,
766 // tasksQueue
767 // }
768 // }
769
770 /**
771 * Removes the given worker from the pool worker nodes.
772 *
773 * @param worker - The worker.
774 */
775 private removeWorkerNode (worker: Worker): void {
776 const workerNodeKey = this.getWorkerNodeKey(worker)
777 if (workerNodeKey !== -1) {
778 this.workerNodes.splice(workerNodeKey, 1)
779 this.workerChoiceStrategyContext.remove(workerNodeKey)
780 }
781 }
782
783 private executeTask (workerNodeKey: number, task: Task<Data>): void {
784 this.beforeTaskExecutionHook(workerNodeKey)
785 this.sendToWorker(this.workerNodes[workerNodeKey].worker, task)
786 }
787
788 private enqueueTask (workerNodeKey: number, task: Task<Data>): number {
789 return this.workerNodes[workerNodeKey].tasksQueue.enqueue(task)
790 }
791
792 private dequeueTask (workerNodeKey: number): Task<Data> | undefined {
793 return this.workerNodes[workerNodeKey].tasksQueue.dequeue()
794 }
795
796 private tasksQueueSize (workerNodeKey: number): number {
797 return this.workerNodes[workerNodeKey].tasksQueue.size
798 }
799
800 private flushTasksQueue (workerNodeKey: number): void {
801 if (this.tasksQueueSize(workerNodeKey) > 0) {
802 for (let i = 0; i < this.tasksQueueSize(workerNodeKey); i++) {
803 this.executeTask(
804 workerNodeKey,
805 this.dequeueTask(workerNodeKey) as Task<Data>
806 )
807 }
808 }
809 }
810
811 private flushTasksQueues (): void {
812 for (const [workerNodeKey] of this.workerNodes.entries()) {
813 this.flushTasksQueue(workerNodeKey)
814 }
815 }
816
817 private setWorkerStatistics (worker: Worker): void {
818 this.sendToWorker(worker, {
819 statistics: {
820 runTime:
821 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
822 .runTime,
823 waitTime:
824 this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
825 .waitTime,
826 elu: this.workerChoiceStrategyContext.getTaskStatisticsRequirements()
827 .elu
828 }
829 })
830 }
831
832 private getWorkerUsage (worker: Worker): WorkerUsage {
833 return {
834 tasks: this.getTaskStatistics(this, worker),
835 runTime: {
836 aggregation: 0,
837 average: 0,
838 median: 0,
839 history: new CircularArray()
840 },
841 waitTime: {
842 aggregation: 0,
843 average: 0,
844 median: 0,
845 history: new CircularArray()
846 },
847 elu: undefined
848 }
849 }
850
851 private getTaskStatistics (
852 self: AbstractPool<Worker, Data, Response>,
853 worker: Worker
854 ): TaskStatistics {
855 return {
856 executed: 0,
857 executing: 0,
858 get queued (): number {
859 return self.tasksQueueSize(self.getWorkerNodeKey(worker))
860 },
861 failed: 0
862 }
863 }
864 }