@rivetkit/workflow-engine 0.0.0-sqlite-query-metrics-v2.f5d3605 → 0.0.0-sqlite-sync-3-0-1-release.95f6e2d

package/dist/tsup/{chunk-XRITRTKT.cjs → chunk-AO3IZAQ4.cjs}

@@ -1994,6 +1994,37 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
   markVisited(key) {
     this.visitedKeys.add(key);
   }
+  /**
+   * Mark all retained history under `location` as visited.
+   *
+   * Replay does not re-enter a settled scope, but its entries stay in history.
+   * Without this, an enclosing validateComplete or getVersion treats them as
+   * leftovers from changed workflow code.
+   */
+  markSettledHistoryVisited(location) {
+    for (const [key, entry] of this.storage.history.entries) {
+      if (isLocationPrefix(location, entry.location)) {
+        this.markVisited(key);
+      }
+    }
+  }
+  /**
+   * Mark the retained history of a resumed loop's finished iterations as
+   * visited, because the loop continues from `resumeIteration` without
+   * replaying them.
+   */
+  markFinishedIterationsVisited(loopLocation, resumeIteration) {
+    const loopSegment = loopLocation[loopLocation.length - 1];
+    for (const [key, entry] of this.storage.history.entries) {
+      if (!isLocationPrefix(loopLocation, entry.location)) {
+        continue;
+      }
+      const iterationSegment = entry.location[loopLocation.length];
+      if (iterationSegment && typeof iterationSegment !== "number" && iterationSegment.loop === loopSegment && iterationSegment.iteration < resumeIteration) {
+        this.markVisited(key);
+      }
+    }
+  }
   /**
    * Check if a name has already been used at the current location in this execution.
    * Throws HistoryDivergedError if duplicate detected.
@@ -2521,6 +2552,9 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
         this.driver,
         existing.id
       );
+      if (metadata.status === "completed" || loopData.output !== void 0) {
+        this.markSettledHistoryVisited(location);
+      }
       if (rollbackMode) {
         if (loopData.output !== void 0) {
           return loopData.output;
@@ -2538,6 +2572,7 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
       entry = existing;
       state = loopData.state;
       iteration = loopData.iteration;
+      this.markFinishedIterationsVisited(location, iteration);
       if (rollbackMode) {
         rollbackOutput = loopData.output;
         rollbackIterationRan = rollbackOutput !== void 0;
@@ -3208,19 +3243,21 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
             `Expected join branch "${branchName}" in "${name}"`
           );
         }
+        const branchLocation = appendName(
+          this.storage,
+          location,
+          branchName
+        );
         if (branchStatus.status === "completed") {
+          this.markSettledHistoryVisited(branchLocation);
           results[branchName] = branchStatus.output;
           return;
         }
         if (branchStatus.status === "failed") {
+          this.markSettledHistoryVisited(branchLocation);
           errors[branchName] = new Error(branchStatus.error);
           return;
         }
-        const branchLocation = appendName(
-          this.storage,
-          location,
-          branchName
-        );
         const branchCtx = this.createBranch(branchLocation);
         branchStatus.status = "running";
         branchStatus.error = void 0;
@@ -3312,6 +3349,7 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
       entry = existing;
       const raceKind = existing.kind;
       if (raceKind.data.winner !== null) {
+        this.markSettledHistoryVisited(location);
         const winnerStatus = raceKind.data.branches[raceKind.data.winner];
         return {
           winner: raceKind.data.winner,
@@ -3370,6 +3408,7 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
       }
     }
     if (hasWinner && winnerName !== null) {
+      this.markSettledHistoryVisited(location);
       return { winner: winnerName, value: winnerValue };
     }
     for (const branch of branches) {
@@ -3379,14 +3418,15 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
           `Expected race branch "${branch.name}" in "${name}"`
         );
       }
-      if (branchStatus.status !== "pending" && branchStatus.status !== "running") {
-        continue;
-      }
       const branchLocation = appendName(
         this.storage,
         location,
         branch.name
       );
+      if (branchStatus.status !== "pending" && branchStatus.status !== "running") {
+        this.markSettledHistoryVisited(branchLocation);
+        continue;
+      }
       const branchCtx = this.createBranch(
         branchLocation,
         raceAbortController
@@ -3394,7 +3434,12 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
       branchStatus.status = "running";
       branchStatus.error = void 0;
       entry.dirty = true;
-      const branchPromise = branch.run(branchCtx).then(
+      const runBranch = async () => {
+        const output = await branch.run(branchCtx);
+        branchCtx.validateComplete();
+        return output;
+      };
+      const branchPromise = runBranch().then(
         async (output) => {
           if (hasWinner) {
             branchStatus.status = "completed";
@@ -3413,7 +3458,6 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
           hasWinner = true;
           winnerName = branch.name;
           winnerValue = output;
-          branchCtx.validateComplete();
           branchStatus.status = "completed";
           branchStatus.output = output;
           branchStatus.error = void 0;
@@ -3548,6 +3592,7 @@ var WorkflowContextImpl = (_class = class _WorkflowContextImpl {
           `Expected ${originalType} or removed at ${key}, found ${existing.kind.type}`
         );
       }
+      this.markSettledHistoryVisited(location);
       return;
     }
     const entry = createEntry(location, {
@@ -4402,4 +4447,4 @@ async function executeWorkflow(workflowId, workflowFn, input, driver, messageDri
 
 
 exports.extractErrorInfo = extractErrorInfo; exports.CriticalError = CriticalError; exports.RollbackError = RollbackError; exports.RollbackCheckpointError = RollbackCheckpointError; exports.SleepError = SleepError; exports.MessageWaitError = MessageWaitError; exports.EvictedError = EvictedError; exports.HistoryDivergedError = HistoryDivergedError; exports.StepExhaustedError = StepExhaustedError; exports.StepFailedError = StepFailedError; exports.JoinError = JoinError; exports.RaceError = RaceError; exports.CancelledError = CancelledError; exports.EntryInProgressError = EntryInProgressError; exports.keyStartsWith = keyStartsWith; exports.compareKeys = compareKeys; exports.keyToHex = keyToHex; exports.isLoopIterationMarker = isLoopIterationMarker; exports.registerName = registerName; exports.resolveName = resolveName; exports.locationToKey = locationToKey; exports.appendName = appendName; exports.appendLoopIteration = appendLoopIteration; exports.emptyLocation = emptyLocation; exports.parentLocation = parentLocation; exports.isLocationPrefix = isLocationPrefix; exports.locationsEqual = locationsEqual; exports.createStorage = createStorage; exports.createHistorySnapshot = createHistorySnapshot; exports.generateId = generateId; exports.createEntry = createEntry; exports.getOrCreateMetadata = getOrCreateMetadata; exports.loadStorage = loadStorage; exports.loadMetadata = loadMetadata; exports.flush = flush; exports.deleteEntriesWithPrefix = deleteEntriesWithPrefix; exports.getEntry = getEntry; exports.setEntry = setEntry; exports.DEFAULT_MAX_RETRIES = DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_STEP_TIMEOUT = DEFAULT_STEP_TIMEOUT; exports.WorkflowContextImpl = WorkflowContextImpl; exports.sleep = sleep; exports.Loop = Loop; exports.runWorkflow = runWorkflow; exports.replayWorkflowFromStep = replayWorkflowFromStep;
-//# sourceMappingURL=chunk-XRITRTKT.cjs.map
+//# sourceMappingURL=chunk-AO3IZAQ4.cjs.map