X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=bef82826350d772c146656d4676f942923d33165;hb=f313c91822165ee5ddf81417df4cad1106f3a4a5;hp=c9145f571186a9d5783531236414aff7eadd463e;hpb=84d0f4f2937987e5adbb8cfa94839eaf050c7502;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index c9145f57..bef82826 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -1,4 +1,6 @@ import { + type MessageChannel, + type MessagePort, SHARE_ENV, Worker, type WorkerOptions, @@ -56,12 +58,40 @@ export class FixedThreadPool< /** @inheritDoc */ protected async destroyWorker (worker: Worker): Promise { this.sendToWorker(worker, { kill: true, workerId: worker.threadId }) + this.workerNodes[this.getWorkerNodeKey(worker)].closeChannel() await worker.terminate() } /** @inheritDoc */ protected sendToWorker (worker: Worker, message: MessageValue): void { - worker.postMessage(message) + ( + this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel + ).port1.postMessage(message) + } + + /** @inheritDoc */ + protected sendStartupMessageToWorker (worker: Worker): void { + const port2: MessagePort = ( + this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel + ).port2 + worker.postMessage( + { + ready: false, + workerId: worker.threadId, + port: port2 + }, + [port2] + ) + } + + /** @inheritDoc */ + protected registerWorkerMessageListener( + worker: Worker, + listener: (message: MessageValue) => void + ): void { + ( + this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel + ).port1.on('message', listener) } /** @inheritDoc */