X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=d69055cf0909efa3b9cf45983163e0be6a35c513;hb=9fe8fd698590c2494dc6793cfd8c08026fe88a31;hp=7199baf8fdf5ad52bc43bcfdfe107de84719589a;hpb=c2301b8e5ab9a84e16dd037c6f7e3e85030bb266;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 7199baf8..d69055cf 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -2,6 +2,7 @@ import { type MessageChannel, type MessagePort, SHARE_ENV, + type TransferListItem, Worker, type WorkerOptions, isMainThread @@ -61,12 +62,15 @@ export class FixedThreadPool< // FIXME: wait for tasks to be finished const workerNode = this.workerNodes[workerNodeKey] const worker = workerNode.worker - const waitWorkerExit = new Promise(resolve => { + const waitWorkerExit = new Promise((resolve) => { worker.on('exit', () => { resolve() }) }) - this.sendToWorker(workerNodeKey, { kill: true, workerId: worker.threadId }) + await this.sendKillMessageToWorker( + workerNodeKey, + workerNode.info.id as number + ) workerNode.closeChannel() await worker.terminate() await waitWorkerExit @@ -75,23 +79,24 @@ export class FixedThreadPool< /** @inheritDoc */ protected sendToWorker ( workerNodeKey: number, - message: MessageValue + message: MessageValue, + transferList?: TransferListItem[] ): void { ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel - ).port1.postMessage(message) + this.workerNodes[workerNodeKey].messageChannel as MessageChannel + ).port1.postMessage(message, transferList) } /** @inheritDoc */ protected sendStartupMessageToWorker (workerNodeKey: number): void { - const worker = this.workerNodes[workerNodeKey].worker - const port2: MessagePort = ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel - ).port2 + const workerNode = this.workerNodes[workerNodeKey] + const worker = workerNode.worker + const port2: MessagePort = (workerNode.messageChannel as MessageChannel) + .port2 worker.postMessage( { ready: false, - workerId: worker.threadId, + workerId: workerNode.info.id, port: port2 }, [port2] @@ -104,7 +109,7 @@ export class FixedThreadPool< listener: (message: MessageValue) => void ): void { ( - this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel + this.workerNodes[workerNodeKey].messageChannel as MessageChannel ).port1.on('message', listener) } @@ -126,16 +131,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()