@codemation/core 0.13.0 → 0.13.1

package/dist/testing.cjs

@@ -1,8 +1,8 @@
 const require_di = require('./di-C-2ep8NZ.cjs');
 require('./contracts-CK0x6w_G.cjs');
-const require_runtime = require('./runtime-DjYXgOo0.cjs');
+const require_runtime = require('./runtime-rrH8-Ouq.cjs');
 const require_InMemoryRunEventBusRegistry = require('./InMemoryRunEventBusRegistry-Sa86VxuV.cjs');
-const require_bootstrap = require('./bootstrap-BEu1fJBM.cjs');
+const require_bootstrap = require('./bootstrap-UDyH8OfK.cjs');
 let tsyringe = require("tsyringe");
 tsyringe = require_di.__toESM(tsyringe);
 

package/dist/testing.js

@@ -1,8 +1,8 @@
 import { d as CoreTokens } from "./di-D9Mv3kF3.js";
 import "./contracts-DXdfTdpW.js";
-import { A as PersistedWorkflowTokenRegistry, C as DefaultDrivingScheduler, G as AllWorkflowsActiveWorkflowActivationPolicy, N as NodeExecutor, O as NodeInstanceFactory, S as HintOnlyOffloadPolicy, V as DefaultExecutionContextFactory, a as InMemoryLiveWorkflowRepository, i as RunIntentService, it as emitPorts, k as WorkflowSnapshotCodec, l as Engine, p as InMemoryRunDataFactory, st as DefaultAsyncSleeper, x as InlineDrivingScheduler, yt as WorkflowBuilder, z as InProcessRetryRunner } from "./runtime-6-U2Cou5.js";
+import { A as PersistedWorkflowTokenRegistry, C as DefaultDrivingScheduler, G as AllWorkflowsActiveWorkflowActivationPolicy, N as NodeExecutor, O as NodeInstanceFactory, S as HintOnlyOffloadPolicy, V as DefaultExecutionContextFactory, a as InMemoryLiveWorkflowRepository, i as RunIntentService, it as emitPorts, k as WorkflowSnapshotCodec, l as Engine, p as InMemoryRunDataFactory, st as DefaultAsyncSleeper, x as InlineDrivingScheduler, yt as WorkflowBuilder, z as InProcessRetryRunner } from "./runtime-iHBN1jyD.js";
 import { t as InMemoryRunEventBus } from "./InMemoryRunEventBusRegistry-Bwunvt1T.js";
-import { a as InMemoryWorkflowExecutionRepository, t as EngineRuntimeRegistrar } from "./bootstrap-CSeInbj1.js";
+import { a as InMemoryWorkflowExecutionRepository, t as EngineRuntimeRegistrar } from "./bootstrap-DB3jpo8F.js";
 import { container } from "tsyringe";
 
 //#region src/testing/RejectingCredentialSessionService.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codemation/core",
-  "version": "0.13.0",
+  "version": "0.13.1",
   "publishConfig": {
     "access": "public"
   },

package/src/scheduler/InlineDrivingScheduler.ts

@@ -15,6 +15,8 @@ export class InlineDrivingScheduler implements NodeActivationScheduler {
     Array<Readonly<{ request: NodeActivationRequest; receipt: NodeActivationReceipt }>>
   >();
   private readonly scheduledRuns = new Set<string>();
+  private stopped = false;
+  private readonly activeDrainPromises = new Set<Promise<void>>();
 
   constructor(private readonly nodeExecutor: NodeExecutor) {}
 
@@ -22,6 +24,21 @@ export class InlineDrivingScheduler implements NodeActivationScheduler {
     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(): Promise<void> {
+    this.stopped = true;
+    if (this.activeDrainPromises.size > 0) {
+      await Promise.allSettled(this.activeDrainPromises);
+    }
+  }
+
   async prepareDispatch(request: NodeActivationRequest): Promise<PreparedNodeActivationDispatch> {
     const receipt: NodeActivationReceipt = { receiptId: request.activationId, mode: "local" };
     return {
@@ -77,7 +94,7 @@ export class InlineDrivingScheduler implements NodeActivationScheduler {
   }
 
   private scheduleDrain(runId: string): void {
-    if (this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) {
+    if (this.stopped || this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) {
       return;
     }
     this.scheduledRuns.add(runId);
@@ -85,7 +102,11 @@ export class InlineDrivingScheduler implements NodeActivationScheduler {
     // giving queued HTTP writes and WS frames a chance to flush between activations.
     setImmediate(() => {
       this.scheduledRuns.delete(runId);
-      void this.drainRun(runId);
+      if (this.stopped) return;
+      const p = this.drainRun(runId).finally(() => {
+        this.activeDrainPromises.delete(p);
+      });
+      this.activeDrainPromises.add(p);
     });
   }