type MessageChannel,
type MessagePort,
SHARE_ENV,
+ type TransferListItem,
Worker,
type WorkerOptions,
isMainThread
}
/** @inheritDoc */
- protected async destroyWorker (worker: Worker): Promise<void> {
- this.sendToWorker(worker, { kill: true, workerId: worker.threadId })
- const workerInfo = this.getWorkerInfoByWorker(worker)
- workerInfo.messageChannel?.port1.close()
- workerInfo.messageChannel?.port2.close()
+ protected async destroyWorkerNode (workerNodeKey: number): Promise<void> {
+ this.flushTasksQueue(workerNodeKey)
+ // FIXME: wait for tasks to be finished
+ const workerNode = this.workerNodes[workerNodeKey]
+ const worker = workerNode.worker
+ const waitWorkerExit = new Promise<void>(resolve => {
+ worker.on('exit', () => {
+ resolve()
+ })
+ })
+ this.sendToWorker(workerNodeKey, { kill: true, workerId: worker.threadId })
+ workerNode.closeChannel()
await worker.terminate()
+ await waitWorkerExit
}
/** @inheritDoc */
- protected sendToWorker (worker: Worker, message: MessageValue<Data>): void {
+ protected sendToWorker (
+ workerNodeKey: number,
+ message: MessageValue<Data>,
+ transferList?: TransferListItem[]
+ ): void {
(
- this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel
- ).port1.postMessage(message)
+ this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel
+ ).port1.postMessage(message, transferList)
}
/** @inheritDoc */
- protected sendStartupMessageToWorker (worker: Worker): void {
+ protected sendStartupMessageToWorker (workerNodeKey: number): void {
+ const worker = this.workerNodes[workerNodeKey].worker
const port2: MessagePort = (
- this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel
+ this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel
).port2
worker.postMessage(
{
ready: false,
- workerId: this.getWorkerInfoByWorker(worker).id as number,
+ workerId: worker.threadId,
port: port2
},
[port2]
/** @inheritDoc */
protected registerWorkerMessageListener<Message extends Data | Response>(
- worker: Worker,
+ workerNodeKey: number,
listener: (message: MessageValue<Message>) => void
): void {
(
- this.getWorkerInfoByWorker(worker).messageChannel as MessageChannel
+ this.getWorkerInfo(workerNodeKey).messageChannel as MessageChannel
).port1.on('message', listener)
}