1 import cluster
, { type Worker
} from
'node:cluster'
2 import type { MessageValue
} from
'../../utility-types'
3 import { AbstractPool
} from
'../abstract-pool'
4 import { type PoolOptions
, type PoolType
, PoolTypes
} from
'../pool'
5 import { type WorkerType
, WorkerTypes
} from
'../worker'
8 * A cluster pool with a fixed number of workers.
10 * @typeParam Data - Type of data sent to the worker. This can only be structured-cloneable data.
11 * @typeParam Response - Type of execution response. This can only be structured-cloneable data.
12 * @author [Christopher Quadflieg](https://github.com/Shinigami92)
15 export class FixedClusterPool
<
18 > extends AbstractPool
<Worker
, Data
, Response
> {
20 * Constructs a new poolifier fixed cluster pool.
22 * @param numberOfWorkers - Number of workers for this pool.
23 * @param filePath - Path to an implementation of a `ClusterWorker` file, which can be relative or absolute.
24 * @param opts - Options for this fixed cluster pool.
27 numberOfWorkers
: number,
29 protected readonly opts
: PoolOptions
<Worker
> = {}
31 super(numberOfWorkers
, filePath
, opts
)
35 protected setupHook (): void {
36 cluster
.setupPrimary({ ...this.opts
.settings
, exec
: this.filePath
})
40 protected isMain (): boolean {
41 return cluster
.isPrimary
45 protected sendToWorker (
46 workerNodeKey
: number,
47 message
: MessageValue
<Data
>
49 this.workerNodes
[workerNodeKey
].worker
.send({
51 workerId
: this.getWorkerInfo(workerNodeKey
).id
as number
56 protected sendStartupMessageToWorker (workerNodeKey
: number): void {
57 this.sendToWorker(workerNodeKey
, {
63 protected registerWorkerMessageListener
<Message
extends Data
| Response
>(
64 workerNodeKey
: number,
65 listener
: (message
: MessageValue
<Message
>) => void
67 this.workerNodes
[workerNodeKey
].worker
.on('message', listener
)
71 protected registerOnceWorkerMessageListener
<Message
extends Data
| Response
>(
72 workerNodeKey
: number,
73 listener
: (message
: MessageValue
<Message
>) => void
75 this.workerNodes
[workerNodeKey
].worker
.once('message', listener
)
79 protected deregisterWorkerMessageListener
<Message
extends Data
| Response
>(
80 workerNodeKey
: number,
81 listener
: (message
: MessageValue
<Message
>) => void
83 this.workerNodes
[workerNodeKey
].worker
.off('message', listener
)
87 protected get
type (): PoolType
{
88 return PoolTypes
.fixed
92 protected get
worker (): WorkerType
{
93 return WorkerTypes
.cluster
97 protected get
busy (): boolean {
98 return this.internalBusy()