worker-lib 1.0.2 → 1.0.4

package/dist/cjs/node.d.ts

@@ -0,0 +1,27 @@
+import { type Worker } from "node:worker_threads";
+type WorkerType = {
+    [key: string]: (...args: any) => any;
+};
+/**
+ * createWorker
+ *
+ * @template T
+ * @param {() => Worker} builder
+ * @param {number} [limit=0]
+ * @return {*}
+ */
+export declare const createWorker: <T extends WorkerType>(builder: () => Worker, limit?: number) => {
+    execute: <K extends keyof T>(name: K, ...value: Parameters<T[K]>) => Promise<Awaited<ReturnType<T[K]>>>;
+    waitAll: () => Promise<void>;
+    close: () => void;
+    setLimit: (limit: number) => void;
+};
+/**
+ *
+ *
+ * @template T
+ * @param {T} WorkerProc
+ * @return {*}
+ */
+export declare const initWorker: <T extends WorkerType>(WorkerProc: T) => T;
+export {};

package/dist/cjs/node.js

@@ -0,0 +1,150 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initWorker = exports.createWorker = void 0;
+const node_worker_threads_1 = require("node:worker_threads");
+const init = (worker) => {
+    return new Promise((resolve) => {
+        worker.once("message", () => {
+            resolve(worker);
+        });
+    });
+};
+const exec = (worker, name, ...value) => {
+    return new Promise((resolve, reject) => {
+        const p = async (data) => {
+            switch (data.type) {
+                case "callback":
+                    const r = value[data.payload.index](data.payload.value);
+                    worker.postMessage({
+                        type: "callback_result",
+                        payload: { id: data.payload.id, result: await r },
+                    });
+                    break;
+                case "result":
+                    worker.removeListener("message", p);
+                    resolve(data.payload);
+                    break;
+                case "error":
+                    worker.removeListener("message", p);
+                    reject(data.payload);
+                    break;
+            }
+        };
+        worker.addListener("message", p);
+        worker.postMessage({
+            type: "function",
+            payload: {
+                name,
+                value: value.map((v) => !(typeof v === "function") && v),
+                callback: value.map((v) => typeof v === "function"),
+            },
+        });
+    });
+};
+/**
+ * createWorker
+ *
+ * @template T
+ * @param {() => Worker} builder
+ * @param {number} [limit=0]
+ * @return {*}
+ */
+const createWorker = (builder, limit = 4) => {
+    let workers = Array(limit)
+        .fill(undefined)
+        .map(() => ({}));
+    const getResolver = async () => {
+        while (true) {
+            const target = workers.find(({ resultResolver }) => !resultResolver);
+            if (target) {
+                target.resultResolver = Promise.withResolvers();
+                if (!target.worker)
+                    target.worker = await init(builder());
+                return target;
+            }
+            await Promise.race(workers.map(({ resultResolver }) => resultResolver?.promise));
+        }
+    };
+    const execute = async (name, ...value) => {
+        const target = await getResolver();
+        const { resultResolver } = target;
+        if (!resultResolver)
+            throw new Error("Unexpected error");
+        exec(target.worker, name, ...value)
+            .then(resultResolver.resolve)
+            .catch(resultResolver.reject)
+            .finally(() => {
+            target.resultResolver = undefined;
+        });
+        return resultResolver.promise;
+    };
+    const waitAll = async () => {
+        while (workers.find(({ resultResolver }) => resultResolver)) {
+            await Promise.all(workers.flatMap(({ resultResolver }) => resultResolver ? [resultResolver.promise] : []));
+        }
+    };
+    const close = () => {
+        for (const { worker } of workers) {
+            worker?.terminate();
+        }
+    };
+    const setLimit = (limit) => {
+        workers = Array(limit)
+            .fill(undefined)
+            .map(() => ({}));
+    };
+    return { execute, waitAll, close, setLimit };
+};
+exports.createWorker = createWorker;
+/**
+ *
+ *
+ * @template T
+ * @param {T} WorkerProc
+ * @return {*}
+ */
+const initWorker = (WorkerProc) => {
+    const worker = node_worker_threads_1.parentPort;
+    if (!worker) {
+        throw new Error("This is not a worker thread");
+    }
+    worker.addListener("message", async (data) => {
+        if (data.type === "function") {
+            const { name, value, callback, } = data.payload;
+            const proc = WorkerProc[name];
+            if (proc) {
+                try {
+                    const params = value.map((v, index) => callback[index]
+                        ? (...params) => callbackProc(worker, index, params)
+                        : v);
+                    worker.postMessage({
+                        type: "result",
+                        payload: await proc(...params),
+                    });
+                }
+                catch (e) {
+                    worker.postMessage({ type: "error", payload: String(e) });
+                }
+            }
+        }
+    });
+    worker.postMessage(undefined);
+    return WorkerProc;
+};
+exports.initWorker = initWorker;
+const callbackProc = (worker, index, params) => {
+    const id = WorkerValue.id++;
+    return new Promise((resolve) => {
+        worker.once("message", (data) => {
+            if (data.type === "callback_result" && data.payload.id === id) {
+                resolve(data.payload.result);
+            }
+        });
+        worker.postMessage({
+            type: "callback",
+            payload: { id, index, value: params },
+        });
+    });
+};
+const WorkerValue = { id: 0, promises: {} };
+//# sourceMappingURL=node.js.map

