X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fworker%2Fcluster-worker.ts;h=d972692ec6eb4f182479cac008c47a1f20b75d8d;hb=8b3fccaac505706915540210b9d123e1a12e79ab;hp=b22c50fe4160a17ac83b1a525df8090ac1ae9538;hpb=90d7d101196cf9702ccc1d220dd33cca67a427b0;p=poolifier.git diff --git a/src/worker/cluster-worker.ts b/src/worker/cluster-worker.ts index b22c50fe..d972692e 100644 --- a/src/worker/cluster-worker.ts +++ b/src/worker/cluster-worker.ts @@ -32,21 +32,24 @@ export class ClusterWorker< taskFunctions: TaskFunction | TaskFunctions, opts: WorkerOptions = {} ) { - super( - 'worker-cluster-pool:poolifier', - cluster.isPrimary, - cluster.worker as Worker, - taskFunctions, - opts - ) + super(cluster.isPrimary, cluster.worker as Worker, taskFunctions, opts) } /** @inheritDoc */ protected handleReadyMessage (message: MessageValue): void { - if (message.workerId === this.id && message.ready != null) { - this.getMainWorker()?.on('message', this.messageListener.bind(this)) - this.sendTaskFunctionsListToMainWorker() - this.sendToMainWorker({ ready: true, workerId: this.id }) + if (message.workerId === this.id && message.ready === false) { + try { + this.getMainWorker().on('message', this.messageListener.bind(this)) + this.sendToMainWorker({ + ready: true, + taskFunctionNames: this.listTaskFunctionNames() + }) + } catch { + this.sendToMainWorker({ + ready: false, + taskFunctionNames: this.listTaskFunctionNames() + }) + } } } @@ -56,7 +59,9 @@ export class ClusterWorker< } /** @inheritDoc */ - protected sendToMainWorker (message: MessageValue): void { - this.getMainWorker().send(message) + protected readonly sendToMainWorker = ( + message: MessageValue + ): void => { + this.getMainWorker().send({ ...message, workerId: this.id }) } }