@rivetkit/workflow-engine 2.3.17 → 2.3.18-rc.1

package/dist/tsup/{chunk-KLHCG2KY.cjs → chunk-32422MOM.cjs}

@@ -1437,6 +1437,7 @@ function deserializeName(bytes) {
 // src/storage.ts
 var MAX_KV_BATCH_ENTRIES = 128;
 var MAX_KV_BATCH_PAYLOAD_BYTES = 976 * 1024;
+var MAX_CONCURRENT_DELETES = 64;
 function createStorage() {
   return {
     nameRegistry: [],
@@ -1616,18 +1617,8 @@ async function flush(storage, driver, onHistoryUpdated, pendingDeletions) {
     }
   }
   if (pendingDeletions) {
-    const deleteOps = [];
-    for (const prefix of pendingDeletions.prefixes) {
-      deleteOps.push(driver.deletePrefix(prefix));
-    }
-    for (const range of pendingDeletions.ranges) {
-      deleteOps.push(driver.deleteRange(range.start, range.end));
-    }
-    for (const key of pendingDeletions.keys) {
-      deleteOps.push(driver.delete(key));
-    }
-    if (deleteOps.length > 0) {
-      await Promise.all(deleteOps);
+    const didChange = await runDeletes(driver, pendingDeletions);
+    if (didChange) {
       historyUpdated = true;
     }
   }
@@ -1669,10 +1660,36 @@ function splitBatchWrites(writes) {
   }
   return chunks;
 }
+function splitBatchDeletes(keys) {
+  const chunks = [];
+  for (let i = 0; i < keys.length; i += MAX_KV_BATCH_ENTRIES) {
+    chunks.push(keys.slice(i, i + MAX_KV_BATCH_ENTRIES));
+  }
+  return chunks;
+}
+async function runDeletes(driver, deletions) {
+  const ops = [
+    ...deletions.prefixes.map((prefix) => () => driver.deletePrefix(prefix)),
+    ...deletions.ranges.map(
+      (range) => () => driver.deleteRange(range.start, range.end)
+    ),
+    ...splitBatchDeletes(deletions.keys).map(
+      (chunk) => () => driver.batchDelete(chunk)
+    )
+  ];
+  if (ops.length === 0) {
+    return false;
+  }
+  for (let i = 0; i < ops.length; i += MAX_CONCURRENT_DELETES) {
+    await Promise.all(
+      ops.slice(i, i + MAX_CONCURRENT_DELETES).map((op) => op())
+    );
+  }
+  return true;
+}
 async function deleteEntriesWithPrefix(storage, driver, prefixLocation, onHistoryUpdated) {
   const deletions = collectDeletionsForPrefix(storage, prefixLocation);
-  await driver.deletePrefix(deletions.prefixes[0]);
-  await Promise.all(deletions.keys.map((key) => driver.delete(key)));
+  await runDeletes(driver, deletions);
   if (deletions.keys.length > 0 && onHistoryUpdated) {
     onHistoryUpdated();
   }
@@ -4321,4 +4338,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-KLHCG2KY.cjs.map
+//# sourceMappingURL=chunk-32422MOM.cjs.map