X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fcluster%2Ffixed.ts;h=b74f034bc199d9d4e89f77f0856b00855cf1685a;hb=fa548cda5120ac0708d82f37cd0ce1260d7f96c1;hp=9470cccd857321388d70e0bf7f09de1a20505cfe;hpb=2c93d3cf3183abd425f1578b66fc3b38fae85349;p=poolifier.git diff --git a/src/pools/cluster/fixed.ts b/src/pools/cluster/fixed.ts index 9470cccd..b74f034b 100644 --- a/src/pools/cluster/fixed.ts +++ b/src/pools/cluster/fixed.ts @@ -61,19 +61,21 @@ export class FixedClusterPool< /** @inheritDoc */ protected async destroyWorkerNode (workerNodeKey: number): Promise { + this.flagWorkerNodeAsNotReady(workerNodeKey) this.flushTasksQueue(workerNodeKey) // FIXME: wait for tasks to be finished const workerNode = this.workerNodes[workerNodeKey] const worker = workerNode.worker const waitWorkerExit = new Promise(resolve => { - worker.on('exit', () => { + worker.once('exit', () => { resolve() }) }) - worker.on('disconnect', () => { + worker.once('disconnect', () => { worker.kill() }) await this.sendKillMessageToWorker(workerNodeKey) + workerNode.removeAllListeners() worker.disconnect() await waitWorkerExit } @@ -85,7 +87,7 @@ export class FixedClusterPool< ): void { this.workerNodes[workerNodeKey].worker.send({ ...message, - workerId: this.workerNodes[workerNodeKey].info.id as number + workerId: this.getWorkerInfo(workerNodeKey).id as number }) } @@ -104,6 +106,22 @@ export class FixedClusterPool< this.workerNodes[workerNodeKey].worker.on('message', listener) } + /** @inheritDoc */ + protected registerOnceWorkerMessageListener( + workerNodeKey: number, + listener: (message: MessageValue) => void + ): void { + this.workerNodes[workerNodeKey].worker.once('message', listener) + } + + /** @inheritDoc */ + protected deregisterWorkerMessageListener( + workerNodeKey: number, + listener: (message: MessageValue) => void + ): void { + this.workerNodes[workerNodeKey].worker.off('message', listener) + } + /** @inheritDoc */ protected createWorker (): Worker { return cluster.fork(this.opts.env)