@rivetkit/workflow-engine 2.3.4 → 2.3.5

package/dist/tsup/index.d.cts

@@ -571,6 +571,11 @@ interface KVWrite {
  * See architecture.md "Isolation Model" for details.
  */
 interface EngineDriver {
+    /**
+     * Requires each logical storage flush to reach `batch` as one indivisible
+     * unit. The driver must reject an oversized unit instead of splitting it.
+     */
+    readonly atomicBatch?: boolean;
     /**
      * Get a value by key.
      * Returns null if the key doesn't exist.

package/dist/tsup/index.d.ts

@@ -571,6 +571,11 @@ interface KVWrite {
  * See architecture.md "Isolation Model" for details.
  */
 interface EngineDriver {
+    /**
+     * Requires each logical storage flush to reach `batch` as one indivisible
+     * unit. The driver must reject an oversized unit instead of splitting it.
+     */
+    readonly atomicBatch?: boolean;
     /**
      * Get a value by key.
      * Returns null if the key doesn't exist.

package/dist/tsup/index.js

@@ -43,7 +43,7 @@ import {
   resolveName,
   runWorkflow,
   setEntry
-} from "./chunk-Z3WC5MXC.js";
+} from "./chunk-DTS4S7RT.js";
 export {
   CancelledError,
   CriticalError,

package/dist/tsup/testing.cjs

@@ -47,7 +47,7 @@
 
 
 
-var _chunkOCUQRGRScjs = require('./chunk-OCUQRGRS.cjs');
+var _chunkKLHCG2KYcjs = require('./chunk-KLHCG2KY.cjs');
 
 // src/testing.ts
 var InMemoryWorkflowMessageDriver = class {
@@ -97,7 +97,7 @@ var InMemoryWorkflowMessageDriver = class {
   }
   async waitForMessages(messageNames, abortSignal) {
     if (abortSignal.aborted) {
-      throw new (0, _chunkOCUQRGRScjs.EvictedError)();
+      throw new (0, _chunkKLHCG2KYcjs.EvictedError)();
     }
     const nameSet = messageNames.length > 0 ? new Set(messageNames) : void 0;
     if (this.#messages.some(
@@ -118,7 +118,7 @@ var InMemoryWorkflowMessageDriver = class {
         },
         abortSignal,
         onAbort: () => {
-          waiter.reject(new (0, _chunkOCUQRGRScjs.EvictedError)());
+          waiter.reject(new (0, _chunkKLHCG2KYcjs.EvictedError)());
         }
       };
       abortSignal.addEventListener("abort", waiter.onAbort, {
@@ -158,56 +158,56 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
   __init4() {this.workerPollInterval = 100}
   __init5() {this.messageDriver = this.#inMemoryMessageDriver}
   async get(key) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
-    const entry = this.kv.get(_chunkOCUQRGRScjs.keyToHex.call(void 0, key));
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
+    const entry = this.kv.get(_chunkKLHCG2KYcjs.keyToHex.call(void 0, key));
     return _nullishCoalesce((entry == null ? void 0 : entry.value), () => ( null));
   }
   async set(key, value) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
-    this.kv.set(_chunkOCUQRGRScjs.keyToHex.call(void 0, key), { key, value });
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
+    this.kv.set(_chunkKLHCG2KYcjs.keyToHex.call(void 0, key), { key, value });
   }
   async delete(key) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
-    this.kv.delete(_chunkOCUQRGRScjs.keyToHex.call(void 0, key));
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
+    this.kv.delete(_chunkKLHCG2KYcjs.keyToHex.call(void 0, key));
   }
   async deletePrefix(prefix) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkOCUQRGRScjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+      if (_chunkKLHCG2KYcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
         this.kv.delete(hexKey);
       }
     }
   }
   async deleteRange(start, end) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkOCUQRGRScjs.compareKeys.call(void 0, entry.key, start) >= 0 && _chunkOCUQRGRScjs.compareKeys.call(void 0, entry.key, end) < 0) {
+      if (_chunkKLHCG2KYcjs.compareKeys.call(void 0, entry.key, start) >= 0 && _chunkKLHCG2KYcjs.compareKeys.call(void 0, entry.key, end) < 0) {
         this.kv.delete(hexKey);
       }
     }
   }
   async list(prefix) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     const results = [];
     for (const entry of this.kv.values()) {
-      if (_chunkOCUQRGRScjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+      if (_chunkKLHCG2KYcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
         results.push({ key: entry.key, value: entry.value });
       }
     }
-    return results.sort((a, b) => _chunkOCUQRGRScjs.compareKeys.call(void 0, a.key, b.key));
+    return results.sort((a, b) => _chunkKLHCG2KYcjs.compareKeys.call(void 0, a.key, b.key));
   }
   async batch(writes) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     for (const { key, value } of writes) {
-      this.kv.set(_chunkOCUQRGRScjs.keyToHex.call(void 0, key), { key, value });
+      this.kv.set(_chunkKLHCG2KYcjs.keyToHex.call(void 0, key), { key, value });
     }
   }
   async setAlarm(workflowId, wakeAt) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     this.alarms.set(workflowId, wakeAt);
   }
   async clearAlarm(workflowId) {
-    await _chunkOCUQRGRScjs.sleep.call(void 0, this.latency);
+    await _chunkKLHCG2KYcjs.sleep.call(void 0, this.latency);
     this.alarms.delete(workflowId);
   }
   async waitForMessages(messageNames, abortSignal) {
@@ -218,7 +218,7 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
     }
     while (true) {
       if (abortSignal.aborted) {
-        throw new (0, _chunkOCUQRGRScjs.EvictedError)();
+        throw new (0, _chunkKLHCG2KYcjs.EvictedError)();
       }
       const messages = await this.messageDriver.receiveMessages({
         names: messageNames.length > 0 ? messageNames : void 0,
@@ -228,7 +228,7 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
       if (messages.length > 0) {
         return;
       }
-      await _chunkOCUQRGRScjs.sleep.call(void 0, Math.max(1, this.latency));
+      await _chunkKLHCG2KYcjs.sleep.call(void 0, Math.max(1, this.latency));
     }
   }
   /**
@@ -324,5 +324,5 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
 
 
 
-exports.CancelledError = _chunkOCUQRGRScjs.CancelledError; exports.CriticalError = _chunkOCUQRGRScjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkOCUQRGRScjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkOCUQRGRScjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkOCUQRGRScjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkOCUQRGRScjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkOCUQRGRScjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkOCUQRGRScjs.EntryInProgressError; exports.EvictedError = _chunkOCUQRGRScjs.EvictedError; exports.HistoryDivergedError = _chunkOCUQRGRScjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkOCUQRGRScjs.JoinError; exports.Loop = _chunkOCUQRGRScjs.Loop; exports.MessageWaitError = _chunkOCUQRGRScjs.MessageWaitError; exports.RaceError = _chunkOCUQRGRScjs.RaceError; exports.RollbackCheckpointError = _chunkOCUQRGRScjs.RollbackCheckpointError; exports.RollbackError = _chunkOCUQRGRScjs.RollbackError; exports.SleepError = _chunkOCUQRGRScjs.SleepError; exports.StepExhaustedError = _chunkOCUQRGRScjs.StepExhaustedError; exports.StepFailedError = _chunkOCUQRGRScjs.StepFailedError; exports.WorkflowContextImpl = _chunkOCUQRGRScjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkOCUQRGRScjs.appendLoopIteration; exports.appendName = _chunkOCUQRGRScjs.appendName; exports.createEntry = _chunkOCUQRGRScjs.createEntry; exports.createHistorySnapshot = _chunkOCUQRGRScjs.createHistorySnapshot; exports.createStorage = _chunkOCUQRGRScjs.createStorage; exports.deleteEntriesWithPrefix = _chunkOCUQRGRScjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkOCUQRGRScjs.emptyLocation; exports.extractErrorInfo = _chunkOCUQRGRScjs.extractErrorInfo; exports.flush = _chunkOCUQRGRScjs.flush; exports.generateId = _chunkOCUQRGRScjs.generateId; exports.getEntry = _chunkOCUQRGRScjs.getEntry; exports.getOrCreateMetadata = _chunkOCUQRGRScjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkOCUQRGRScjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkOCUQRGRScjs.isLoopIterationMarker; exports.loadMetadata = _chunkOCUQRGRScjs.loadMetadata; exports.loadStorage = _chunkOCUQRGRScjs.loadStorage; exports.locationToKey = _chunkOCUQRGRScjs.locationToKey; exports.locationsEqual = _chunkOCUQRGRScjs.locationsEqual; exports.parentLocation = _chunkOCUQRGRScjs.parentLocation; exports.registerName = _chunkOCUQRGRScjs.registerName; exports.replayWorkflowFromStep = _chunkOCUQRGRScjs.replayWorkflowFromStep; exports.resolveName = _chunkOCUQRGRScjs.resolveName; exports.runWorkflow = _chunkOCUQRGRScjs.runWorkflow; exports.setEntry = _chunkOCUQRGRScjs.setEntry;
+exports.CancelledError = _chunkKLHCG2KYcjs.CancelledError; exports.CriticalError = _chunkKLHCG2KYcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkKLHCG2KYcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkKLHCG2KYcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkKLHCG2KYcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkKLHCG2KYcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkKLHCG2KYcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkKLHCG2KYcjs.EntryInProgressError; exports.EvictedError = _chunkKLHCG2KYcjs.EvictedError; exports.HistoryDivergedError = _chunkKLHCG2KYcjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkKLHCG2KYcjs.JoinError; exports.Loop = _chunkKLHCG2KYcjs.Loop; exports.MessageWaitError = _chunkKLHCG2KYcjs.MessageWaitError; exports.RaceError = _chunkKLHCG2KYcjs.RaceError; exports.RollbackCheckpointError = _chunkKLHCG2KYcjs.RollbackCheckpointError; exports.RollbackError = _chunkKLHCG2KYcjs.RollbackError; exports.SleepError = _chunkKLHCG2KYcjs.SleepError; exports.StepExhaustedError = _chunkKLHCG2KYcjs.StepExhaustedError; exports.StepFailedError = _chunkKLHCG2KYcjs.StepFailedError; exports.WorkflowContextImpl = _chunkKLHCG2KYcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkKLHCG2KYcjs.appendLoopIteration; exports.appendName = _chunkKLHCG2KYcjs.appendName; exports.createEntry = _chunkKLHCG2KYcjs.createEntry; exports.createHistorySnapshot = _chunkKLHCG2KYcjs.createHistorySnapshot; exports.createStorage = _chunkKLHCG2KYcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkKLHCG2KYcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkKLHCG2KYcjs.emptyLocation; exports.extractErrorInfo = _chunkKLHCG2KYcjs.extractErrorInfo; exports.flush = _chunkKLHCG2KYcjs.flush; exports.generateId = _chunkKLHCG2KYcjs.generateId; exports.getEntry = _chunkKLHCG2KYcjs.getEntry; exports.getOrCreateMetadata = _chunkKLHCG2KYcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkKLHCG2KYcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkKLHCG2KYcjs.isLoopIterationMarker; exports.loadMetadata = _chunkKLHCG2KYcjs.loadMetadata; exports.loadStorage = _chunkKLHCG2KYcjs.loadStorage; exports.locationToKey = _chunkKLHCG2KYcjs.locationToKey; exports.locationsEqual = _chunkKLHCG2KYcjs.locationsEqual; exports.parentLocation = _chunkKLHCG2KYcjs.parentLocation; exports.registerName = _chunkKLHCG2KYcjs.registerName; exports.replayWorkflowFromStep = _chunkKLHCG2KYcjs.replayWorkflowFromStep; exports.resolveName = _chunkKLHCG2KYcjs.resolveName; exports.runWorkflow = _chunkKLHCG2KYcjs.runWorkflow; exports.setEntry = _chunkKLHCG2KYcjs.setEntry;
 //# sourceMappingURL=testing.cjs.map

package/dist/tsup/testing.js

@@ -47,7 +47,7 @@ import {
   runWorkflow,
   setEntry,
   sleep
-} from "./chunk-Z3WC5MXC.js";
+} from "./chunk-DTS4S7RT.js";
 
 // src/testing.ts
 var InMemoryWorkflowMessageDriver = class {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/workflow-engine",
-	"version": "2.3.4",
+	"version": "2.3.5",
 	"description": "Durable workflow engine with reentrant execution",
 	"license": "Apache-2.0",
 	"keywords": [

package/src/driver.ts

@@ -30,6 +30,12 @@ export interface KVWrite {
  * See architecture.md "Isolation Model" for details.
  */
 export interface EngineDriver {
+	/**
+	 * Requires each logical storage flush to reach `batch` as one indivisible
+	 * unit. The driver must reject an oversized unit instead of splitting it.
+	 */
+	readonly atomicBatch?: boolean;
+
 	// === KV Operations ===
 
 	/**

package/src/storage.ts

@@ -318,8 +318,12 @@ export async function flush(
 	}
 
 	if (writes.length > 0) {
-		for (const chunk of splitBatchWrites(writes)) {
-			await driver.batch(chunk);
+		if (driver.atomicBatch) {
+			await driver.batch(writes);
+		} else {
+			for (const chunk of splitBatchWrites(writes)) {
+				await driver.batch(chunk);
+			}
 		}
 	}