X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fthread%2Ffixed.ts;h=a61c10b2346c8a44c4094605dfe6aa25787b8f09;hb=deb85c12b77faf6974551cefcd9676e62a392086;hp=bed8f3a7dba8ae87ed6085df040723a93eb0e739;hpb=c97c7edb14ea0699cd82bce5d0ffe50ae26af667;p=poolifier.git diff --git a/src/pools/thread/fixed.ts b/src/pools/thread/fixed.ts index bed8f3a7..a61c10b2 100644 --- a/src/pools/thread/fixed.ts +++ b/src/pools/thread/fixed.ts @@ -3,33 +3,41 @@ import type { Draft, MessageValue } from '../../utility-types' import type { PoolOptions } from '../abstract-pool' import { AbstractPool } from '../abstract-pool' +/** + * A thread worker with message channels for communication between main thread and thread worker. + */ export type ThreadWorkerWithMessageChannel = Worker & Draft /** - * A thread pool with a static number of threads, is possible to execute tasks in sync or async mode as you prefer. + * A thread pool with a fixed number of threads. + * + * It is possible to perform tasks in sync or asynchronous mode as you prefer. * - * This pool will select the worker thread in a round robin fashion. + * This pool selects the threads in a round robin fashion. + * + * @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 */ -// eslint-disable-next-line @typescript-eslint/no-explicit-any -export class FixedThreadPool extends AbstractPool< - ThreadWorkerWithMessageChannel, - Data, - Response -> { +export class FixedThreadPool< + Data = unknown, + Response = unknown +> extends AbstractPool { /** - * @param numThreads Num of threads for this worker pool. - * @param filePath A file path with implementation of `ThreadWorker` class, relative path is fine. - * @param opts An object with possible options for example `errorHandler`, `onlineHandler`. Default: `{ maxTasks: 1000 }` + * Constructs a new poolifier fixed thread pool. + * + * @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 }` */ public constructor ( - numThreads: number, + numberOfThreads: number, filePath: string, opts: PoolOptions = { maxTasks: 1000 } ) { - super(numThreads, filePath, opts) + super(numberOfThreads, filePath, opts) } protected isMain (): boolean { @@ -39,6 +47,7 @@ export class FixedThreadPool extends AbstractPool< protected async destroyWorker ( worker: ThreadWorkerWithMessageChannel ): Promise { + this.sendToWorker(worker, { kill: 1 }) await worker.terminate() } @@ -49,29 +58,27 @@ export class FixedThreadPool extends AbstractPool< worker.postMessage(message) } - protected registerWorkerMessageListener ( - port: ThreadWorkerWithMessageChannel, - listener: (message: MessageValue) => void + protected registerWorkerMessageListener ( + messageChannel: ThreadWorkerWithMessageChannel, + listener: (message: MessageValue) => void ): void { - port.port2?.on('message', listener) + messageChannel.port2?.on('message', listener) } - protected unregisterWorkerMessageListener ( - port: ThreadWorkerWithMessageChannel, - listener: (message: MessageValue) => void + protected unregisterWorkerMessageListener ( + messageChannel: ThreadWorkerWithMessageChannel, + listener: (message: MessageValue) => void ): void { - port.port2?.removeListener('message', listener) + messageChannel.port2?.removeListener('message', listener) } - protected newWorker (): ThreadWorkerWithMessageChannel { + protected createWorker (): ThreadWorkerWithMessageChannel { return new Worker(this.filePath, { env: SHARE_ENV }) } - protected afterNewWorkerPushed ( - worker: ThreadWorkerWithMessageChannel - ): void { + protected afterWorkerSetup (worker: ThreadWorkerWithMessageChannel): void { const { port1, port2 } = new MessageChannel() worker.postMessage({ parent: port1 }, [port1]) worker.port1 = port1