X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fcluster%2Ffixed.ts;h=9844c4a8a912570c10043e4980b8bb3113aab6f1;hb=0295156c310b5872970aa5dd6674f165b77954e4;hp=b522947f25cb6f9497c7935fdcd131d1747e8b34;hpb=7c0ba92006a5c188738ffc5ff642c51f172df3d6;p=poolifier.git diff --git a/src/pools/cluster/fixed.ts b/src/pools/cluster/fixed.ts index b522947f..9844c4a8 100644 --- a/src/pools/cluster/fixed.ts +++ b/src/pools/cluster/fixed.ts @@ -1,4 +1,4 @@ -import { fork, isMaster, setupMaster, Worker } from 'cluster' +import cluster, { Worker } from 'cluster' import type { MessageValue } from '../../utility-types' import type { PoolOptions } from '../abstract-pool' import { AbstractPool } from '../abstract-pool' @@ -24,9 +24,8 @@ export interface ClusterPoolOptions extends PoolOptions { * * This pool selects the workers 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. - * + * @template DataType of data sent to the worker. This can only be serializable data. + * @template ResponseType of response of execution. This can only be serializable data. * @author [Christopher Quadflieg](https://github.com/Shinigami92) * @since 2.0.0 */ @@ -39,7 +38,7 @@ export class FixedClusterPool< * * @param numberOfWorkers Number of workers for this pool. * @param filePath Path to an implementation of a `ClusterWorker` file, which can be relative or absolute. - * @param opts Options for this fixed cluster pool. Default: `{}` + * @param [opts={}] Options for this fixed cluster pool. */ public constructor ( numberOfWorkers: number, @@ -49,14 +48,16 @@ export class FixedClusterPool< super(numberOfWorkers, filePath, opts) } + /** @inheritdoc */ protected setupHook (): void { - setupMaster({ + cluster.setupPrimary({ exec: this.filePath }) } + /** @inheritdoc */ protected isMain (): boolean { - return isMaster + return cluster.isPrimary } /** @inheritdoc */ @@ -65,6 +66,7 @@ export class FixedClusterPool< worker.kill() } + /** @inheritdoc */ protected sendToWorker (worker: Worker, message: MessageValue): void { worker.send(message) } @@ -77,10 +79,12 @@ export class FixedClusterPool< worker.on('message', listener) } + /** @inheritdoc */ protected createWorker (): Worker { - return fork(this.opts.env) + return cluster.fork(this.opts.env) } + /** @inheritdoc */ protected afterWorkerSetup (worker: Worker): void { // Listen worker messages. this.registerWorkerMessageListener(worker, super.workerListener())