Improvements for #161 (#169)

[poolifier.git] / src / utility-types.ts

diff --git a/src/utility-types.ts b/src/utility-types.ts
index aa4d058cce142ffa5214c8d2ce2965e428309d14..d680fa16ddc7e4748a41f2e878def0384eb2e305 100644 (file)
--- a/src/utility-types.ts
+++ b/src/utility-types.ts
@@ -1,3 +1,7 @@
+import type { Worker } from 'cluster'
+import type { MessagePort } from 'worker_threads'
+import type { KillBehavior } from './worker/worker-options'
+
 /**
  * Make all properties in T non-readonly
  */
@@ -24,7 +28,10 @@ export type JSONArray = Array<JSONValue>
 /**
  * Message object that is passed between worker and main worker.
  */
-export interface MessageValue<Data = unknown> {
+export interface MessageValue<
+  Data = unknown,
+  MainWorker extends Worker | MessagePort | unknown = unknown
+> {
   /**
    * Input data that will be passed to the worker.
    */
@@ -36,7 +43,7 @@ export interface MessageValue<Data = unknown> {
   /**
    * Kill code.
    */
-  readonly kill?: number
+  readonly kill?: KillBehavior | 1
   /**
    * Error.
    */
@@ -46,5 +53,5 @@ export interface MessageValue<Data = unknown> {
    *
    * _Only for internal use_
    */
-  readonly parent?: MessagePort
+  readonly parent?: MainWorker
 }