UNPKG

@mondaydotcomorg/atp-compiler

Version:

Production-ready compiler for transforming async iteration patterns into resumable operations with checkpoint-based state management

61 lines 2.34 kB
import { getCheckpointManager } from './checkpoint-manager.js'; import { InfiniteLoopDetectionError } from './errors.js'; const MAX_ITERATIONS = 1000000; export async function resumableForOf(items, callback, loopId) { const checkpointManager = getCheckpointManager(); const checkpoint = await checkpointManager.load(loopId); const startIndex = checkpoint?.currentIndex || 0; for (let i = startIndex; i < items.length; i++) { if (i > MAX_ITERATIONS) { throw new InfiniteLoopDetectionError(loopId, i); } await callback(items[i], i); const newCheckpoint = { loopId, currentIndex: i + 1, timestamp: Date.now(), }; await checkpointManager.save(newCheckpoint); } await checkpointManager.clear(loopId); } export async function resumableWhile(conditionFn, bodyFn, loopId) { const checkpointManager = getCheckpointManager(); const checkpoint = await checkpointManager.load(loopId); let iteration = checkpoint?.currentIndex || 0; while (await Promise.resolve(conditionFn())) { if (iteration > MAX_ITERATIONS) { throw new InfiniteLoopDetectionError(loopId, iteration); } await bodyFn(iteration); const newCheckpoint = { loopId, currentIndex: iteration + 1, timestamp: Date.now(), }; await checkpointManager.save(newCheckpoint); iteration++; } await checkpointManager.clear(loopId); } export async function resumableForLoop(initValue, conditionFn, incrementFn, bodyFn, loopId) { const checkpointManager = getCheckpointManager(); const checkpoint = await checkpointManager.load(loopId); let i = checkpoint?.currentIndex !== undefined ? checkpoint.currentIndex : initValue; let iterations = 0; while (conditionFn(i)) { if (iterations++ > MAX_ITERATIONS) { throw new InfiniteLoopDetectionError(loopId, iterations); } await bodyFn(i); const newCheckpoint = { loopId, currentIndex: incrementFn(i), timestamp: Date.now(), }; await checkpointManager.save(newCheckpoint); i = incrementFn(i); } await checkpointManager.clear(loopId); } //# sourceMappingURL=resumable-loops.js.map