package/dist/cjs/node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"node.js","sourceRoot":"","sources":["../../src/node.ts"],"names":[],"mappings":";;;AAAA,6DAA8D;AAsB9D,MAAM,IAAI,GAAG,CAAC,MAAc,EAAmB,EAAE;IAC/C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;YAC1B,OAAO,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AACF,MAAM,IAAI,GAAG,CACX,MAAc,EACd,IAAa,EACb,GAAG,KAA6B,EACC,EAAE;IACnC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,CAAC,GAAG,KAAK,EAAE,IAAwB,EAAE,EAAE;YAC3C,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,UAAU;oBACb,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBACxD,MAAM,CAAC,WAAW,CAAC;wBACjB,IAAI,EAAE,iBAAiB;wBACvB,OAAO,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE;qBAClD,CAAC,CAAC;oBACH,MAAM;gBACR,KAAK,QAAQ;oBACX,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;oBACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACtB,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACrB,MAAM;YACV,CAAC;QACH,CAAC,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,WAAW,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE;gBACP,IAAI;gBACJ,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC;gBACjE,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC;aAC7D;SACoB,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;;;;;;GAOG;AACI,MAAM,YAAY,GAAG,CAC1B,OAAqB,EACrB,KAAK,GAAG,CAAC,EACT,EAAE;IACF,IAAI,OAAO,GAGL,KAAK,CAAC,KAAK,CAAC;SACf,IAAI,CAAC,SAAS,CAAC;SACf,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACnB,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;QAC7B,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC;YACrE,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,EAAW,CAAC;gBACzD,IAAI,CAAC,MAAM,CAAC,MAAM;oBAAE,MAAM,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,MAAM,OAAO,CAAC,IAAI,CAChB,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,EAAE,OAAO,CAAC,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,KAAK,EACnB,IAAO,EACP,GAAG,KAAuB,EACU,EAAE;QACtC,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;QACnC,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAClC,IAAI,CAAC,cAAc;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,MAAO,EAAE,IAAc,EAAE,GAAG,KAAK,CAAC;aAC3C,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;aAC5B,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC;aAC5B,OAAO,CAAC,GAAG,EAAE;YACZ,MAAM,CAAC,cAAc,GAAG,SAAS,CAAC;QACpC,CAAC,CAAC,CAAC;QACL,OAAO,cAAc,CAAC,OAA6C,CAAC;IACtE,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACzB,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5D,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrC,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAC/C,CACF,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IACF,MAAM,KAAK,GAAG,GAAG,EAAE;QACjB,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;YACjC,MAAM,EAAE,SAAS,EAAE,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IACF,MAAM,QAAQ,GAAG,CAAC,KAAa,EAAE,EAAE;QACjC,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC;aACnB,IAAI,CAAC,SAAS,CAAC;aACf,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACrB,CAAC,CAAC;IACF,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;AAC/C,CAAC,CAAC;AA3DW,QAAA,YAAY,gBA2DvB;AACF;;;;;;GAMG;AACI,MAAM,UAAU,GAAG,CAAuB,UAAa,EAAE,EAAE;IAChE,MAAM,MAAM,GAAG,gCAAU,CAAC;IAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,KAAK,EAAE,IAAwB,EAAE,EAAE;QAC/D,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM,EACJ,IAAI,EACJ,KAAK,EACL,QAAQ,GACT,GAIG,IAAI,CAAC,OAAO,CAAC;YACjB,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAC9B,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CACpC,QAAQ,CAAC,KAAK,CAAC;wBACb,CAAC,CAAC,CAAC,GAAG,MAAiB,EAAE,EAAE,CACvB,YAAY,CAAI,MAAe,EAAE,KAAK,EAAE,MAAM,CAAC;wBACnD,CAAC,CAAC,CAAC,CACN,CAAC;oBACF,MAAM,CAAC,WAAW,CAAC;wBACjB,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,MAAM,IAAI,CAAC,GAAG,MAAM,CAAC;qBAC/B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IACH,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAC9B,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AArCW,QAAA,UAAU,cAqCrB;AAEF,MAAM,YAAY,GAAG,CAAI,MAAc,EAAE,KAAa,EAAE,MAAiB,EAAE,EAAE;IAC3E,MAAM,EAAE,GAAG,WAAW,CAAC,EAAE,EAAE,CAAC;IAC5B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,IAAwB,EAAE,EAAE;YAClD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC9D,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE;SACtC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAGxC,CAAC","sourcesContent":["import { parentPort, type Worker } from \"node:worker_threads\";\ntype WorkerType = { [key: string]: (...args: any) => any };\ntype WorkerRecvEvent<T> =\n  | {\n      type: \"function\";\n      payload: { name: keyof T; callback: boolean[]; value: unknown[] };\n    }\n  | { type: \"callback_result\"; payload: { id: number; result: unknown } };\ntype WorkerSendEvent<T extends WorkerType> =\n  | {\n      type: \"callback\";\n      payload: { id: number; result: unknown; index: number; value: T };\n    }\n  | {\n      type: \"result\";\n      payload: ReturnType<T[keyof T]>;\n    }\n  | {\n      type: \"error\";\n      payload: unknown;\n    };\n\nconst init = (worker: Worker): Promise<Worker> => {\n  return new Promise((resolve) => {\n    worker.once(\"message\", () => {\n      resolve(worker);\n    });\n  });\n};\nconst exec = <T extends WorkerType>(\n  worker: Worker,\n  name: keyof T,\n  ...value: Parameters<T[keyof T]>\n): Promise<ReturnType<T[keyof T]>> => {\n  return new Promise((resolve, reject) => {\n    const p = async (data: WorkerSendEvent<T>) => {\n      switch (data.type) {\n        case \"callback\":\n          const r = value[data.payload.index](data.payload.value);\n          worker.postMessage({\n            type: \"callback_result\",\n            payload: { id: data.payload.id, result: await r },\n          });\n          break;\n        case \"result\":\n          worker.removeListener(\"message\", p);\n          resolve(data.payload);\n          break;\n        case \"error\":\n          worker.removeListener(\"message\", p);\n          reject(data.payload);\n          break;\n      }\n    };\n    worker.addListener(\"message\", p);\n    worker.postMessage({\n      type: \"function\",\n      payload: {\n        name,\n        value: value.map((v: unknown) => !(typeof v === \"function\") && v),\n        callback: value.map((v: unknown) => typeof v === \"function\"),\n      },\n    } as WorkerRecvEvent<T>);\n  });\n};\n\n/**\n * createWorker\n *\n * @template T\n * @param {() => Worker} builder\n * @param {number} [limit=0]\n * @return {*}\n */\nexport const createWorker = <T extends WorkerType>(\n  builder: () => Worker,\n  limit = 4\n) => {\n  let workers: {\n    worker?: Worker;\n    resultResolver?: PromiseWithResolvers<unknown>;\n  }[] = Array(limit)\n    .fill(undefined)\n    .map(() => ({}));\n  const getResolver = async () => {\n    while (true) {\n      const target = workers.find(({ resultResolver }) => !resultResolver);\n      if (target) {\n        target.resultResolver = Promise.withResolvers<unknown>();\n        if (!target.worker) target.worker = await init(builder());\n        return target;\n      }\n      await Promise.race(\n        workers.map(({ resultResolver }) => resultResolver?.promise)\n      );\n    }\n  };\n\n  const execute = async <K extends keyof T>(\n    name: K,\n    ...value: Parameters<T[K]>\n  ): Promise<Awaited<ReturnType<T[K]>>> => {\n    const target = await getResolver();\n    const { resultResolver } = target;\n    if (!resultResolver) throw new Error(\"Unexpected error\");\n    exec(target.worker!, name as string, ...value)\n      .then(resultResolver.resolve)\n      .catch(resultResolver.reject)\n      .finally(() => {\n        target.resultResolver = undefined;\n      });\n    return resultResolver.promise as Promise<Awaited<ReturnType<T[K]>>>;\n  };\n  const waitAll = async () => {\n    while (workers.find(({ resultResolver }) => resultResolver)) {\n      await Promise.all(\n        workers.flatMap(({ resultResolver }) =>\n          resultResolver ? [resultResolver.promise] : []\n        )\n      );\n    }\n  };\n  const close = () => {\n    for (const { worker } of workers) {\n      worker?.terminate();\n    }\n  };\n  const setLimit = (limit: number) => {\n    workers = Array(limit)\n      .fill(undefined)\n      .map(() => ({}));\n  };\n  return { execute, waitAll, close, setLimit };\n};\n/**\n *\n *\n * @template T\n * @param {T} WorkerProc\n * @return {*}\n */\nexport const initWorker = <T extends WorkerType>(WorkerProc: T) => {\n  const worker = parentPort;\n  if (!worker) {\n    throw new Error(\"This is not a worker thread\");\n  }\n  worker.addListener(\"message\", async (data: WorkerRecvEvent<T>) => {\n    if (data.type === \"function\") {\n      const {\n        name,\n        value,\n        callback,\n      }: {\n        name: keyof T;\n        value: unknown[];\n        callback: boolean[];\n      } = data.payload;\n      const proc = WorkerProc[name];\n      if (proc) {\n        try {\n          const params = value.map((v, index) =>\n            callback[index]\n              ? (...params: unknown[]) =>\n                  callbackProc<T>(worker as never, index, params)\n              : v\n          );\n          worker.postMessage({\n            type: \"result\",\n            payload: await proc(...params),\n          });\n        } catch (e) {\n          worker.postMessage({ type: \"error\", payload: String(e) });\n        }\n      }\n    }\n  });\n  worker.postMessage(undefined);\n  return WorkerProc;\n};\n\nconst callbackProc = <T>(worker: Worker, index: number, params: unknown[]) => {\n  const id = WorkerValue.id++;\n  return new Promise((resolve) => {\n    worker.once(\"message\", (data: WorkerRecvEvent<T>) => {\n      if (data.type === \"callback_result\" && data.payload.id === id) {\n        resolve(data.payload.result);\n      }\n    });\n    worker.postMessage({\n      type: \"callback\",\n      payload: { id, index, value: params },\n    });\n  });\n};\n\nconst WorkerValue = { id: 0, promises: {} } as {\n  id: number;\n  promises: { [key: number]: Promise<unknown> };\n};\n"]}

