X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fcluster%2Ffixed.ts;h=3f56c858b0ed5b363fc19da06eb1b9a295a66dd5;hb=2c039e4373e86714cdf27e77440b12ee8eb2e4db;hp=ed4f6fed2826078bfdeac466bcec7e8c4c80bef5;hpb=56960396878dec81e3ebeea5b76387efca8cc2dc;p=poolifier.git diff --git a/src/pools/cluster/fixed.ts b/src/pools/cluster/fixed.ts index ed4f6fed..3f56c858 100644 --- a/src/pools/cluster/fixed.ts +++ b/src/pools/cluster/fixed.ts @@ -1,9 +1,13 @@ -import type { ClusterSettings, Worker } from 'node:cluster' -import cluster from 'node:cluster' +import cluster, { type ClusterSettings, type Worker } from 'node:cluster' import type { MessageValue } from '../../utility-types' import { AbstractPool } from '../abstract-pool' -import type { PoolOptions } from '../pool' -import { PoolType } from '../pool' +import { + type PoolOptions, + type PoolType, + PoolTypes, + type WorkerType, + WorkerTypes +} from '../pool' /** * Options for a poolifier cluster pool. @@ -14,8 +18,7 @@ export interface ClusterPoolOptions extends PoolOptions { * * @see https://nodejs.org/api/cluster.html#cluster_cluster_fork_env */ - // eslint-disable-next-line @typescript-eslint/no-explicit-any - env?: any + env?: Record /** * Cluster settings. * @@ -29,8 +32,6 @@ export interface ClusterPoolOptions extends PoolOptions { * * It is possible to perform tasks in sync or asynchronous mode as you prefer. * - * This pool selects the workers in a round robin fashion. - * * @typeParam Data - Type of data sent to the worker. This can only be serializable data. * @typeParam Response - Type of execution response. This can only be serializable data. * @author [Christopher Quadflieg](https://github.com/Shinigami92) @@ -50,7 +51,7 @@ export class FixedClusterPool< public constructor ( numberOfWorkers: number, filePath: string, - public readonly opts: ClusterPoolOptions = {} + protected readonly opts: ClusterPoolOptions = {} ) { super(numberOfWorkers, filePath, opts) } @@ -68,7 +69,10 @@ export class FixedClusterPool< /** @inheritDoc */ protected destroyWorker (worker: Worker): void { this.sendToWorker(worker, { kill: 1 }) - worker.kill() + worker.on('disconnect', () => { + worker.kill() + }) + worker.disconnect() } /** @inheritDoc */ @@ -96,13 +100,23 @@ export class FixedClusterPool< } /** @inheritDoc */ - public get type (): PoolType { - return PoolType.FIXED + protected get type (): PoolType { + return PoolTypes.fixed + } + + /** @inheritDoc */ + protected get worker (): WorkerType { + return WorkerTypes.cluster + } + + /** @inheritDoc */ + 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 */