X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=7977cd61abd5263023b855fe9f7f23ba5135de10;hb=b3b1857ba11fd35d0f7118ef8046c38676473a15;hp=197830279be020e45741fb00930bb82ffcc12ea3;hpb=07e0c9e591f9fa6715ba94e52c647b7ee3d2b9c7;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 19783027..7977cd61 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -1,14 +1,18 @@ import { - type MessageChannel, - type MessagePort, + isMainThread, type TransferListItem, - type Worker, - isMainThread + type Worker } from 'node:worker_threads' -import type { MessageValue } from '../../utility-types' -import { AbstractPool } from '../abstract-pool' -import { type PoolOptions, type PoolType, PoolTypes } from '../pool' -import { type WorkerType, WorkerTypes } from '../worker' + +import type { MessageValue } from '../../utility-types.js' +import { AbstractPool } from '../abstract-pool.js' +import { type PoolOptions, type PoolType, PoolTypes } from '../pool.js' +import { type WorkerType, WorkerTypes } from '../worker.js' + +/** + * Options for a poolifier thread pool. + */ +export type ThreadPoolOptions = PoolOptions /** * A thread pool with a fixed number of threads. @@ -32,9 +36,10 @@ export class FixedThreadPool< public constructor ( numberOfThreads: number, filePath: string, - protected readonly opts: PoolOptions = {} + opts: ThreadPoolOptions = {}, + maximumNumberOfThreads?: number ) { - super(numberOfThreads, filePath, opts) + super(numberOfThreads, filePath, opts, maximumNumberOfThreads) } /** @inheritDoc */ @@ -46,10 +51,13 @@ export class FixedThreadPool< protected sendToWorker ( workerNodeKey: number, message: MessageValue, - transferList?: TransferListItem[] + transferList?: readonly TransferListItem[] ): void { - this.workerNodes[workerNodeKey].messageChannel?.port1?.postMessage( - { ...message, workerId: this.getWorkerInfo(workerNodeKey).id }, + this.workerNodes[workerNodeKey]?.messageChannel?.port1.postMessage( + { + ...message, + workerId: this.getWorkerInfo(workerNodeKey)?.id + } satisfies MessageValue, transferList ) } @@ -57,14 +65,14 @@ export class FixedThreadPool< /** @inheritDoc */ protected sendStartupMessageToWorker (workerNodeKey: number): void { const workerNode = this.workerNodes[workerNodeKey] - const port2: MessagePort = (workerNode.messageChannel as MessageChannel) - .port2 + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + const port2 = workerNode.messageChannel!.port2 workerNode.worker.postMessage( { ready: false, - workerId: this.getWorkerInfo(workerNodeKey).id, + workerId: this.getWorkerInfo(workerNodeKey)?.id, port: port2 - }, + } satisfies MessageValue, [port2] ) } @@ -74,7 +82,7 @@ export class FixedThreadPool< workerNodeKey: number, listener: (message: MessageValue) => void ): void { - this.workerNodes[workerNodeKey].messageChannel?.port1?.on( + this.workerNodes[workerNodeKey].messageChannel?.port1.on( 'message', listener ) @@ -85,7 +93,7 @@ export class FixedThreadPool< workerNodeKey: number, listener: (message: MessageValue) => void ): void { - this.workerNodes[workerNodeKey].messageChannel?.port1?.once( + this.workerNodes[workerNodeKey].messageChannel?.port1.once( 'message', listener ) @@ -96,12 +104,22 @@ export class FixedThreadPool< workerNodeKey: number, listener: (message: MessageValue) => void ): void { - this.workerNodes[workerNodeKey].messageChannel?.port1?.off( + this.workerNodes[workerNodeKey].messageChannel?.port1.off( 'message', listener ) } + /** @inheritDoc */ + protected shallCreateDynamicWorker (): boolean { + return false + } + + /** @inheritDoc */ + protected checkAndEmitDynamicWorkerCreationEvents (): void { + /* noop */ + } + /** @inheritDoc */ protected get type (): PoolType { return PoolTypes.fixed