refactor: remove unneeded encapsulation around tasks usage handling

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

diff --git a/src/pools/pool-internal.ts b/src/pools/pool-internal.ts
index d641fcf9d9fec246ee26a92a67a78d348b3dacf3..9da6495266ab6c6e4654def94eca460b4630cb24 100644 (file)
--- a/src/pools/pool-internal.ts
+++ b/src/pools/pool-internal.ts
@@ -1,81 +1,86 @@
-import EventEmitter from 'events'
-import type { MessageValue } from '../utility-types'
-import type { IWorker } from './abstract-pool'
 import type { IPool } from './pool'
+import type { IPoolWorker } from './pool-worker'
 
 /**
- * Internal poolifier pool emitter.
+ * Internal pool types.
  */
-export class PoolEmitter extends EventEmitter {}
+export enum PoolType {
+  FIXED = 'fixed',
+  DYNAMIC = 'dynamic'
+}
+
+/**
+ * Internal tasks usage statistics.
+ */
+export interface TasksUsage {
+  run: number
+  running: number
+  runTime: number
+  avgRunTime: number
+}
+
+/**
+ * Internal worker type.
+ *
+ * @typeParam Worker - Type of worker which manages this pool.
+ */
+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 IWorker,
+  Worker extends IPoolWorker,
   Data = unknown,
   Response = unknown
 > extends IPool<Data, Response> {
   /**
-   * List of currently available workers.
+   * Pool workers map.
    */
-  readonly workers: Worker[]
+  readonly workers: Map<number, WorkerType<Worker>>
 
   /**
-   * The tasks map.
+   * Pool type.
    *
-   * - `key`: The `Worker`
-   * - `value`: Number of tasks currently in progress on the worker.
+   * If it is `'dynamic'`, it provides the `max` property.
    */
-  readonly tasks: Map<Worker, number>
+  readonly type: PoolType
 
   /**
-   * Emitter on which events can be listened to.
+   * Whether the pool is busy or not.
    *
-   * Events that can currently be listened to:
-   *
-   * - `'FullPool'`
-   */
-  readonly emitter: PoolEmitter
-
-  /**
-   * Whether the pool is dynamic or not.
-   *
-   * If it is dynamic, it provides the `max` property.
+   * The pool busyness boolean status.
    */
-  readonly dynamic: boolean
+  readonly busy: boolean
 
   /**
-   * Maximum number of workers that can be created by this pool.
+   * Number of tasks currently concurrently running.
    */
-  readonly max?: number
+  readonly numberOfRunningTasks: number
 
   /**
-   * Creates a new worker for this pool and sets it up completely.
+   * Finds a free worker based on the number of tasks the worker has applied.
    *
-   * @returns New, completely set up worker.
-   */
-  createAndSetupWorker(): Worker
-
-  /**
-   * Shut down given worker.
+   * If a worker is found with `0` running tasks, it is detected as free and returned.
+   *
+   * If no free worker is found, `false` is returned.
    *
-   * @param worker A worker within `workers`.
+   * @returns A free worker if there is one, otherwise `false`.
    */
-  destroyWorker(worker: Worker): void | Promise<void>
+  findFreeWorker: () => Worker | false
 
   /**
-   * Register a listener callback on a given worker.
+   * Gets worker tasks usage.
    *
-   * @param worker A worker.
-   * @param listener A message listener callback.
+   * @param worker - The worker.
+   * @returns The tasks usage on the worker.
    */
-  registerWorkerMessageListener<Message extends Data | Response>(
-    worker: Worker,
-    listener: (message: MessageValue<Message>) => void
-  ): void
+  getWorkerTasksUsage: (worker: Worker) => TasksUsage | undefined
 }