import {
type MessageChannel,
type MessagePort,
- SHARE_ENV,
type TransferListItem,
- Worker,
- type WorkerOptions,
+ type Worker,
isMainThread
} from 'node:worker_threads'
import type { MessageValue } from '../../utility-types'
import { type PoolOptions, type PoolType, PoolTypes } from '../pool'
import { type WorkerType, WorkerTypes } from '../worker'
-/**
- * Options for a poolifier thread pool.
- */
-export interface ThreadPoolOptions extends PoolOptions<Worker> {
- /**
- * Worker options.
- *
- * @see https://nodejs.org/api/worker_threads.html#new-workerfilename-options
- */
- workerOptions?: WorkerOptions
-}
-
/**
* A thread pool with a fixed number of threads.
*
public constructor (
numberOfThreads: number,
filePath: string,
- protected readonly opts: ThreadPoolOptions = {}
+ opts: PoolOptions<Worker> = {},
+ maximumNumberOfThreads?: number
) {
- super(numberOfThreads, filePath, opts)
+ super(numberOfThreads, filePath, opts, maximumNumberOfThreads)
}
/** @inheritDoc */
return isMainThread
}
- /** @inheritDoc */
- 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()
- })
- })
- await this.sendKillMessageToWorker(workerNodeKey)
- workerNode.closeChannel()
- await worker.terminate()
- await waitWorkerExit
- }
-
/** @inheritDoc */
protected sendToWorker (
workerNodeKey: number,
message: MessageValue<Data>,
transferList?: TransferListItem[]
): void {
- (
- this.workerNodes[workerNodeKey].messageChannel as MessageChannel
- ).port1.postMessage(
- { ...message, workerId: this.workerNodes[workerNodeKey].info.id },
+ this.workerNodes[workerNodeKey].messageChannel?.port1?.postMessage(
+ { ...message, workerId: this.getWorkerInfo(workerNodeKey).id },
transferList
)
}
workerNode.worker.postMessage(
{
ready: false,
- workerId: workerNode.info.id,
+ workerId: this.getWorkerInfo(workerNodeKey).id,
port: port2
},
[port2]
workerNodeKey: number,
listener: (message: MessageValue<Message>) => void
): void {
- (
- this.workerNodes[workerNodeKey].messageChannel as MessageChannel
- ).port1.on('message', listener)
+ this.workerNodes[workerNodeKey].messageChannel?.port1?.on(
+ 'message',
+ listener
+ )
}
/** @inheritDoc */
- protected createWorker (): Worker {
- return new Worker(this.filePath, {
- env: SHARE_ENV,
- ...this.opts.workerOptions
- })
+ protected registerOnceWorkerMessageListener<Message extends Data | Response>(
+ workerNodeKey: number,
+ listener: (message: MessageValue<Message>) => void
+ ): void {
+ this.workerNodes[workerNodeKey].messageChannel?.port1?.once(
+ 'message',
+ listener
+ )
+ }
+
+ /** @inheritDoc */
+ protected deregisterWorkerMessageListener<Message extends Data | Response>(
+ workerNodeKey: number,
+ listener: (message: MessageValue<Message>) => void
+ ): void {
+ this.workerNodes[workerNodeKey].messageChannel?.port1?.off(
+ 'message',
+ listener
+ )
}
/** @inheritDoc */