X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=a9d8f685fec3432db4fc6829dabcfd84f146babe;hb=4f487526a63c873d168386250b40ad8103c5a4d8;hp=9939a04b9e497f5e78a84396493db7819f837508;hpb=6b27d40762317ec8502657663bdc839e358cda03;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 9939a04b..a9d8f685 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -2,11 +2,30 @@ import { MessageChannel, SHARE_ENV, Worker, + type WorkerOptions, isMainThread } from 'node:worker_threads' import type { Draft, MessageValue } from '../../utility-types' import { AbstractPool } from '../abstract-pool' -import { type PoolOptions, type PoolType, PoolTypes } from '../pool' +import { + type PoolOptions, + type PoolType, + PoolTypes, + type WorkerType, + WorkerTypes +} from '../pool' + +/** + * Options for a poolifier thread pool. + */ +export interface ThreadPoolOptions extends PoolOptions { + /** + * Worker options. + * + * @see https://nodejs.org/api/worker_threads.html#new-workerfilename-options + */ + workerOptions?: WorkerOptions +} /** * A thread worker with message channels for communication between main thread and thread worker. @@ -39,7 +58,7 @@ export class FixedThreadPool< public constructor ( numberOfThreads: number, filePath: string, - opts: PoolOptions = {} + protected readonly opts: ThreadPoolOptions = {} ) { super(numberOfThreads, filePath, opts) } @@ -76,7 +95,8 @@ export class FixedThreadPool< /** @inheritDoc */ protected createWorker (): ThreadWorkerWithMessageChannel { return new Worker(this.filePath, { - env: SHARE_ENV + env: SHARE_ENV, + ...this.opts.workerOptions }) } @@ -91,23 +111,23 @@ export class FixedThreadPool< } /** @inheritDoc */ - public get type (): PoolType { + protected get type (): PoolType { return PoolTypes.fixed } /** @inheritDoc */ - protected get minSize (): number { - return this.numberOfWorkers + protected get worker (): WorkerType { + return WorkerTypes.thread } /** @inheritDoc */ - protected get maxSize (): number { + protected get minSize (): number { return this.numberOfWorkers } /** @inheritDoc */ - protected get full (): boolean { - return this.workerNodes.length >= this.numberOfWorkers + protected get maxSize (): number { + return this.numberOfWorkers } /** @inheritDoc */