X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=db278485a097fcb40a1846902ee8a7b36b548c3e;hb=refs%2Ftags%2Fv2.7.4;hp=2e708d973891be533e0d66f632ee5e4d983fed7a;hpb=7d91a8cd5c644cf409ec7a174d72ec84fcc3e4f7;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 2e708d97..db278485 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -63,11 +63,11 @@ export class FixedThreadPool< const workerNode = this.workerNodes[workerNodeKey] const worker = workerNode.worker const waitWorkerExit = new Promise(resolve => { - worker.on('exit', () => { + worker.once('exit', () => { resolve() }) }) - this.sendToWorker(workerNodeKey, { kill: true, workerId: worker.threadId }) + await this.sendKillMessageToWorker(workerNodeKey) workerNode.closeChannel() await worker.terminate() await waitWorkerExit @@ -80,20 +80,22 @@ export class FixedThreadPool< transferList?: TransferListItem[] ): void { ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel - ).port1.postMessage(message, transferList) + this.workerNodes[workerNodeKey].messageChannel as MessageChannel + ).port1.postMessage( + { ...message, workerId: this.workerNodes[workerNodeKey].info.id }, + transferList + ) } /** @inheritDoc */ protected sendStartupMessageToWorker (workerNodeKey: number): void { - const worker = this.workerNodes[workerNodeKey].worker - const port2: MessagePort = ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel - ).port2 - worker.postMessage( + const workerNode = this.workerNodes[workerNodeKey] + const port2: MessagePort = (workerNode.messageChannel as MessageChannel) + .port2 + workerNode.worker.postMessage( { ready: false, - workerId: worker.threadId, + workerId: workerNode.info.id, port: port2 }, [port2] @@ -106,10 +108,30 @@ export class FixedThreadPool< listener: (message: MessageValue) => void ): void { ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel + this.workerNodes[workerNodeKey].messageChannel as MessageChannel ).port1.on('message', listener) } + /** @inheritDoc */ + protected registerOnceWorkerMessageListener( + workerNodeKey: number, + listener: (message: MessageValue) => void + ): void { + ( + this.workerNodes[workerNodeKey].messageChannel as MessageChannel + ).port1.once('message', listener) + } + + /** @inheritDoc */ + protected deregisterWorkerMessageListener( + workerNodeKey: number, + listener: (message: MessageValue) => void + ): void { + ( + this.workerNodes[workerNodeKey].messageChannel as MessageChannel + ).port1.off('message', listener) + } + /** @inheritDoc */ protected createWorker (): Worker { return new Worker(this.filePath, { @@ -128,16 +150,6 @@ export class FixedThreadPool< return WorkerTypes.thread } - /** @inheritDoc */ - protected get minSize (): number { - return this.numberOfWorkers - } - - /** @inheritDoc */ - protected get maxSize (): number { - return this.numberOfWorkers - } - /** @inheritDoc */ protected get busy (): boolean { return this.internalBusy()