X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Futility-types.ts;h=05af4e3d396b7c89d22ba0c895ceb6c4649e5f42;hb=3fafb1b2f068533b39818d66b5b0b9972f76f577;hp=2403b8a257b0377e441a4dc2568b49571be4cdc2;hpb=deb85c12b77faf6974551cefcd9676e62a392086;p=poolifier.git

diff --git a/src/utility-types.ts b/src/utility-types.ts
index 2403b8a2..05af4e3d 100644
--- a/src/utility-types.ts
+++ b/src/utility-types.ts
@@ -1,6 +1,7 @@
-import type { Worker } from 'cluster'
-import type { MessagePort } from 'worker_threads'
+import type { Worker as ClusterWorker } from 'node:cluster'
+import type { MessagePort } from 'node:worker_threads'
 import type { KillBehavior } from './worker/worker-options'
+import type { IPoolWorker } from './pools/pool-worker'
 
 /**
  * Make all properties in T non-readonly.
@@ -12,16 +13,16 @@ export type Draft<T> = { -readonly [P in keyof T]?: T[P] }
  */
 export interface MessageValue<
   Data = unknown,
-  MainWorker extends Worker | MessagePort | unknown = unknown
+  MainWorker extends ClusterWorker | MessagePort | unknown = unknown
 > {
   /**
    * Input data that will be passed to the worker.
    */
   readonly data?: Data
   /**
-   * ID of the message.
+   * Id of the message.
    */
-  readonly id?: number
+  readonly id?: string
   /**
    * Kill code.
    */
@@ -30,10 +31,38 @@ export interface MessageValue<
    * Error.
    */
   readonly error?: string
+  /**
+   * Runtime.
+   */
+  readonly runTime?: number
   /**
    * Reference to main worker.
    *
-   * _Only for internal use_
+   * Only for internal use.
    */
   readonly parent?: MainWorker
 }
+
+/**
+ * An object holding the execution response promise resolve/reject callbacks.
+ *
+ * @typeParam Worker - Type of worker.
+ * @typeParam Response - Type of execution response. This can only be serializable data.
+ */
+export interface PromiseResponseWrapper<
+  Worker extends IPoolWorker,
+  Response = unknown
+> {
+  /**
+   * Resolve callback to fulfill the promise.
+   */
+  readonly resolve: (value: Response) => void
+  /**
+   * Reject callback to reject the promise.
+   */
+  readonly reject: (reason?: string) => void
+  /**
+   * The worker handling the promise.
+   */
+  readonly worker: Worker
+}