X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fpools%2Fthread%2Ffixed.ts;h=e8be44f2e3b5b02671f44ce6739df7a87b61273c;hb=a8884ffdad6658f854f4048d768f7e56cf8d5032;hp=1c91caa752702fefcccecb88074875ea4a27031c;hpb=be0676b3936d75f22ce55b0f71a1fb03d008a01c;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index 1c91caa7..e8be44f2 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -1,7 +1,8 @@ import { isMainThread, MessageChannel, SHARE_ENV, Worker } from 'worker_threads' import type { Draft, MessageValue } from '../../utility-types' -import type { PoolOptions } from '../abstract-pool' import { AbstractPool } from '../abstract-pool' +import type { PoolOptions } from '../pool' +import { PoolType } from '../pool-internal' /** * A thread worker with message channels for communication between main thread and thread worker. @@ -17,7 +18,6 @@ export type ThreadWorkerWithMessageChannel = Worker & Draft * * @template Data Type of data sent to the worker. This can only be serializable data. * @template Response Type of response of execution. This can only be serializable data. - * * @author [Alessandro Pio Ardizio](https://github.com/pioardi) * @since 0.0.1 */ @@ -30,21 +30,22 @@ export class FixedThreadPool< * * @param numberOfThreads Number of threads for this pool. * @param filePath Path to an implementation of a `ThreadWorker` file, which can be relative or absolute. - * @param opts Options for this fixed thread pool. Default: `{ maxTasks: 1000 }` + * @param opts Options for this fixed thread pool. */ public constructor ( numberOfThreads: number, filePath: string, - opts: PoolOptions = { maxTasks: 1000 } + opts: PoolOptions = {} ) { super(numberOfThreads, filePath, opts) } + /** @inheritDoc */ protected isMain (): boolean { return isMainThread } - /** @inheritdoc */ + /** @inheritDoc */ public async destroyWorker ( worker: ThreadWorkerWithMessageChannel ): Promise { @@ -52,6 +53,7 @@ export class FixedThreadPool< await worker.terminate() } + /** @inheritDoc */ protected sendToWorker ( worker: ThreadWorkerWithMessageChannel, message: MessageValue @@ -59,7 +61,7 @@ export class FixedThreadPool< worker.postMessage(message) } - /** @inheritdoc */ + /** @inheritDoc */ public registerWorkerMessageListener ( messageChannel: ThreadWorkerWithMessageChannel, listener: (message: MessageValue) => void @@ -67,20 +69,30 @@ export class FixedThreadPool< messageChannel.port2?.on('message', listener) } + /** @inheritDoc */ protected createWorker (): ThreadWorkerWithMessageChannel { return new Worker(this.filePath, { env: SHARE_ENV }) } + /** @inheritDoc */ protected afterWorkerSetup (worker: ThreadWorkerWithMessageChannel): void { const { port1, port2 } = new MessageChannel() worker.postMessage({ parent: port1 }, [port1]) worker.port1 = port1 worker.port2 = port2 - // We will attach a listener for every task, - // when the task is completed the listener will be removed but to avoid warnings we are increasing the max listeners size. - worker.port2.setMaxListeners(this.opts.maxTasks ?? 1000) + // Listen to worker messages. this.registerWorkerMessageListener(worker, super.workerListener()) } + + /** @inheritDoc */ + public get type (): PoolType { + return PoolType.FIXED + } + + /** @inheritDoc */ + public get busy (): boolean { + return this.internalGetBusyStatus() + } }