@coherentglobal/wasm-runner 0.0.103 → 0.1.19

package/dist/node/template/main.template.ejs

@@ -3,17 +3,17 @@
 /* eslint-disable no-useless-catch */
 /* eslint-disable no-param-reassign */
 /* eslint-disable no-underscore-dangle */
-const { parentPort, workerData } = require('worker_threads');
+const WorkerThread = require('<%= modules.workerThread %>').default;
+const { threadId } = require('worker_threads');
 const model = require('<%= runtime %>');
-const externalResolver = require('<%= modules.externalResolver %>');
 const { logger } = require('<%= modules.logger %>');
 const metadata = <%- metadata %>;
 const formspec = <%- formspec %>;
-
-var events = require('events');
-var eventEmitter = new events.EventEmitter();
-
 const modelName = '<%= modelName %>';
+const DISPATCH_TYPE = {
+  SPARK_SERVICE: 0,
+  EXTERNAL_API: 1,
+};
 
 logger.debug({
   TimeStamp: Date.now(),
@@ -22,20 +22,24 @@ logger.debug({
 
 });
 
-let wasm;
 let wb;
-
 let isReady = false;
-
 let execute = () => { return Promise.resolve(false) };
 let _execute = () => { return Promise.resolve(false) };
-
 let modelIsBusy = false;
 
-const DISPATCH_TYPE = {
-  SPARK_SERVICE: 0,
-  EXTERNAL_API: 1,
-};
+/**
+ * Place holder of the worker thread instance.
+ * Actual definition happens during Worker initialization.
+ */
+let workerThreadInstance = null;
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+const start = Date.now();
+let readyRes = undefined;
 
 try {
   logger.debug({
@@ -62,23 +66,20 @@ try {
           const resultData = new Uint8Array(
             instance.HEAPU8.subarray(argPtr, argPtr + argsOffSet)
           );
-          const result = dnc.decode(resultData).slice();
 
+          const result = dnc.decode(resultData).slice();
           const requestData = JSON.parse(result);
 
           switch (service) {
             case DISPATCH_TYPE.SPARK_SERVICE: {
-              const dispatchRes = await externalResolver.sparkService(requestData, ctx)
-              return dispatchRes
-            }
-            case DISPATCH_TYPE.EXTERNAL_API: {
-              // eslint-disable-next-line no-return-await
-              return await externalResolver.externalApi(requestData, ctx);
+              const resultSparkService = await workerThreadInstance.requestMessage(requestData, ctx);
+              return resultSparkService;
             }
             default: {
               logger.error(
                 `Not implemented for service ${service} data ${result}`
               );
+              return '';
             }
           }
         };
@@ -87,7 +88,7 @@ try {
         execute = function (req) {
           try {
             if (req === "isReady") {
-              return Promise.resolve(isReady);
+              return Promise.resolve(readyRes);
             }
 
             if (req === "destroy") {
@@ -95,7 +96,7 @@ try {
               return Promise.resolve(true);
             }
 
-            if(!req.request_data) req.request_data = {}
+            if (!req.request_data) req.request_data = {}
 
             let rawString = false;
             if (req.request_data._raw) {
@@ -115,7 +116,7 @@ try {
               }
             }
 
-            if(!req.request_meta) req.request_meta = {}
+            if (!req.request_meta) req.request_meta = {}
 
             if (req.request_meta?._ctx?.correlationId) {
               ctx.correlationId = req.request_meta._ctx.correlationId;
@@ -149,22 +150,34 @@ try {
               delete req.request_meta._ctx.servicemap;
             }
 
-            if(req.request_meta._ctx?.xcallChainId) {
+            if (req.request_meta._ctx?.xcallChainId) {
               ctx.xcallChainId = req.request_meta._ctx?.xcallChainId;
               delete req.request_meta._ctx.xcallChainId;
             }
 
-            if(req.callid) {
+            if (req.request_meta._ctx?.callId) {
+              ctx.callid = req.request_meta._ctx.callId;
+              delete req.request_meta._ctx.callId;
+            } else if (req.callid) {
               ctx.callid = req.callid;
               delete req.callid;
             }
 
+            if (!req.request_meta?.allow_range_address) {
+              req.request_meta.allow_range_address = false;
+            }
+
+            if (req.request_meta?._ctx?.trace) {
+              ctx.trace = req.request_meta._ctx.trace;
+              delete req.request_meta._ctx.trace;
+            }
+
             const startTime = Date.now();
             logger.debug({
               TimeStamp: Date.now(),
               EventType: 'Runner.ModelThread.Execute',
-             threadCallId: ctx.callid,
-
+              threadCallId: ctx.callid,
+              correlationId: ctx.correlationId,
             });
 
             const input = JSON.stringify(req);
@@ -191,21 +204,23 @@ try {
                   [wb, inBuffer, encoded.length, outBuffer],
                   { async: true }
                 ).then((outPtr) => {
+
                   var offset = instance.getValue(outBuffer, "i32");
                   const resultData = new Uint8Array(instance.HEAPU8.subarray(outPtr, outPtr + offset));
                   var result = dnc.decode(resultData).slice();
+
                   instance._free(inBuffer);
                   instance._free(outBuffer);
                   instance._free(outPtr);
 
-                  if(isMetadataSubSvc) {
+                  if (isMetadataSubSvc) {
                     result = JSON.parse(result);
-                    if(!result.response_data) {
+                    if (!result.response_data) {
                       result.response_data = {}
                     }
                     const hasImageOutputs = metadata?.ImageOutputs ?? []
-                    const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce( (prev, curr) => {
-                      if(!curr?.ImageName) return prev
+                    const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce((prev, curr) => {
+                      if (!curr?.ImageName) return prev
                       prev[curr.ImageName] = curr?.Base64Content || ""
                       return prev
                     }, {}) : {}
@@ -220,14 +235,14 @@ try {
                   }
 
 
-                  if(isFormSpecSubSvc) {
+                  if (isFormSpecSubSvc) {
                     result = JSON.parse(result);
-                    if(!result.response_data) {
+                    if (!result.response_data) {
                       result.response_data = {}
                     }
 
                     // result.response_data.outputs = Object.assign({}, result.response_data.outputs, formspec);
-                     result.response_data.outputs = formspec;
+                    result.response_data.outputs = formspec;
                     return rawString ? JSON.stringify(result) : result;
                   }
 
@@ -237,6 +252,7 @@ try {
                     Message: `Complete execute model ${modelName}`,
                     innerExecuteDuration: Date.now() - startTime,
                     threadCallId: ctx.callid,
+                    correlationId: ctx.correlationId,
                   });
 
                   return rawString ? result : JSON.parse(result);
@@ -257,14 +273,14 @@ try {
                   instance._free(outBuffer);
                   instance._free(outPtr);
 
-                  if(isMetadataSubSvc) {
+                  if (isMetadataSubSvc) {
                     result = JSON.parse(result);
-                    if(!result.response_data) {
+                    if (!result.response_data) {
                       result.response_data = {}
                     }
                     const hasImageOutputs = metadata?.ImageOutputs ?? []
-                    const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce( (prev, curr) => {
-                      if(!curr?.ImageName) return prev
+                    const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce((prev, curr) => {
+                      if (!curr?.ImageName) return prev
                       prev[curr.ImageName] = curr?.Base64Content || ""
                       return prev
                     }, {}) : {}
@@ -273,17 +289,17 @@ try {
                       ...metadata?.Outputs,
                       ...imageOutputs,
                     }
-                    result.response_data.outputs = Object.assign({},  metadataOutput);
+                    result.response_data.outputs = Object.assign({}, metadataOutput);
                     return rawString ? JSON.stringify(result) : result;
                   }
 
-                  if(isFormSpecSubSvc) {
+                  if (isFormSpecSubSvc) {
                     result = JSON.parse(result);
-                    if(!result.response_data) {
+                    if (!result.response_data) {
                       result.response_data = {}
                     }
 
-                     // result.response_data.outputs = Object.assign({}, result.response_data.outputs, formspec);
+                    // result.response_data.outputs = Object.assign({}, result.response_data.outputs, formspec);
                     result.response_data.outputs = formspec;
                     return rawString ? JSON.stringify(result) : result;
                   }
@@ -293,11 +309,11 @@ try {
                     Message: `Complete execute model ${modelName}`,
                     EventType: 'Runner.ModelThread.Execute',
                     innerExecuteDuration: Date.now() - startTime,
-                   threadCallId: ctx.callid,
-
+                    threadCallId: ctx.callid,
+                    correlationId: ctx.correlationId,
                   });
 
-                  return rawString ? result : JSON.parse(result);
+                  return rawString ? result.concat('') : JSON.parse(result);
                 });
               }
             } else {
@@ -316,14 +332,15 @@ try {
                 instance._free(outBuffer);
                 instance._free(outPtr);
 
-                if(isMetadataSubSvc) {
+
+                if (isMetadataSubSvc) {
                   result = JSON.parse(result);
-                  if(!result.response_data) {
+                  if (!result.response_data) {
                     result.response_data = {}
                   }
                   const hasImageOutputs = metadata?.ImageOutputs ?? []
-                  const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce( (prev, curr) => {
-                    if(!curr?.ImageName) return prev
+                  const imageOutputs = hasImageOutputs ? hasImageOutputs.reduce((prev, curr) => {
+                    if (!curr?.ImageName) return prev
                     prev[curr.ImageName] = curr?.Base64Content || null
                     return prev
                   }, {}) : {}
@@ -336,13 +353,13 @@ try {
                   return rawString ? JSON.stringify(result) : result;
                 }
 
-                if(isFormSpecSubSvc) {
+                if (isFormSpecSubSvc) {
                   result = JSON.parse(result);
-                  if(!result.response_data) {
+                  if (!result.response_data) {
                     result.response_data = {}
                   }
 
-                   //result.response_data.outputs = Object.assign({}, result.response_data.outputs, formspec);
+                  //result.response_data.outputs = Object.assign({}, result.response_data.outputs, formspec);
                   result.response_data.outputs = formspec;
                   return rawString ? JSON.stringify(result) : result;
                 }
@@ -352,11 +369,11 @@ try {
                   Message: `Complete execute model ${modelName}`,
                   EventType: 'Runner.ModelThread.Execute',
                   innerExecuteDuration: Date.now() - startTime,
-                 threadCallId: ctx.callid,
-
+                  threadCallId: ctx.callid,
+                  correlationId: ctx.correlationId,
                 });
 
-                return rawString ? result : JSON.parse(result);
+                return rawString ? result.concat('') : JSON.parse(result);
               });
             }
 
@@ -368,34 +385,33 @@ try {
         };
 
 
-        _execute = function (req) {
-          return new Promise((resolve, reject) => {
-            if(!modelIsBusy) {
-              logger.debug({
-                TimeStamp: Date.now(),
-                Message: `Execute ${modelName}`,
-                EventType: 'Execute',
-              })
+        _execute = async function (req) {
+          while (1) {
+            if (!modelIsBusy) {
               modelIsBusy = true;
-              return _execute(req)
-                  .then((result) => {
-                    resolve(result);
-                  }).catch((err) => {
-                    reject(err);
-                  }).finally(() => {
-                    modelIsBusy = false
-                  });
+              const start = Date.now();
+
+              try {
+                return {
+                  payload: await execute(req),
+                  memoryUsage: process.memoryUsage().rss,
+                  threadId,
+                  time: Date.now() - start
+                }
+              } catch (err) {
+                return {
+                  error: err,
+                  memoryUsage: process.memoryUsage().rss,
+                  threadId,
+                  time: Date.now() - start
+                }
+              } finally {
+                modelIsBusy = false;
+              }
             } else {
-              setTimeout(() => {
-                logger.debug({
-                  TimeStamp: Date.now(),
-                  Message: `Execute.wait ${modelName}`,
-                  EventType: 'Execute.wait',
-                });
-                execute(req)
-              }, 1)
+              await sleep(5)
             }
-          });
+          }
         }
 
         logger.debug({
@@ -412,6 +428,13 @@ try {
         });
 
         isReady = true;
+        readyRes = {
+          memoryUsage: process.memoryUsage().rss,
+          threadId,
+          time: Date.now() - start,
+          readyTs: Date.now()
+        };
+
         return Promise.resolve();
       } catch (err) {
         instance = undefined
@@ -419,12 +442,29 @@ try {
         return Promise.reject(err);
       }
     })
-    .catch((err) => logger.error('Instance Error: ', err));
+    .catch((err) => logger.error('Instance Error: ' + err.message, err));
 } catch (err) {
   logger.error('Error', err)
 }
 
-parentPort.on("message", (input) => {
+/**
+ * Main worker function that triggers the `execute` logic which communicates to
+ * the WASM functions.
+ *
+ * @param {String|Object} input
+ * Can be 'isReady', JSON String or an Object.
+ *
+ * @returns {*}
+ */
+async function main(input) {
+  // IMPORTANT: `postMessage` from parent also triggers this function, better not to pass `data` from parent
+  // and use `pool.execute()` instead.
+  if (!input) {
+    return;
+  }
+
+  // Private function init
+  workerThreadInstance = this;
 
   if (
     typeof input !== "string" &&
@@ -434,7 +474,7 @@ parentPort.on("message", (input) => {
     throw new Error("Param must be a json string or json object");
   }
 
-  const promise = execute(input);
+  return _execute(input);
+};
 
-  return promise.then((result) => parentPort.postMessage(result));
-});
+module.exports = new WorkerThread(main);

package/dist/node/threads/mockWorkerThread.d.ts

@@ -0,0 +1,28 @@
+declare class WorkerThread extends ThreadWorker<any, any> {
+    constructor(taskFunctions: any, opts: any);
+    /**
+     * Sends message back to the main thread.
+     *
+     * @param {Object} message
+     */
+    postMessage(message: any): void;
+    /**
+     * Receives message from the main thread or other threads.
+     *
+     * @param {Function} callback
+     * Function to receive the message data.
+     */
+    onMessage(callback: Function): void;
+    /**
+     * Sends a request to the main thread and waits for a response.
+     *
+     * @param {Object} payload
+     * Object or details that needs to be passed to main thread as `payload`.
+     *
+     * @returns {Promise<Object>}
+     * Response from main thread.
+     */
+    requestMessage(payload: any, context: any): Promise<any>;
+}
+import { ThreadWorker } from "poolifier";
+export { WorkerThread as default };

package/dist/node/threads/mockWorkerThread.js

@@ -0,0 +1,89 @@
+const events = require("events");
+const { isMainThread, parentPort } = require("worker_threads");
+const { createId } = require("@paralleldrive/cuid2");
+const { ThreadWorker } = require("poolifier");
+// eslint-disable-next-line import/no-unresolved, import/extensions
+const { THREAD_EVENT_TYPES } = require("../../mockConstants");
+
+class WorkerThread extends ThreadWorker {
+  // eslint-disable-next-line no-useless-constructor
+  constructor(taskFunctions, opts) {
+    super(taskFunctions, opts);
+  }
+
+  /**
+   * Sends message back to the main thread.
+   *
+   * @param {Object} message
+   */
+  postMessage(message) {
+    if (isMainThread) {
+      throw Error("Failed to post message using main thread.");
+    } else {
+      parentPort.postMessage(message);
+    }
+  }
+
+  /**
+   * Receives message from the main thread or other threads.
+   *
+   * @param {Function} callback
+   * Function to receive the message data.
+   */
+  onMessage(callback) {
+    if (isMainThread) {
+      throw Error("Failed to receive message using main thread.");
+    } else {
+      parentPort.on("message", callback);
+    }
+  }
+
+  /**
+   * Sends a request to the main thread and waits for a response.
+   *
+   * @param {Object} payload
+   * Object or details that needs to be passed to main thread as `payload`.
+   *
+   * @returns {Promise<Object>}
+   * Response from main thread.
+   */
+  async requestMessage(payload, context) {
+    return new Promise((resolve, reject) => {
+      const eventId = createId();
+      const eventEmitter = new events.EventEmitter();
+
+      // Prepare to process the response from parent
+      eventEmitter.once(eventId, (response) => {
+        eventEmitter.removeAllListeners(eventId);
+        if (response.type === THREAD_EVENT_TYPES.ERROR) {
+          reject(response?.payload);
+        } else {
+          resolve(response?.payload);
+        }
+      });
+
+      // Send the request to parent
+      this.postMessage({
+        payload,
+        context,
+        type: THREAD_EVENT_TYPES.REQUEST_EXECUTE,
+        eventId,
+      });
+
+      // Receive the response from parent once ready
+      this.onMessage((message) => {
+        if (
+          message.eventId === eventId &&
+          (message?.type === THREAD_EVENT_TYPES.RESPONSE ||
+            message?.type === THREAD_EVENT_TYPES.ERROR)
+        ) {
+          eventEmitter.emit(eventId, message);
+        }
+      });
+    });
+  }
+}
+
+module.exports = {
+  default: WorkerThread,
+};

package/dist/node/threads/mockWorkerThread.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mockWorkerThread.js","sourceRoot":"","sources":["../../../src/node/threads/mockWorkerThread.js"],"names":[],"mappings":";;;;;;;;;AAAA,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;AACjC,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAC/D,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;AACrD,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;AAC9C,mEAAmE;AACnE,MAAM,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;AAE9D,MAAM,YAAa,SAAQ,YAAY;IACrC,kDAAkD;IAClD,YAAY,aAAa,EAAE,IAAI;QAC7B,KAAK,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACH,WAAW,CAAC,OAAO;QACjB,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,QAAQ;QAChB,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACG,cAAc,CAAC,OAAO,EAAE,OAAO;;YACnC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACrC,MAAM,OAAO,GAAG,QAAQ,EAAE,CAAC;gBAC3B,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBAE/C,8CAA8C;gBAC9C,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE;oBACtC,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;oBACzC,IAAI,QAAQ,CAAC,IAAI,KAAK,kBAAkB,CAAC,KAAK,EAAE,CAAC;wBAC/C,MAAM,CAAC,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,OAAO,CAAC,CAAC;oBAC5B,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,OAAO,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,6BAA6B;gBAC7B,IAAI,CAAC,WAAW,CAAC;oBACf,OAAO;oBACP,OAAO;oBACP,IAAI,EAAE,kBAAkB,CAAC,eAAe;oBACxC,OAAO;iBACR,CAAC,CAAC;gBAEH,8CAA8C;gBAC9C,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE;oBACzB,IACE,OAAO,CAAC,OAAO,KAAK,OAAO;wBAC3B,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,MAAK,kBAAkB,CAAC,QAAQ;4BAC5C,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,MAAK,kBAAkB,CAAC,KAAK,CAAC,EAC7C,CAAC;wBACD,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACtC,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;CACF;AAED,MAAM,CAAC,OAAO,GAAG;IACf,OAAO,EAAE,YAAY;CACtB,CAAC"}

package/dist/node/threads/workerPool.d.ts

@@ -0,0 +1,17 @@
+import { FixedThreadPool } from "poolifier";
+import type { ThreadPoolOptions } from "poolifier";
+import { EventResponseData } from "../../types";
+declare class WorkerPool extends FixedThreadPool {
+    constructor(numberOfThreads: number, filePath: string, opts?: ThreadPoolOptions, maximumNumberOfThreads?: number);
+    /**
+     * Sends message to a specific worker thread or broadcasts the message
+     * to all running threads.
+     *
+     * @param message
+     * Any data that needs to passed to the worker.
+     *
+     */
+    postMessage(message: EventResponseData): void;
+}
+export default WorkerPool;
+export type { FixedThreadPool };

package/dist/node/threads/workerPool.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const poolifier_1 = require("poolifier");
+class WorkerPool extends poolifier_1.FixedThreadPool {
+    constructor(numberOfThreads, filePath, opts, maximumNumberOfThreads) {
+        super(numberOfThreads, filePath, opts);
+    }
+    /**
+     * Sends message to a specific worker thread or broadcasts the message
+     * to all running threads.
+     *
+     * @param message
+     * Any data that needs to passed to the worker.
+     *
+     */
+    postMessage(message) {
+        let workerIndex = -1;
+        if (message.threadId) {
+            workerIndex = this.workerNodes.findIndex((w) => { var _a; return ((_a = w.info) === null || _a === void 0 ? void 0 : _a.id) === message.threadId; });
+        }
+        if (workerIndex > -1) {
+            this.workerNodes[workerIndex].worker.postMessage(message);
+        }
+    }
+}
+exports.default = WorkerPool;
+//# sourceMappingURL=workerPool.js.map

package/dist/node/threads/workerPool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workerPool.js","sourceRoot":"","sources":["../../../src/node/threads/workerPool.ts"],"names":[],"mappings":";;AAAA,yCAA4C;AAI5C,MAAM,UAAW,SAAQ,2BAAe;IACtC,YACE,eAAuB,EACvB,QAAgB,EAChB,IAAwB,EACxB,sBAA+B;QAE/B,KAAK,CAAC,eAAe,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED;;;;;;;OAOG;IACH,WAAW,CAAC,OAA0B;QACpC,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC;QACrB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CACtC,CAAC,CAAC,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,CAAC,CAAC,IAAI,0CAAE,EAAE,MAAK,OAAO,CAAC,QAAQ,CAAA,EAAA,CACvC,CAAC;QACJ,CAAC;QAED,IAAI,WAAW,GAAG,CAAC,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;CACF;AAED,kBAAe,UAAU,CAAC"}

package/dist/node/threads/workerPool.ts

@@ -0,0 +1,38 @@
+import { FixedThreadPool } from "poolifier";
+import type { ThreadPoolOptions } from "poolifier";
+import { EventResponseData } from "../../types";
+
+class WorkerPool extends FixedThreadPool {
+  constructor(
+    numberOfThreads: number,
+    filePath: string,
+    opts?: ThreadPoolOptions,
+    maximumNumberOfThreads?: number
+  ) {
+    super(numberOfThreads, filePath, opts);
+  }
+
+  /**
+   * Sends message to a specific worker thread or broadcasts the message
+   * to all running threads.
+   *
+   * @param message
+   * Any data that needs to passed to the worker.
+   *
+   */
+  postMessage(message: EventResponseData) {
+    let workerIndex = -1;
+    if (message.threadId) {
+      workerIndex = this.workerNodes.findIndex(
+        (w) => w.info?.id === message.threadId
+      );
+    }
+
+    if (workerIndex > -1) {
+      this.workerNodes[workerIndex].worker.postMessage(message);
+    }
+  }
+}
+
+export default WorkerPool;
+export type { FixedThreadPool };

package/dist/node/threads/workerThread.d.ts

@@ -0,0 +1,31 @@
+import { ThreadWorker } from "poolifier";
+import type { TaskFunctions, TaskFunction, WorkerOptions } from "poolifier";
+import type { EventRequestData } from "../../types";
+declare class WorkerThread extends ThreadWorker {
+    constructor(taskFunctions: TaskFunction<unknown, Response> | TaskFunctions<unknown, Response>, opts?: WorkerOptions);
+    /**
+     * Sends message back to the main thread.
+     *
+     * @param {EventRequestData} message
+     */
+    postMessage(message: EventRequestData): void;
+    /**
+     * Receives message from the main thread or other threads.
+     *
+     * @param {Function} callback
+     * Function to receive the message data.
+     */
+    onMessage(callback: (value?: any) => void): void;
+    private cleanupListener;
+    /**
+     * Sends a request to the main thread and waits for a response.
+     *
+     * @param {Object} payload
+     * Object or details that needs to be passed to main thread as `payload`.
+     *
+     * @returns {Promise<Object>}
+     * Response from main thread.
+     */
+    requestMessage(payload: any, context: any): Promise<unknown>;
+}
+export default WorkerThread;