X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fworker-node.ts;h=09e4a7b5cd8831360365a7a88e338e3b921eaf91;hb=46e29227a97fe04f886969199f1c14338af50543;hp=64880fcb9820c098e37fb103bd27b8563073833a;hpb=671d515455c745dc74f4c385fe23683975bfc3df;p=poolifier.git diff --git a/src/pools/worker-node.ts b/src/pools/worker-node.ts index 64880fcb..09e4a7b5 100644 --- a/src/pools/worker-node.ts +++ b/src/pools/worker-node.ts @@ -28,9 +28,9 @@ implements IWorkerNode { public messageChannel?: MessageChannel /** @inheritdoc */ public usage: WorkerUsage - private readonly tasksUsage: Map + private readonly taskFunctionsUsage: Map private readonly tasksQueue: Queue> - private readonly tasksQueueBackPressureMaxSize: number + private readonly tasksQueueBackPressureSize: number /** * Constructs a new worker node. @@ -40,15 +40,31 @@ implements IWorkerNode { * @param poolMaxSize - The pool maximum size. */ constructor (worker: Worker, workerType: WorkerType, poolMaxSize: number) { + if (worker == null) { + throw new Error('Cannot construct a worker node without a worker') + } + if (workerType == null) { + throw new Error('Cannot construct a worker node without a worker type') + } + if (poolMaxSize == null) { + throw new Error( + 'Cannot construct a worker node without a pool maximum size' + ) + } + if (isNaN(poolMaxSize)) { + throw new Error( + 'Cannot construct a worker node with a NaN pool maximum size' + ) + } this.worker = worker this.info = this.initWorkerInfo(worker, workerType) if (workerType === WorkerTypes.thread) { this.messageChannel = new MessageChannel() } this.usage = this.initWorkerUsage() - this.tasksUsage = new Map() + this.taskFunctionsUsage = new Map() this.tasksQueue = new Queue>() - this.tasksQueueBackPressureMaxSize = Math.pow(poolMaxSize, 2) + this.tasksQueueBackPressureSize = Math.pow(poolMaxSize, 2) } /** @inheritdoc */ @@ -82,13 +98,13 @@ implements IWorkerNode { /** @inheritdoc */ public hasBackPressure (): boolean { - return this.tasksQueueSize() >= this.tasksQueueBackPressureMaxSize + return this.tasksQueue.size >= this.tasksQueueBackPressureSize } /** @inheritdoc */ public resetUsage (): void { this.usage = this.initWorkerUsage() - this.tasksUsage.clear() + this.taskFunctionsUsage.clear() } /** @inheritdoc */ @@ -103,23 +119,27 @@ implements IWorkerNode { } /** @inheritdoc */ - public getTaskWorkerUsage (name: string): WorkerUsage | undefined { + public getTaskFunctionWorkerUsage (name: string): WorkerUsage | undefined { if (!Array.isArray(this.info.taskFunctions)) { throw new Error( - `Cannot get task worker usage for task function name '${name}' when task function names list is not yet defined` + `Cannot get task function worker usage for task function name '${name}' when task function names list is not yet defined` ) } if ( - name === DEFAULT_TASK_NAME && Array.isArray(this.info.taskFunctions) && - this.info.taskFunctions.length > 1 + this.info.taskFunctions.length < 3 ) { + throw new Error( + `Cannot get task function worker usage for task function name '${name}' when task function names list has less than 3 elements` + ) + } + if (name === DEFAULT_TASK_NAME) { name = this.info.taskFunctions[1] } - if (!this.tasksUsage.has(name)) { - this.tasksUsage.set(name, this.initTaskWorkerUsage(name)) + if (!this.taskFunctionsUsage.has(name)) { + this.taskFunctionsUsage.set(name, this.initTaskFunctionWorkerUsage(name)) } - return this.tasksUsage.get(name) + return this.taskFunctionsUsage.get(name) } private initWorkerInfo (worker: Worker, workerType: WorkerType): WorkerInfo { @@ -167,7 +187,7 @@ implements IWorkerNode { } } - private initTaskWorkerUsage (name: string): WorkerUsage { + private initTaskFunctionWorkerUsage (name: string): WorkerUsage { const getTaskQueueSize = (): number => { let taskQueueSize = 0 for (const task of this.tasksQueue) {