package/dist/esm/node.d.ts

@@ -0,0 +1,27 @@
+import { type Worker } from "node:worker_threads";
+type WorkerType = {
+    [key: string]: (...args: any) => any;
+};
+/**
+ * createWorker
+ *
+ * @template T
+ * @param {() => Worker} builder
+ * @param {number} [limit=0]
+ * @return {*}
+ */
+export declare const createWorker: <T extends WorkerType>(builder: () => Worker, limit?: number) => {
+    execute: <K extends keyof T>(name: K, ...value: Parameters<T[K]>) => Promise<Awaited<ReturnType<T[K]>>>;
+    waitAll: () => Promise<void>;
+    close: () => void;
+    setLimit: (limit: number) => void;
+};
+/**
+ *
+ *
+ * @template T
+ * @param {T} WorkerProc
+ * @return {*}
+ */
+export declare const initWorker: <T extends WorkerType>(WorkerProc: T) => T;
+export {};

package/dist/esm/node.js

@@ -0,0 +1,145 @@
+import { parentPort } from "node:worker_threads";
+const init = (worker) => {
+    return new Promise((resolve) => {
+        worker.once("message", () => {
+            resolve(worker);
+        });
+    });
+};
+const exec = (worker, name, ...value) => {
+    return new Promise((resolve, reject) => {
+        const p = async (data) => {
+            switch (data.type) {
+                case "callback":
+                    const r = value[data.payload.index](data.payload.value);
+                    worker.postMessage({
+                        type: "callback_result",
+                        payload: { id: data.payload.id, result: await r },
+                    });
+                    break;
+                case "result":
+                    worker.removeListener("message", p);
+                    resolve(data.payload);
+                    break;
+                case "error":
+                    worker.removeListener("message", p);
+                    reject(data.payload);
+                    break;
+            }
+        };
+        worker.addListener("message", p);
+        worker.postMessage({
+            type: "function",
+            payload: {
+                name,
+                value: value.map((v) => !(typeof v === "function") && v),
+                callback: value.map((v) => typeof v === "function"),
+            },
+        });
+    });
+};
+/**
+ * createWorker
+ *
+ * @template T
+ * @param {() => Worker} builder
+ * @param {number} [limit=0]
+ * @return {*}
+ */
+export const createWorker = (builder, limit = 4) => {
+    let workers = Array(limit)
+        .fill(undefined)
+        .map(() => ({}));
+    const getResolver = async () => {
+        while (true) {
+            const target = workers.find(({ resultResolver }) => !resultResolver);
+            if (target) {
+                target.resultResolver = Promise.withResolvers();
+                if (!target.worker)
+                    target.worker = await init(builder());
+                return target;
+            }
+            await Promise.race(workers.map(({ resultResolver }) => resultResolver?.promise));
+        }
+    };
+    const execute = async (name, ...value) => {
+        const target = await getResolver();
+        const { resultResolver } = target;
+        if (!resultResolver)
+            throw new Error("Unexpected error");
+        exec(target.worker, name, ...value)
+            .then(resultResolver.resolve)
+            .catch(resultResolver.reject)
+            .finally(() => {
+            target.resultResolver = undefined;
+        });
+        return resultResolver.promise;
+    };
+    const waitAll = async () => {
+        while (workers.find(({ resultResolver }) => resultResolver)) {
+            await Promise.all(workers.flatMap(({ resultResolver }) => resultResolver ? [resultResolver.promise] : []));
+        }
+    };
+    const close = () => {
+        for (const { worker } of workers) {
+            worker?.terminate();
+        }
+    };
+    const setLimit = (limit) => {
+        workers = Array(limit)
+            .fill(undefined)
+            .map(() => ({}));
+    };
+    return { execute, waitAll, close, setLimit };
+};
+/**
+ *
+ *
+ * @template T
+ * @param {T} WorkerProc
+ * @return {*}
+ */
+export const initWorker = (WorkerProc) => {
+    const worker = parentPort;
+    if (!worker) {
+        throw new Error("This is not a worker thread");
+    }
+    worker.addListener("message", async (data) => {
+        if (data.type === "function") {
+            const { name, value, callback, } = data.payload;
+            const proc = WorkerProc[name];
+            if (proc) {
+                try {
+                    const params = value.map((v, index) => callback[index]
+                        ? (...params) => callbackProc(worker, index, params)
+                        : v);
+                    worker.postMessage({
+                        type: "result",
+                        payload: await proc(...params),
+                    });
+                }
+                catch (e) {
+                    worker.postMessage({ type: "error", payload: String(e) });
+                }
+            }
+        }
+    });
+    worker.postMessage(undefined);
+    return WorkerProc;
+};
+const callbackProc = (worker, index, params) => {
+    const id = WorkerValue.id++;
+    return new Promise((resolve) => {
+        worker.once("message", (data) => {
+            if (data.type === "callback_result" && data.payload.id === id) {
+                resolve(data.payload.result);
+            }
+        });
+        worker.postMessage({
+            type: "callback",
+            payload: { id, index, value: params },
+        });
+    });
+};
+const WorkerValue = { id: 0, promises: {} };
+//# sourceMappingURL=node.js.map

