@codemation/core 0.13.0 → 0.13.2

package/dist/{runtime-DjYXgOo0.cjs → runtime-_VdHwGkJ.cjs}

@@ -1,4 +1,4 @@
-const require_di = require('./di-C-2ep8NZ.cjs');
+const require_di = require('./di-LP2qSHkY.cjs');
 let zod = require("zod");
 zod = require_di.__toESM(zod);
 let node_stream_web = require("node:stream/web");
@@ -5003,12 +5003,26 @@ var InlineDrivingScheduler = class {
 	drainingRuns = /* @__PURE__ */ new Set();
 	queuesByRunId = /* @__PURE__ */ new Map();
 	scheduledRuns = /* @__PURE__ */ new Set();
+	stopped = false;
+	activeDrainPromises = /* @__PURE__ */ new Set();
 	constructor(nodeExecutor) {
 		this.nodeExecutor = nodeExecutor;
 	}
 	setContinuation(continuation) {
 		this.continuation = continuation;
 	}
+	/**
+	* Prevents new drain cycles from being scheduled and waits for all currently-running
+	* drains to complete. Call before closing the DB connection or rolling back test transactions
+	* to ensure no in-flight writes hit a closed/rolled-back connection.
+	*
+	* Bounded: only the drains already in-flight at call time are awaited. Runs that keep
+	* re-scheduling will not schedule new work once stopped is set.
+	*/
+	async stop() {
+		this.stopped = true;
+		if (this.activeDrainPromises.size > 0) await Promise.allSettled(this.activeDrainPromises);
+	}
 	async prepareDispatch(request) {
 		const receipt = {
 			receiptId: request.activationId,
@@ -5058,11 +5072,15 @@ var InlineDrivingScheduler = class {
 		}
 	}
 	scheduleDrain(runId) {
-		if (this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) return;
+		if (this.stopped || this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) return;
 		this.scheduledRuns.add(runId);
 		setImmediate(() => {
 			this.scheduledRuns.delete(runId);
-			this.drainRun(runId);
+			if (this.stopped) return;
+			const p = this.drainRun(runId).finally(() => {
+				this.activeDrainPromises.delete(p);
+			});
+			this.activeDrainPromises.add(p);
 		});
 	}
 	async resumeAfterExecutionResult(continuation, request, outputs) {
@@ -7133,4 +7151,4 @@ Object.defineProperty(exports, 'tool', {
     return tool;
   }
 });
-//# sourceMappingURL=runtime-DjYXgOo0.cjs.map
+//# sourceMappingURL=runtime-_VdHwGkJ.cjs.map