X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fworker%2Fthread-worker.ts;h=bc143a1f914b7e7dba14b395ac456be40c59ed80;hb=de868be6cc9bcfb6d341ffb14e6407c24a1a0e17;hp=e208eb4063656dccf1ed25a8c12463c6de1302e5;hpb=29d8b961b1907797d251871a97092973d342c63c;p=poolifier.git diff --git a/src/worker/thread-worker.ts b/src/worker/thread-worker.ts index e208eb40..bc143a1f 100644 --- a/src/worker/thread-worker.ts +++ b/src/worker/thread-worker.ts @@ -30,7 +30,8 @@ export class ThreadWorker< /** * Message port used to communicate with the main worker. */ - private port!: MessagePort + private port?: MessagePort + /** * Constructs a new poolifier thread worker. * @@ -41,25 +42,29 @@ export class ThreadWorker< taskFunctions: TaskFunction | TaskFunctions, opts: WorkerOptions = {} ) { - super( - 'worker-thread-pool:poolifier', - isMainThread, - parentPort as MessagePort, - taskFunctions, - opts - ) + super(isMainThread, parentPort as MessagePort, taskFunctions, opts) } /** @inheritDoc */ protected handleReadyMessage (message: MessageValue): void { if ( message.workerId === this.id && - message.ready != null && + message.ready === false && message.port != null ) { - this.port = message.port - this.port.on('message', this.messageListener.bind(this)) - this.sendToMainWorker({ ready: true, workerId: this.id }) + try { + this.port = message.port + this.port.on('message', this.messageListener.bind(this)) + this.sendToMainWorker({ + ready: true, + taskFunctionNames: this.listTaskFunctionNames() + }) + } catch { + this.sendToMainWorker({ + ready: false, + taskFunctionNames: this.listTaskFunctionNames() + }) + } } } @@ -76,12 +81,17 @@ export class ThreadWorker< } /** @inheritDoc */ - protected sendToMainWorker (message: MessageValue): void { - this.port.postMessage(message) + protected readonly sendToMainWorker = ( + message: MessageValue + ): void => { + this.port?.postMessage({ ...message, workerId: this.id }) } - /** @inheritDoc */ - protected handleError (e: Error | string): string { - return e as string + /** + * @inheritDoc + * @override + */ + protected handleError (error: Error | string): string { + return error as string } }