perf: use a single map to store pool workers and their related data

[poolifier.git] / src / pools / pool-internal.ts

diff --git a/src/pools/pool-internal.ts b/src/pools/pool-internal.ts
index f5f1d1908a14d60f2e26f9d6baa75c627468e96b..97e5d5d2d56ae75435b19d1ccb8afae10ff2e8ed 100644 (file)
--- a/src/pools/pool-internal.ts
+++ b/src/pools/pool-internal.ts
@@ -1,9 +1,8 @@
-import EventEmitter from 'events'
-import type { AbstractPoolWorker } from './abstract-pool-worker'
 import type { IPool } from './pool'
+import type { IPoolWorker } from './pool-worker'
 
 /**
- * Pool types.
+ * Internal pool types.
  */
 export enum PoolType {
   FIXED = 'fixed',
@@ -11,7 +10,7 @@ export enum PoolType {
 }
 
 /**
- * Tasks usage statistics.
+ * Internal tasks usage statistics.
  */
 export interface TasksUsage {
   run: number
@@ -21,35 +20,31 @@ export interface TasksUsage {
 }
 
 /**
- * Internal poolifier pool emitter.
+ * Internal worker type.
+ *
+ * @typeParam Worker - Type of worker which manages this pool.
  */
-export class PoolEmitter extends EventEmitter {}
+export interface WorkerType<Worker extends IPoolWorker> {
+  worker: Worker
+  tasksUsage: TasksUsage
+}
 
 /**
  * Internal contract definition for a poolifier pool.
  *
- * @template Worker Type of worker which manages this pool.
- * @template Data Type of data sent to the worker.
- * @template Response Type of response of execution.
+ * @typeParam Worker - Type of worker which manages this pool.
+ * @typeParam Data - Type of data sent to the worker.
+ * @typeParam Response - Type of response of execution.
  */
 export interface IPoolInternal<
-  Worker extends AbstractPoolWorker,
+  Worker extends IPoolWorker,
   Data = unknown,
   Response = unknown
 > extends IPool<Data, Response> {
   /**
-   * List of currently available workers.
-   */
-  readonly workers: Worker[]
-
-  /**
-   * Emitter on which events can be listened to.
-   *
-   * Events that can currently be listened to:
-   *
-   * - `'busy'`
+   * Map of workers.
    */
-  readonly emitter?: PoolEmitter
+  readonly workers: Map<number, WorkerType<Worker>>
 
   /**
    * Pool type.
@@ -58,11 +53,6 @@ export interface IPoolInternal<
    */
   readonly type: PoolType
 
-  /**
-   * Maximum number of workers that can be created by this pool.
-   */
-  readonly max?: number
-
   /**
    * Whether the pool is busy or not.
    *
@@ -76,7 +66,7 @@ export interface IPoolInternal<
   readonly numberOfRunningTasks: number
 
   /**
-   * Find a free worker based on the number of tasks the worker has applied.
+   * Finds a free worker based on the number of tasks the worker has applied.
    *
    * If a worker is found with `0` running tasks, it is detected as free and returned.
    *
@@ -84,29 +74,21 @@ export interface IPoolInternal<
    *
    * @returns A free worker if there is one, otherwise `false`.
    */
-  findFreeWorker(): Worker | false
-
-  /**
-   * Get worker index.
-   *
-   * @param worker The worker.
-   * @returns The worker index.
-   */
-  getWorkerIndex(worker: Worker): number
+  findFreeWorker: () => Worker | false
 
   /**
-   * Get worker running tasks.
+   * Gets worker running tasks.
    *
-   * @param worker The worker.
+   * @param worker - The worker.
    * @returns The number of tasks currently running on the worker.
    */
-  getWorkerRunningTasks(worker: Worker): number | undefined
+  getWorkerRunningTasks: (worker: Worker) => number | undefined
 
   /**
-   * Get worker average tasks run time.
+   * Gets worker average tasks runtime.
    *
-   * @param worker The worker.
-   * @returns The average tasks run time on the worker.
+   * @param worker - The worker.
+   * @returns The average tasks runtime on the worker.
    */
-  getWorkerAverageTasksRunTime(worker: Worker): number | undefined
+  getWorkerAverageTasksRunTime: (worker: Worker) => number | undefined
 }