X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=2ad94a7bf16083072bf1303745834c2b7c06506a;hb=e4c07d066abc51e978a18f44a973a548f24fb7ad;hp=32107e48635314e25269cc37e2c0ad7fdd3fcd75;hpb=c38a523624d95c0329fac3d0748883c361db9d93;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 32107e48..2ad94a7b 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -62,12 +62,12 @@ 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() }) }) - await this.sendKillMessageToWorker(workerNodeKey, worker.threadId) + await this.sendKillMessageToWorker(workerNodeKey) workerNode.closeChannel() await worker.terminate() await waitWorkerExit @@ -81,19 +81,21 @@ export class FixedThreadPool< ): void { ( this.workerNodes[workerNodeKey].messageChannel as MessageChannel - ).port1.postMessage(message, transferList) + ).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.workerNodes[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]