deepline 0.1.182 → 0.1.184

package/dist/bundling-sources/apps/play-runner-workers/src/runtime/harness-receipt-store.ts

@@ -6,16 +6,87 @@ import {
   harnessFailRuntimeReceipt,
   harnessFailRuntimeReceipts,
   harnessGetRuntimeReceipt,
+  harnessHeartbeatRuntimeReceipts,
+  harnessReleaseRuntimeReceipt,
 } from '../../../../sdk/src/plays/harness-stub';
 import type { PreloadedRuntimeDbSessionInput } from '../../../play-harness-worker/src/rpc-types';
 import type { WorkerRuntimeReceiptStore } from './receipts';
 
+const RUNTIME_TEST_FAULT_NAMES = new Set([
+  'receipt_complete_write_fail',
+  'receipt_fail_write_fail',
+  'worker_receipt_complete_write_fail',
+]);
+
+function createReceiptCompletionFaultForwarder(
+  header: string | null | undefined,
+): () => string | null {
+  const value = header?.trim();
+  if (!value) return () => null;
+  const counts = new Map<string, number>();
+  for (const rawPart of value.split(',')) {
+    const part = rawPart.trim();
+    if (!part) continue;
+    const [rawName, rawCount] = part.split(':');
+    const name = rawName?.trim();
+    if (!name) continue;
+    if (!RUNTIME_TEST_FAULT_NAMES.has(name)) {
+      throw new Error(`Unsupported runtime test fault "${name}".`);
+    }
+    const count =
+      rawCount === undefined || rawCount.trim() === '' ? 1 : Number(rawCount);
+    if (!Number.isInteger(count) || count < 1) {
+      throw new Error(
+        `Runtime test fault "${name}" count must be a positive integer.`,
+      );
+    }
+    counts.set(name, (counts.get(name) ?? 0) + count);
+  }
+  return () => {
+    const forwarded: string[] = [];
+    for (const [name, count] of counts) {
+      if (name !== 'worker_receipt_complete_write_fail') {
+        forwarded.push(`${name}:${count}`);
+        continue;
+      }
+      if (count <= 0) continue;
+      forwarded.push('receipt_complete_write_fail:1');
+      // Keep repeated worker-scoped fault forwards distinct for the direct
+      // runtime fault consumer, whose state is keyed by runId + header value.
+      forwarded.push(`${name}:${count}`);
+      if (count === 1) {
+        counts.delete(name);
+      } else {
+        counts.set(name, count - 1);
+      }
+    }
+    return forwarded.length > 0 ? forwarded.join(',') : null;
+  };
+}
+
 export function createHarnessWorkerReceiptStore(input: {
   executorToken: string;
   orgId: string;
   preloadedDbSessions?: PreloadedRuntimeDbSessionInput[] | null;
   userEmail?: string | null;
+  runtimeTestFaultHeader?: string | null;
+  runAttempt?: number | null;
+  receiptLeaseTtlMs?: number | null;
 }): WorkerRuntimeReceiptStore {
+  const runAttempt =
+    typeof input.runAttempt === 'number' && Number.isFinite(input.runAttempt)
+      ? Math.max(0, Math.floor(input.runAttempt))
+      : null;
+  const runAttemptField = runAttempt == null ? {} : { runAttempt };
+  const leaseTtlField =
+    typeof input.receiptLeaseTtlMs === 'number' &&
+    Number.isFinite(input.receiptLeaseTtlMs) &&
+    input.receiptLeaseTtlMs > 0
+      ? { leaseTtlMs: Math.floor(input.receiptLeaseTtlMs) }
+      : {};
+  const receiptCompletionFaultHeader = createReceiptCompletionFaultForwarder(
+    input.runtimeTestFaultHeader,
+  );
   return {
     getReceipt(command) {
       return harnessGetRuntimeReceipt({
@@ -23,7 +94,9 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
         runId: '',
+        ...runAttemptField,
         ...command,
       });
     },
@@ -33,6 +106,9 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...runAttemptField,
+        ...leaseTtlField,
         ...command,
       });
     },
@@ -42,6 +118,9 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...runAttemptField,
+        ...leaseTtlField,
         ...command,
       });
     },
@@ -51,6 +130,8 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: receiptCompletionFaultHeader(),
+        ...runAttemptField,
         ...command,
       });
     },
@@ -60,7 +141,12 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: receiptCompletionFaultHeader(),
         ...command,
+        receipts: command.receipts.map((receipt) => ({
+          ...runAttemptField,
+          ...receipt,
+        })),
       });
     },
     failReceipt(command) {
@@ -69,6 +155,19 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...runAttemptField,
+        ...command,
+      });
+    },
+    releaseReceipt(command) {
+      return harnessReleaseRuntimeReceipt({
+        executorToken: input.executorToken,
+        orgId: input.orgId,
+        preloadedDbSessions: input.preloadedDbSessions ?? null,
+        userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...runAttemptField,
         ...command,
       });
     },
@@ -78,6 +177,23 @@ export function createHarnessWorkerReceiptStore(input: {
         orgId: input.orgId,
         preloadedDbSessions: input.preloadedDbSessions ?? null,
         userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...command,
+        receipts: command.receipts.map((receipt) => ({
+          ...runAttemptField,
+          ...receipt,
+        })),
+      });
+    },
+    heartbeatReceipts(command) {
+      return harnessHeartbeatRuntimeReceipts({
+        executorToken: input.executorToken,
+        orgId: input.orgId,
+        preloadedDbSessions: input.preloadedDbSessions ?? null,
+        userEmail: input.userEmail ?? null,
+        runtimeTestFaultHeader: input.runtimeTestFaultHeader ?? null,
+        ...runAttemptField,
+        ...leaseTtlField,
         ...command,
       });
     },