X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fcluster%2Ffixed.ts;h=c7c95c4b3dad537b39dfd44d37481178035fcb4c;hb=f6bc9f26d8a0246bbee14b2b03d0bcc41b8aeb52;hp=dbf1d470b7e7f36eb12dec0bba28c081fac60786;hpb=a038b5175071af7c51dae9e25c7423c7572f3949;p=poolifier.git diff --git a/src/pools/cluster/fixed.ts b/src/pools/cluster/fixed.ts index dbf1d470..c7c95c4b 100644 --- a/src/pools/cluster/fixed.ts +++ b/src/pools/cluster/fixed.ts @@ -60,33 +60,45 @@ export class FixedClusterPool< } /** @inheritDoc */ - protected destroyWorker (worker: Worker): void { - this.sendToWorker(worker, { kill: true, workerId: worker.id }) + protected async destroyWorkerNode (workerNodeKey: number): Promise { + this.flushTasksQueue(workerNodeKey) + // FIXME: wait for tasks to be finished + const worker = this.workerNodes[workerNodeKey].worker + const waitWorkerExit = new Promise((resolve) => { + worker.on('exit', () => { + resolve() + }) + }) worker.on('disconnect', () => { worker.kill() }) + await this.sendKillMessageToWorker(workerNodeKey, worker.id) worker.disconnect() + await waitWorkerExit } /** @inheritDoc */ - protected sendToWorker (worker: Worker, message: MessageValue): void { - worker.send(message) + protected sendToWorker ( + workerNodeKey: number, + message: MessageValue + ): void { + this.workerNodes[workerNodeKey].worker.send(message) } /** @inheritDoc */ - protected sendStartupMessageToWorker (worker: Worker): void { - this.sendToWorker(worker, { + protected sendStartupMessageToWorker (workerNodeKey: number): void { + this.sendToWorker(workerNodeKey, { ready: false, - workerId: worker.id + workerId: this.workerNodes[workerNodeKey].worker.id }) } /** @inheritDoc */ protected registerWorkerMessageListener( - worker: Worker, + workerNodeKey: number, listener: (message: MessageValue) => void ): void { - worker.on('message', listener) + this.workerNodes[workerNodeKey].worker.on('message', listener) } /** @inheritDoc */ @@ -104,16 +116,6 @@ export class FixedClusterPool< return WorkerTypes.cluster } - /** @inheritDoc */ - protected get minSize (): number { - return this.numberOfWorkers - } - - /** @inheritDoc */ - protected get maxSize (): number { - return this.numberOfWorkers - } - /** @inheritDoc */ protected get busy (): boolean { return this.internalBusy()