@rivetkit/workflow-engine 0.0.0-06-12-chore-rivetkit-wasm-re-enable-wasm-build.4be04fe → 0.0.0-06-09-refactor-rivetkit-split-workflow-context-into-workflowcontext-workflowstepcontext.e0c9540

package/dist/tsup/{chunk-I4AXJJGO.js → chunk-DFNXCZ47.js}

@@ -1664,6 +1664,30 @@ function setEntry(storage, location, entry) {
   storage.history.entries.set(key, entry);
 }
 
+// src/utils.ts
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+var TIMEOUT_MAX = 2147483647;
+function setLongTimeout(listener, after) {
+  let timeout;
+  function start(remaining) {
+    if (remaining <= TIMEOUT_MAX) {
+      timeout = setTimeout(listener, remaining);
+    } else {
+      timeout = setTimeout(() => {
+        start(remaining - TIMEOUT_MAX);
+      }, TIMEOUT_MAX);
+    }
+  }
+  start(after);
+  return {
+    abort: () => {
+      if (timeout !== void 0) clearTimeout(timeout);
+    }
+  };
+}
+
 // src/context.ts
 var DEFAULT_MAX_RETRIES = 3;
 var DEFAULT_RETRY_BACKOFF_BASE = 100;
@@ -2015,34 +2039,20 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
    * The event listener uses { once: true } to auto-remove after firing,
    * preventing memory leaks if this method is called multiple times.
    */
-  /**
-   * Wait for `ms`, rejecting early with EvictedError if the workflow is
-   * evicted. Both the timer and the abort listener are torn down on either
-   * outcome, so a completed sleep never leaves a dangling listener on the
-   * long-lived run abort signal.
-   */
-  async sleepOrEvict(ms) {
-    if (this.abortSignal.aborted) {
-      throw new EvictedError();
-    }
-    let timer;
-    let onAbort;
-    try {
-      await new Promise((resolve, reject) => {
-        timer = setTimeout(resolve, ms);
-        onAbort = () => reject(new EvictedError());
-        this.abortSignal.addEventListener("abort", onAbort, {
-          once: true
-        });
-      });
-    } finally {
-      if (timer !== void 0) {
-        clearTimeout(timer);
-      }
-      if (onAbort) {
-        this.abortSignal.removeEventListener("abort", onAbort);
+  waitForEviction() {
+    return new Promise((_, reject) => {
+      if (this.abortSignal.aborted) {
+        reject(new EvictedError());
+        return;
       }
-    }
+      this.abortSignal.addEventListener(
+        "abort",
+        () => {
+          reject(new EvictedError());
+        },
+        { once: true }
+      );
+    });
   }
   // === Step ===
   async step(nameOrConfig, run) {
@@ -2685,7 +2695,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
       return;
     }
     if (remaining < this.driver.workerPollInterval) {
-      await this.sleepOrEvict(remaining);
+      await Promise.race([sleep(remaining), this.waitForEviction()]);
       this.checkEvicted();
       if (entry.kind.type === "sleep") {
         entry.kind.data.state = "completed";
@@ -3496,30 +3506,6 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
   }
 };
 
-// src/utils.ts
-function sleep(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-var TIMEOUT_MAX = 2147483647;
-function setLongTimeout(listener, after) {
-  let timeout;
-  function start(remaining) {
-    if (remaining <= TIMEOUT_MAX) {
-      timeout = setTimeout(listener, remaining);
-    } else {
-      timeout = setTimeout(() => {
-        start(remaining - TIMEOUT_MAX);
-      }, TIMEOUT_MAX);
-    }
-  }
-  start(after);
-  return {
-    abort: () => {
-      if (timeout !== void 0) clearTimeout(timeout);
-    }
-  };
-}
-
 // src/index.ts
 var Loop = {
   continue: (state) => ({
@@ -3743,41 +3729,25 @@ async function executeLiveWorkflow(workflowId, workflowFn, input, driver, messag
     const hasMessages = result.waitingForMessages !== void 0;
     const hasDeadline = result.sleepUntil !== void 0;
     if (hasMessages && hasDeadline) {
-      const iterationAbort = new AbortController();
-      const onRunAbort = () => iterationAbort.abort();
-      if (abortController.signal.aborted) {
-        iterationAbort.abort();
-      } else {
-        abortController.signal.addEventListener("abort", onRunAbort, {
-          once: true
-        });
-      }
-      const messagePromise = awaitWithEviction(
-        driver.waitForMessages(
-          result.waitingForMessages,
-          iterationAbort.signal
-        ),
-        iterationAbort.signal
-      );
-      const sleepPromise = waitForSleep(
-        runtime,
-        result.sleepUntil,
-        iterationAbort.signal
-      );
-      messagePromise.catch(() => {
-      });
-      sleepPromise.catch(() => {
-      });
       try {
+        const messagePromise = awaitWithEviction(
+          driver.waitForMessages(
+            result.waitingForMessages,
+            abortController.signal
+          ),
+          abortController.signal
+        );
+        const sleepPromise = waitForSleep(
+          runtime,
+          result.sleepUntil,
+          abortController.signal
+        );
         await Promise.race([messagePromise, sleepPromise]);
       } catch (error) {
         if (error instanceof EvictedError) {
           return lastResult;
         }
         throw error;
-      } finally {
-        iterationAbort.abort();
-        abortController.signal.removeEventListener("abort", onRunAbort);
       }
       continue;
     }
@@ -4220,15 +4190,15 @@ export {
   deleteEntriesWithPrefix,
   getEntry,
   setEntry,
+  sleep,
   DEFAULT_MAX_RETRIES,
   DEFAULT_RETRY_BACKOFF_BASE,
   DEFAULT_RETRY_BACKOFF_MAX,
   DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL,
   DEFAULT_STEP_TIMEOUT,
   WorkflowContextImpl,
-  sleep,
   Loop,
   runWorkflow,
   replayWorkflowFromStep
 };
-//# sourceMappingURL=chunk-I4AXJJGO.js.map
+//# sourceMappingURL=chunk-DFNXCZ47.js.map