@danielsimonjr/memoryjs 2.8.0 → 2.8.1

package/dist/index.cjs

@@ -3444,6 +3444,10 @@ var init_WorkerTaskManager = __esm({
       touchedPoolIds = /* @__PURE__ */ new Set();
       cancelledTaskIds = /* @__PURE__ */ new Set();
       handleStatus = /* @__PURE__ */ new Map();
+      // Live `pool.exec(...)` promises by task id. Populated when a task starts
+      // running so the handle can call `.cancel()` on the workerpool promise
+      // for mid-execution cancellation. Cleared on completion / failure.
+      liveExecPromises = /* @__PURE__ */ new Map();
       constructor(opts = {}) {
         this.queue = new TaskQueue({
           concurrency: opts.concurrency,
@@ -3483,7 +3487,13 @@ var init_WorkerTaskManager = __esm({
             this.handleStatus.set(id, "running" /* RUNNING */);
             const pool = this.poolManager.getPool(input.workerType, opts.poolConfig);
             this.touchedPoolIds.add(input.workerType);
-            return await pool.exec(input.methodName, input.args);
+            const execPromise = pool.exec(input.methodName, input.args);
+            this.liveExecPromises.set(id, execPromise);
+            try {
+              return await execPromise;
+            } finally {
+              this.liveExecPromises.delete(id);
+            }
           }
         };
         const result = this.queue.enqueue(task).then(async (res) => {
@@ -3505,8 +3515,21 @@ var init_WorkerTaskManager = __esm({
             const evicted = this.queue.cancel(id);
             if (evicted) {
               this.handleStatus.set(id, "cancelled" /* CANCELLED */);
+              return true;
+            }
+            const live = this.liveExecPromises.get(id);
+            if (live && typeof live.cancel === "function") {
+              if (live.pending === false) return false;
+              try {
+                live.cancel();
+                this.handleStatus.set(id, "cancelled" /* CANCELLED */);
+                return true;
+              } catch (e) {
+                logger.warn(`[WorkerTaskManager] cancel() on live exec for ${id} threw:`, e);
+                return false;
+              }
             }
-            return evicted;
+            return false;
           },
           status: () => this.handleStatus.get(id) ?? "pending" /* PENDING */
         };