X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fabstract-pool.ts;h=dceef6decb534f083510d90a947a05861c2ec2df;hb=a1763c54c962c69b5e02a30c0909724986495fcd;hp=7ed6ab0351f3ad1cf44c6961b1273a2d78fc593f;hpb=ce9e97f17ed0d1b2928919244dce5a7887f843f5;p=poolifier.git diff --git a/src/pools/abstract-pool.ts b/src/pools/abstract-pool.ts index 7ed6ab03..dceef6de 100644 --- a/src/pools/abstract-pool.ts +++ b/src/pools/abstract-pool.ts @@ -118,7 +118,10 @@ export abstract class AbstractPool< this.executeTask = this.executeTask.bind(this) this.enqueueTask = this.enqueueTask.bind(this) this.dequeueTask = this.dequeueTask.bind(this) - this.checkAndEmitEvents = this.checkAndEmitEvents.bind(this) + this.checkAndEmitTaskExecutionEvents = + this.checkAndEmitTaskExecutionEvents.bind(this) + this.checkAndEmitTaskQueuingEvents = + this.checkAndEmitTaskQueuingEvents.bind(this) if (this.opts.enableEvents === true) { this.emitter = new PoolEmitter() @@ -368,6 +371,9 @@ export abstract class AbstractPool< 0 ) }), + ...(this.opts.enableTasksQueue === true && { + backPressure: this.hasBackPressure() + }), failedTasks: this.workerNodes.reduce( (accumulator, workerNode) => accumulator + workerNode.usage.tasks.failed, @@ -733,7 +739,6 @@ export abstract class AbstractPool< } else { this.enqueueTask(workerNodeKey, task) } - this.checkAndEmitEvents() }) } @@ -1221,7 +1226,7 @@ export abstract class AbstractPool< } } - private checkAndEmitEvents (): void { + private checkAndEmitTaskExecutionEvents (): void { if (this.emitter != null) { if (this.busy) { this.emitter.emit(PoolEvents.busy, this.info) @@ -1229,9 +1234,12 @@ export abstract class AbstractPool< if (this.type === PoolTypes.dynamic && this.full) { this.emitter.emit(PoolEvents.full, this.info) } - if (this.hasBackPressure()) { - this.emitter.emit(PoolEvents.backPressure, this.info) - } + } + } + + private checkAndEmitTaskQueuingEvents (): void { + if (this.hasBackPressure()) { + this.emitter?.emit(PoolEvents.backPressure, this.info) } } @@ -1296,7 +1304,7 @@ export abstract class AbstractPool< this.opts.enableTasksQueue === true && this.workerNodes.findIndex( (workerNode) => !workerNode.hasBackPressure() - ) !== -1 + ) === -1 ) } @@ -1309,10 +1317,13 @@ export abstract class AbstractPool< private executeTask (workerNodeKey: number, task: Task): void { this.beforeTaskExecutionHook(workerNodeKey, task) this.sendToWorker(workerNodeKey, task, task.transferList) + this.checkAndEmitTaskExecutionEvents() } private enqueueTask (workerNodeKey: number, task: Task): number { - return this.workerNodes[workerNodeKey].enqueueTask(task) + const tasksQueueSize = this.workerNodes[workerNodeKey].enqueueTask(task) + this.checkAndEmitTaskQueuingEvents() + return tasksQueueSize } private dequeueTask (workerNodeKey: number): Task | undefined {