X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=e3a94b329d4cfcee6f05cd6e2f8ad06e555a6dfc;hb=dc02fc29818293c82727cb43ad5014e30aa6a64d;hp=c9145f571186a9d5783531236414aff7eadd463e;hpb=21f710aa73abbb5d90328cfb199adfc0f7a70406;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index c9145f57..e3a94b32 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,42 @@ export class FixedThreadPool< /** @inheritDoc */ protected async destroyWorker (worker: Worker): Promise { this.sendToWorker(worker, { kill: true, workerId: worker.threadId }) + const workerInfo = this.getWorkerInfoByWorker(worker) + workerInfo.messageChannel?.port1.close() + workerInfo.messageChannel?.port2.close() 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 */