package/dist/esm/node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"node.js","sourceRoot":"","sources":["../../src/node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAe,MAAM,qBAAqB,CAAC;AAsB9D,MAAM,IAAI,GAAG,CAAC,MAAc,EAAmB,EAAE;IAC/C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;YAC1B,OAAO,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AACF,MAAM,IAAI,GAAG,CACX,MAAc,EACd,IAAa,EACb,GAAG,KAA6B,EACC,EAAE;IACnC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,CAAC,GAAG,KAAK,EAAE,IAAwB,EAAE,EAAE;YAC3C,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,UAAU;oBACb,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBACxD,MAAM,CAAC,WAAW,CAAC;wBACjB,IAAI,EAAE,iBAAiB;wBACvB,OAAO,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE;qBAClD,CAAC,CAAC;oBACH,MAAM;gBACR,KAAK,QAAQ;oBACX,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;oBACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACtB,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACrB,MAAM;YACV,CAAC;QACH,CAAC,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,WAAW,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE;gBACP,IAAI;gBACJ,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC;gBACjE,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC;aAC7D;SACoB,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,CAC1B,OAAqB,EACrB,KAAK,GAAG,CAAC,EACT,EAAE;IACF,IAAI,OAAO,GAGL,KAAK,CAAC,KAAK,CAAC;SACf,IAAI,CAAC,SAAS,CAAC;SACf,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACnB,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;QAC7B,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC;YACrE,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,EAAW,CAAC;gBACzD,IAAI,CAAC,MAAM,CAAC,MAAM;oBAAE,MAAM,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,MAAM,OAAO,CAAC,IAAI,CAChB,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,EAAE,OAAO,CAAC,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,KAAK,EACnB,IAAO,EACP,GAAG,KAAuB,EACU,EAAE;QACtC,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;QACnC,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAClC,IAAI,CAAC,cAAc;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,MAAO,EAAE,IAAc,EAAE,GAAG,KAAK,CAAC;aAC3C,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;aAC5B,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC;aAC5B,OAAO,CAAC,GAAG,EAAE;YACZ,MAAM,CAAC,cAAc,GAAG,SAAS,CAAC;QACpC,CAAC,CAAC,CAAC;QACL,OAAO,cAAc,CAAC,OAA6C,CAAC;IACtE,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACzB,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5D,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrC,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAC/C,CACF,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IACF,MAAM,KAAK,GAAG,GAAG,EAAE;QACjB,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;YACjC,MAAM,EAAE,SAAS,EAAE,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IACF,MAAM,QAAQ,GAAG,CAAC,KAAa,EAAE,EAAE;QACjC,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC;aACnB,IAAI,CAAC,SAAS,CAAC;aACf,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACrB,CAAC,CAAC;IACF,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;AAC/C,CAAC,CAAC;AACF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,CAAuB,UAAa,EAAE,EAAE;IAChE,MAAM,MAAM,GAAG,UAAU,CAAC;IAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,KAAK,EAAE,IAAwB,EAAE,EAAE;QAC/D,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM,EACJ,IAAI,EACJ,KAAK,EACL,QAAQ,GACT,GAIG,IAAI,CAAC,OAAO,CAAC;YACjB,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAC9B,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CACpC,QAAQ,CAAC,KAAK,CAAC;wBACb,CAAC,CAAC,CAAC,GAAG,MAAiB,EAAE,EAAE,CACvB,YAAY,CAAI,MAAe,EAAE,KAAK,EAAE,MAAM,CAAC;wBACnD,CAAC,CAAC,CAAC,CACN,CAAC;oBACF,MAAM,CAAC,WAAW,CAAC;wBACjB,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,MAAM,IAAI,CAAC,GAAG,MAAM,CAAC;qBAC/B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IACH,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAC9B,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,CAAI,MAAc,EAAE,KAAa,EAAE,MAAiB,EAAE,EAAE;IAC3E,MAAM,EAAE,GAAG,WAAW,CAAC,EAAE,EAAE,CAAC;IAC5B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,IAAwB,EAAE,EAAE;YAClD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC9D,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE;SACtC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAGxC,CAAC","sourcesContent":["import { parentPort, type Worker } from \"node:worker_threads\";\ntype WorkerType = { [key: string]: (...args: any) => any };\ntype WorkerRecvEvent<T> =\n  | {\n      type: \"function\";\n      payload: { name: keyof T; callback: boolean[]; value: unknown[] };\n    }\n  | { type: \"callback_result\"; payload: { id: number; result: unknown } };\ntype WorkerSendEvent<T extends WorkerType> =\n  | {\n      type: \"callback\";\n      payload: { id: number; result: unknown; index: number; value: T };\n    }\n  | {\n      type: \"result\";\n      payload: ReturnType<T[keyof T]>;\n    }\n  | {\n      type: \"error\";\n      payload: unknown;\n    };\n\nconst init = (worker: Worker): Promise<Worker> => {\n  return new Promise((resolve) => {\n    worker.once(\"message\", () => {\n      resolve(worker);\n    });\n  });\n};\nconst exec = <T extends WorkerType>(\n  worker: Worker,\n  name: keyof T,\n  ...value: Parameters<T[keyof T]>\n): Promise<ReturnType<T[keyof T]>> => {\n  return new Promise((resolve, reject) => {\n    const p = async (data: WorkerSendEvent<T>) => {\n      switch (data.type) {\n        case \"callback\":\n          const r = value[data.payload.index](data.payload.value);\n          worker.postMessage({\n            type: \"callback_result\",\n            payload: { id: data.payload.id, result: await r },\n          });\n          break;\n        case \"result\":\n          worker.removeListener(\"message\", p);\n          resolve(data.payload);\n          break;\n        case \"error\":\n          worker.removeListener(\"message\", p);\n          reject(data.payload);\n          break;\n      }\n    };\n    worker.addListener(\"message\", p);\n    worker.postMessage({\n      type: \"function\",\n      payload: {\n        name,\n        value: value.map((v: unknown) => !(typeof v === \"function\") && v),\n        callback: value.map((v: unknown) => typeof v === \"function\"),\n      },\n    } as WorkerRecvEvent<T>);\n  });\n};\n\n/**\n * createWorker\n *\n * @template T\n * @param {() => Worker} builder\n * @param {number} [limit=0]\n * @return {*}\n */\nexport const createWorker = <T extends WorkerType>(\n  builder: () => Worker,\n  limit = 4\n) => {\n  let workers: {\n    worker?: Worker;\n    resultResolver?: PromiseWithResolvers<unknown>;\n  }[] = Array(limit)\n    .fill(undefined)\n    .map(() => ({}));\n  const getResolver = async () => {\n    while (true) {\n      const target = workers.find(({ resultResolver }) => !resultResolver);\n      if (target) {\n        target.resultResolver = Promise.withResolvers<unknown>();\n        if (!target.worker) target.worker = await init(builder());\n        return target;\n      }\n      await Promise.race(\n        workers.map(({ resultResolver }) => resultResolver?.promise)\n      );\n    }\n  };\n\n  const execute = async <K extends keyof T>(\n    name: K,\n    ...value: Parameters<T[K]>\n  ): Promise<Awaited<ReturnType<T[K]>>> => {\n    const target = await getResolver();\n    const { resultResolver } = target;\n    if (!resultResolver) throw new Error(\"Unexpected error\");\n    exec(target.worker!, name as string, ...value)\n      .then(resultResolver.resolve)\n      .catch(resultResolver.reject)\n      .finally(() => {\n        target.resultResolver = undefined;\n      });\n    return resultResolver.promise as Promise<Awaited<ReturnType<T[K]>>>;\n  };\n  const waitAll = async () => {\n    while (workers.find(({ resultResolver }) => resultResolver)) {\n      await Promise.all(\n        workers.flatMap(({ resultResolver }) =>\n          resultResolver ? [resultResolver.promise] : []\n        )\n      );\n    }\n  };\n  const close = () => {\n    for (const { worker } of workers) {\n      worker?.terminate();\n    }\n  };\n  const setLimit = (limit: number) => {\n    workers = Array(limit)\n      .fill(undefined)\n      .map(() => ({}));\n  };\n  return { execute, waitAll, close, setLimit };\n};\n/**\n *\n *\n * @template T\n * @param {T} WorkerProc\n * @return {*}\n */\nexport const initWorker = <T extends WorkerType>(WorkerProc: T) => {\n  const worker = parentPort;\n  if (!worker) {\n    throw new Error(\"This is not a worker thread\");\n  }\n  worker.addListener(\"message\", async (data: WorkerRecvEvent<T>) => {\n    if (data.type === \"function\") {\n      const {\n        name,\n        value,\n        callback,\n      }: {\n        name: keyof T;\n        value: unknown[];\n        callback: boolean[];\n      } = data.payload;\n      const proc = WorkerProc[name];\n      if (proc) {\n        try {\n          const params = value.map((v, index) =>\n            callback[index]\n              ? (...params: unknown[]) =>\n                  callbackProc<T>(worker as never, index, params)\n              : v\n          );\n          worker.postMessage({\n            type: \"result\",\n            payload: await proc(...params),\n          });\n        } catch (e) {\n          worker.postMessage({ type: \"error\", payload: String(e) });\n        }\n      }\n    }\n  });\n  worker.postMessage(undefined);\n  return WorkerProc;\n};\n\nconst callbackProc = <T>(worker: Worker, index: number, params: unknown[]) => {\n  const id = WorkerValue.id++;\n  return new Promise((resolve) => {\n    worker.once(\"message\", (data: WorkerRecvEvent<T>) => {\n      if (data.type === \"callback_result\" && data.payload.id === id) {\n        resolve(data.payload.result);\n      }\n    });\n    worker.postMessage({\n      type: \"callback\",\n      payload: { id, index, value: params },\n    });\n  });\n};\n\nconst WorkerValue = { id: 0, promises: {} } as {\n  id: number;\n  promises: { [key: number]: Promise<unknown> };\n};\n"]}

package/package.json

@@ -1,17 +1,35 @@
 {
   "name": "worker-lib",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "main": "./dist/cjs/index.js",
   "types": "./dist/cjs/index.d.ts",
   "exports": {
-    "require": "./dist/cjs/index.js",
-    "import": "./dist/esm/index.js"
+    ".": {
+      "types": "./dist/cjs/index.d.ts",
+      "require": "./dist/cjs/index.js",
+      "import": "./dist/esm/index.js"
+    },
+    "./node": {
+      "type": "./dist/cjs/node.d.ts",
+      "default": "./dist/cjs/node.js"
+    }
+  },
+  "typesVersions": {
+    "*": {
+      "*": [
+        "./dist/cjs/index.d.ts"
+      ],
+      "node": [
+        "./dist/cjs/node.d.ts"
+      ]
+    }
   },
   "license": "MIT",
   "scripts": {
     "build": "tsc && tsc --project ./tsconfig.esm.json && cpy esm dist"
   },
   "devDependencies": {
+    "@types/node": "22.13.5",
     "cpy-cli": "5.0.0",
     "typescript": "^5.7.3"
   },

package/dist/tsconfig.tsbuildinfo

@@ -1 +0,0 @@
-{"root":["../src/index.ts"],"version":"5.7.3"}