@shipfox/api-workflows 2.0.0

package/dist/temporal/workflows/run-orchestration.js

@@ -0,0 +1,243 @@
+import { condition, defineSignal, executeChild, log, ParentClosePolicy, proxyActivities, setHandler } from '@temporalio/workflow';
+import { createRuntimeRunProgress, nonCompletedRuntimeJobIds, recordRuntimeJobResult, recordSkippedRuntimeJob, runtimeJobVersion, shouldContinueStartedRun } from '#core/workflow-scheduling/run-progress.js';
+import { scheduleRuntimeDag } from '#core/workflow-scheduling/schedule-runtime-dag.js';
+import { RUN_CANCEL_SIGNAL } from '../constants.js';
+import { deadlineReached, remainingMs } from './deadline.js';
+import { jobExecutionOrchestration } from './job-execution-orchestration.js';
+import { jobListenerOrchestration } from './job-listener-orchestration.js';
+const { loadRunAttemptDag, evaluateJobActivationsActivity, setRunAttemptStatus, setJobStatus, cancelRunnerJobsActivity } = proxyActivities({
+    startToCloseTimeout: '30s'
+});
+const { failRunAsTimedOutActivity } = proxyActivities({
+    startToCloseTimeout: '30s',
+    retry: {
+        maximumAttempts: 5
+    }
+});
+export const runCancelSignal = defineSignal(RUN_CANCEL_SIGNAL);
+export async function runOrchestration(input) {
+    let cancelRequested = false;
+    setHandler(runCancelSignal, ()=>{
+        cancelRequested = true;
+    });
+    const dag = await loadRunAttemptDag(input.runAttemptId);
+    const runDeadline = Date.now() + dag.runTimeoutMs;
+    let runVersion = dag.runVersion;
+    const { newVersion, status } = await setRunAttemptStatus({
+        runAttemptId: input.runAttemptId,
+        status: 'running',
+        version: runVersion
+    });
+    runVersion = newVersion;
+    if (!shouldContinueStartedRun(status)) return;
+    const progress = createRuntimeRunProgress(dag.jobs);
+    const inFlight = new Map();
+    while(true){
+        if (cancelRequested) {
+            await cancelNonCompletedRunnerJobs(dag, progress);
+            return;
+        }
+        if (deadlineReached(runDeadline)) {
+            await failRunAsTimedOutActivity({
+                runAttemptId: input.runAttemptId
+            });
+            await cancelNonCompletedRunnerJobs(dag, progress);
+            return;
+        }
+        const commands = scheduleRuntimeDag({
+            jobs: dag.jobs,
+            completed: progress.completed,
+            running: new Set(inFlight.keys())
+        });
+        const completeRun = commands.find((command)=>command.kind === 'complete-run');
+        const startJobs = new Map();
+        for (const command of commands){
+            if (command.kind !== 'skip-job') continue;
+            await skipJob(command.job, progress, command.statusReason);
+        }
+        for (const command of commands){
+            if (command.kind !== 'evaluate-job-activation') continue;
+            const activationJobsById = new Map(command.jobs.map((job)=>[
+                    job.id,
+                    job
+                ]));
+            const decisions = await evaluateJobActivationsActivity({
+                runAttemptId: input.runAttemptId,
+                jobs: command.jobs.map((job)=>({
+                        jobId: job.id,
+                        expectedVersion: runtimeJobVersion(job, progress)
+                    }))
+            });
+            for (const decision of decisions){
+                const job = activationJobsById.get(decision.jobId);
+                if (!job) continue;
+                if (decision.kind === 'start-job') {
+                    startJobs.set(job.key, job);
+                    continue;
+                }
+                recordRuntimeJobResult(job, progress, {
+                    status: decision.status,
+                    jobVersion: decision.jobVersion
+                });
+            }
+        }
+        for (const command of commands){
+            if (command.kind !== 'start-job') continue;
+            startJobs.set(command.job.key, command.job);
+        }
+        for (const job of startJobs.values()){
+            if (!inFlight.has(job.key)) {
+                inFlight.set(job.key, launchJob(job, dag, progress));
+            }
+        }
+        if (completeRun) {
+            await setRunAttemptStatus({
+                runAttemptId: input.runAttemptId,
+                status: completeRun.status,
+                version: runVersion
+            });
+            return;
+        }
+        if (startJobs.size === 0 && inFlight.size === 0) continue;
+        const settled = await waitForNextSettlement(inFlight, ()=>cancelRequested, runDeadline);
+        if (settled.kind === 'cancelled') {
+            await cancelNonCompletedRunnerJobs(dag, progress);
+            return;
+        }
+        if (settled.kind === 'timed-out') {
+            await failRunAsTimedOutActivity({
+                runAttemptId: input.runAttemptId
+            });
+            await cancelNonCompletedRunnerJobs(dag, progress);
+            return;
+        }
+        inFlight.delete(settled.job.key);
+        recordRuntimeJobResult(settled.job, progress, settled.result);
+    }
+}
+async function skipJob(job, progress, statusReason) {
+    const version = runtimeJobVersion(job, progress);
+    const { newVersion } = await setJobStatus({
+        jobId: job.id,
+        status: 'skipped',
+        version,
+        statusReason
+    });
+    recordSkippedRuntimeJob(job, progress, newVersion);
+}
+function launchJob(job, run, progress) {
+    if (job.mode === 'listening') {
+        return executeChild(jobListenerOrchestration, {
+            workflowId: `job-listener:${job.id}`,
+            args: [
+                {
+                    workspaceId: run.workspaceId,
+                    workflowRunId: run.workflowRunId,
+                    projectId: run.projectId,
+                    jobId: job.id,
+                    runAttemptId: run.runAttemptId,
+                    jobVersion: runtimeJobVersion(job, progress),
+                    ...job.executionTimeoutMs === undefined ? {} : {
+                        executionTimeoutMs: job.executionTimeoutMs
+                    },
+                    ...job.listeningTimeoutMs === undefined ? {} : {
+                        listeningTimeoutMs: job.listeningTimeoutMs
+                    },
+                    ...job.maxExecutions === undefined ? {} : {
+                        maxExecutions: job.maxExecutions
+                    },
+                    ...job.onResolve === undefined ? {} : {
+                        onResolve: job.onResolve
+                    },
+                    ...job.batchDebounceMs === undefined ? {} : {
+                        batchDebounceMs: job.batchDebounceMs
+                    },
+                    ...job.batchMaxSize === undefined ? {} : {
+                        batchMaxSize: job.batchMaxSize
+                    },
+                    ...job.batchMaxWaitMs === undefined ? {} : {
+                        batchMaxWaitMs: job.batchMaxWaitMs
+                    },
+                    requiredLabels: job.runner
+                }
+            ],
+            parentClosePolicy: ParentClosePolicy.TERMINATE
+        }).then((result)=>({
+                job,
+                result
+            })).catch(async (error)=>{
+            log.warn('listener child failed; marking runtime job failed', {
+                jobId: job.id,
+                error: String(error)
+            });
+            const failed = await setJobStatus({
+                jobId: job.id,
+                status: 'failed',
+                version: runtimeJobVersion(job, progress),
+                statusReason: 'unknown'
+            });
+            return {
+                job,
+                result: {
+                    status: 'failed',
+                    jobVersion: failed.newVersion
+                }
+            };
+        });
+    }
+    if (job.jobExecutionId === undefined) {
+        throw new Error(`Cannot start job without an execution: ${job.id}`);
+    }
+    return executeChild(jobExecutionOrchestration, {
+        workflowId: `job:${job.id}`,
+        args: [
+            {
+                workspaceId: run.workspaceId,
+                workflowRunId: run.workflowRunId,
+                projectId: run.projectId,
+                jobId: job.id,
+                jobExecutionId: job.jobExecutionId,
+                runAttemptId: run.runAttemptId,
+                jobVersion: runtimeJobVersion(job, progress),
+                executionVersion: job.executionVersion ?? runtimeJobVersion(job, progress),
+                ...job.executionTimeoutMs === undefined ? {} : {
+                    executionTimeoutMs: job.executionTimeoutMs
+                },
+                requiredLabels: job.runner
+            }
+        ],
+        parentClosePolicy: ParentClosePolicy.TERMINATE
+    }).then((result)=>({
+            job,
+            result: {
+                status: result.status,
+                jobVersion: result.jobVersion
+            }
+        }));
+}
+async function waitForNextSettlement(inFlight, isCancelRequested, runDeadline) {
+    const remaining = remainingMs(runDeadline) ?? 0;
+    const childSettled = Promise.race([
+        ...inFlight.values()
+    ]).then((settled)=>({
+            kind: 'settled',
+            ...settled
+        }));
+    const cancel = condition(isCancelRequested, remaining).then((woke)=>woke ? {
+            kind: 'cancelled'
+        } : {
+            kind: 'timed-out'
+        });
+    return await Promise.race([
+        childSettled,
+        cancel
+    ]);
+}
+async function cancelNonCompletedRunnerJobs(dag, progress) {
+    const jobIds = nonCompletedRuntimeJobIds(dag.jobs, progress);
+    await cancelRunnerJobsActivity({
+        jobIds
+    });
+}
+
+//# sourceMappingURL=run-orchestration.js.map

package/dist/temporal/workflows/run-orchestration.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/temporal/workflows/run-orchestration.ts"],"sourcesContent":["import {\n  condition,\n  defineSignal,\n  executeChild,\n  log,\n  ParentClosePolicy,\n  proxyActivities,\n  setHandler,\n} from '@temporalio/workflow';\nimport {\n  createRuntimeRunProgress,\n  nonCompletedRuntimeJobIds,\n  type RuntimeRunProgress,\n  recordRuntimeJobResult,\n  recordSkippedRuntimeJob,\n  runtimeJobVersion,\n  shouldContinueStartedRun,\n} from '#core/workflow-scheduling/run-progress.js';\nimport type {RuntimeCompletionStatus} from '#core/workflow-scheduling/runtime-dag.js';\nimport {scheduleRuntimeDag} from '#core/workflow-scheduling/schedule-runtime-dag.js';\n\nimport type {createOrchestrationActivities} from '../activities/index.js';\nimport type {DagJob, RunDag} from '../activities/orchestration-activities.js';\nimport {RUN_CANCEL_SIGNAL} from '../constants.js';\nimport {deadlineReached, remainingMs} from './deadline.js';\nimport {jobExecutionOrchestration} from './job-execution-orchestration.js';\nimport {jobListenerOrchestration} from './job-listener-orchestration.js';\n\nconst {\n  loadRunAttemptDag,\n  evaluateJobActivationsActivity,\n  setRunAttemptStatus,\n  setJobStatus,\n  cancelRunnerJobsActivity,\n} = proxyActivities<ReturnType<typeof createOrchestrationActivities>>({\n  startToCloseTimeout: '30s',\n});\n\nconst {failRunAsTimedOutActivity} = proxyActivities<\n  ReturnType<typeof createOrchestrationActivities>\n>({\n  startToCloseTimeout: '30s',\n  retry: {maximumAttempts: 5},\n});\n\nexport const runCancelSignal = defineSignal<[]>(RUN_CANCEL_SIGNAL);\n\nexport interface RunOrchestrationInput {\n  workflowRunId: string;\n  runAttemptId: string;\n  workspaceId: string;\n}\n\nexport async function runOrchestration(input: RunOrchestrationInput): Promise<void> {\n  let cancelRequested = false;\n  setHandler(runCancelSignal, () => {\n    cancelRequested = true;\n  });\n\n  const dag = await loadRunAttemptDag(input.runAttemptId);\n  const runDeadline = Date.now() + dag.runTimeoutMs;\n\n  let runVersion = dag.runVersion;\n  const {newVersion, status} = await setRunAttemptStatus({\n    runAttemptId: input.runAttemptId,\n    status: 'running',\n    version: runVersion,\n  });\n  runVersion = newVersion;\n  if (!shouldContinueStartedRun(status)) return;\n\n  const progress = createRuntimeRunProgress(dag.jobs);\n  const inFlight = new Map<string, Promise<{job: DagJob; result: LaunchResult}>>();\n\n  while (true) {\n    if (cancelRequested) {\n      await cancelNonCompletedRunnerJobs(dag, progress);\n      return;\n    }\n    if (deadlineReached(runDeadline)) {\n      await failRunAsTimedOutActivity({runAttemptId: input.runAttemptId});\n      await cancelNonCompletedRunnerJobs(dag, progress);\n      return;\n    }\n\n    const commands = scheduleRuntimeDag({\n      jobs: dag.jobs,\n      completed: progress.completed,\n      running: new Set(inFlight.keys()),\n    });\n    const completeRun = commands.find((command) => command.kind === 'complete-run');\n    const startJobs = new Map<string, DagJob>();\n\n    for (const command of commands) {\n      if (command.kind !== 'skip-job') continue;\n      await skipJob(command.job, progress, command.statusReason);\n    }\n\n    for (const command of commands) {\n      if (command.kind !== 'evaluate-job-activation') continue;\n      const activationJobsById = new Map(command.jobs.map((job) => [job.id, job]));\n      const decisions = await evaluateJobActivationsActivity({\n        runAttemptId: input.runAttemptId,\n        jobs: command.jobs.map((job) => ({\n          jobId: job.id,\n          expectedVersion: runtimeJobVersion(job, progress),\n        })),\n      });\n      for (const decision of decisions) {\n        const job = activationJobsById.get(decision.jobId);\n        if (!job) continue;\n        if (decision.kind === 'start-job') {\n          startJobs.set(job.key, job);\n          continue;\n        }\n        recordRuntimeJobResult(job, progress, {\n          status: decision.status,\n          jobVersion: decision.jobVersion,\n        });\n      }\n    }\n\n    for (const command of commands) {\n      if (command.kind !== 'start-job') continue;\n      startJobs.set(command.job.key, command.job);\n    }\n\n    for (const job of startJobs.values()) {\n      if (!inFlight.has(job.key)) {\n        inFlight.set(job.key, launchJob(job, dag, progress));\n      }\n    }\n\n    if (completeRun) {\n      await setRunAttemptStatus({\n        runAttemptId: input.runAttemptId,\n        status: completeRun.status,\n        version: runVersion,\n      });\n      return;\n    }\n\n    if (startJobs.size === 0 && inFlight.size === 0) continue;\n\n    const settled = await waitForNextSettlement(inFlight, () => cancelRequested, runDeadline);\n    if (settled.kind === 'cancelled') {\n      await cancelNonCompletedRunnerJobs(dag, progress);\n      return;\n    }\n    if (settled.kind === 'timed-out') {\n      await failRunAsTimedOutActivity({runAttemptId: input.runAttemptId});\n      await cancelNonCompletedRunnerJobs(dag, progress);\n      return;\n    }\n    inFlight.delete(settled.job.key);\n    recordRuntimeJobResult(settled.job, progress, settled.result);\n  }\n}\n\nasync function skipJob(\n  job: DagJob,\n  progress: RuntimeRunProgress,\n  statusReason: 'default_gate_rejected',\n): Promise<void> {\n  const version = runtimeJobVersion(job, progress);\n  const {newVersion} = await setJobStatus({\n    jobId: job.id,\n    status: 'skipped',\n    version,\n    statusReason,\n  });\n  recordSkippedRuntimeJob(job, progress, newVersion);\n}\n\ninterface LaunchResult {\n  status: RuntimeCompletionStatus;\n  jobVersion: number;\n}\n\nfunction launchJob(\n  job: DagJob,\n  run: RunDag,\n  progress: RuntimeRunProgress,\n): Promise<{job: DagJob; result: LaunchResult}> {\n  if (job.mode === 'listening') {\n    return executeChild(jobListenerOrchestration, {\n      workflowId: `job-listener:${job.id}`,\n      args: [\n        {\n          workspaceId: run.workspaceId,\n          workflowRunId: run.workflowRunId,\n          projectId: run.projectId,\n          jobId: job.id,\n          runAttemptId: run.runAttemptId,\n          jobVersion: runtimeJobVersion(job, progress),\n          ...(job.executionTimeoutMs === undefined\n            ? {}\n            : {executionTimeoutMs: job.executionTimeoutMs}),\n          ...(job.listeningTimeoutMs === undefined\n            ? {}\n            : {listeningTimeoutMs: job.listeningTimeoutMs}),\n          ...(job.maxExecutions === undefined ? {} : {maxExecutions: job.maxExecutions}),\n          ...(job.onResolve === undefined ? {} : {onResolve: job.onResolve}),\n          ...(job.batchDebounceMs === undefined ? {} : {batchDebounceMs: job.batchDebounceMs}),\n          ...(job.batchMaxSize === undefined ? {} : {batchMaxSize: job.batchMaxSize}),\n          ...(job.batchMaxWaitMs === undefined ? {} : {batchMaxWaitMs: job.batchMaxWaitMs}),\n          requiredLabels: job.runner,\n        },\n      ],\n      parentClosePolicy: ParentClosePolicy.TERMINATE,\n    })\n      .then((result) => ({job, result}))\n      .catch(async (error) => {\n        log.warn('listener child failed; marking runtime job failed', {\n          jobId: job.id,\n          error: String(error),\n        });\n        const failed = await setJobStatus({\n          jobId: job.id,\n          status: 'failed',\n          version: runtimeJobVersion(job, progress),\n          statusReason: 'unknown',\n        });\n        return {\n          job,\n          result: {status: 'failed' as const, jobVersion: failed.newVersion},\n        };\n      });\n  }\n\n  if (job.jobExecutionId === undefined) {\n    throw new Error(`Cannot start job without an execution: ${job.id}`);\n  }\n  return executeChild(jobExecutionOrchestration, {\n    workflowId: `job:${job.id}`,\n    args: [\n      {\n        workspaceId: run.workspaceId,\n        workflowRunId: run.workflowRunId,\n        projectId: run.projectId,\n        jobId: job.id,\n        jobExecutionId: job.jobExecutionId,\n        runAttemptId: run.runAttemptId,\n        jobVersion: runtimeJobVersion(job, progress),\n        executionVersion: job.executionVersion ?? runtimeJobVersion(job, progress),\n        ...(job.executionTimeoutMs === undefined\n          ? {}\n          : {executionTimeoutMs: job.executionTimeoutMs}),\n        requiredLabels: job.runner,\n      },\n    ],\n    parentClosePolicy: ParentClosePolicy.TERMINATE,\n  }).then((result) => ({\n    job,\n    result: {status: result.status, jobVersion: result.jobVersion},\n  }));\n}\n\nasync function waitForNextSettlement(\n  inFlight: ReadonlyMap<string, Promise<{job: DagJob; result: LaunchResult}>>,\n  isCancelRequested: () => boolean,\n  runDeadline: number,\n): Promise<\n  {kind: 'settled'; job: DagJob; result: LaunchResult} | {kind: 'cancelled'} | {kind: 'timed-out'}\n> {\n  const remaining = remainingMs(runDeadline) ?? 0;\n  const childSettled = Promise.race([...inFlight.values()]).then((settled) => ({\n    kind: 'settled' as const,\n    ...settled,\n  }));\n  const cancel = condition(isCancelRequested, remaining).then((woke) =>\n    woke ? {kind: 'cancelled' as const} : {kind: 'timed-out' as const},\n  );\n  return await Promise.race([childSettled, cancel]);\n}\n\nasync function cancelNonCompletedRunnerJobs(\n  dag: RunDag,\n  progress: RuntimeRunProgress,\n): Promise<void> {\n  const jobIds = nonCompletedRuntimeJobIds(dag.jobs, progress);\n  await cancelRunnerJobsActivity({jobIds});\n}\n"],"names":["condition","defineSignal","executeChild","log","ParentClosePolicy","proxyActivities","setHandler","createRuntimeRunProgress","nonCompletedRuntimeJobIds","recordRuntimeJobResult","recordSkippedRuntimeJob","runtimeJobVersion","shouldContinueStartedRun","scheduleRuntimeDag","RUN_CANCEL_SIGNAL","deadlineReached","remainingMs","jobExecutionOrchestration","jobListenerOrchestration","loadRunAttemptDag","evaluateJobActivationsActivity","setRunAttemptStatus","setJobStatus","cancelRunnerJobsActivity","startToCloseTimeout","failRunAsTimedOutActivity","retry","maximumAttempts","runCancelSignal","runOrchestration","input","cancelRequested","dag","runAttemptId","runDeadline","Date","now","runTimeoutMs","runVersion","newVersion","status","version","progress","jobs","inFlight","Map","cancelNonCompletedRunnerJobs","commands","completed","running","Set","keys","completeRun","find","command","kind","startJobs","skipJob","job","statusReason","activationJobsById","map","id","decisions","jobId","expectedVersion","decision","get","set","key","jobVersion","values","has","launchJob","size","settled","waitForNextSettlement","delete","result","run","mode","workflowId","args","workspaceId","workflowRunId","projectId","executionTimeoutMs","undefined","listeningTimeoutMs","maxExecutions","onResolve","batchDebounceMs","batchMaxSize","batchMaxWaitMs","requiredLabels","runner","parentClosePolicy","TERMINATE","then","catch","error","warn","String","failed","jobExecutionId","Error","executionVersion","isCancelRequested","remaining","childSettled","Promise","race","cancel","woke","jobIds"],"mappings":"AAAA,SACEA,SAAS,EACTC,YAAY,EACZC,YAAY,EACZC,GAAG,EACHC,iBAAiB,EACjBC,eAAe,EACfC,UAAU,QACL,uBAAuB;AAC9B,SACEC,wBAAwB,EACxBC,yBAAyB,EAEzBC,sBAAsB,EACtBC,uBAAuB,EACvBC,iBAAiB,EACjBC,wBAAwB,QACnB,4CAA4C;AAEnD,SAAQC,kBAAkB,QAAO,oDAAoD;AAIrF,SAAQC,iBAAiB,QAAO,kBAAkB;AAClD,SAAQC,eAAe,EAAEC,WAAW,QAAO,gBAAgB;AAC3D,SAAQC,yBAAyB,QAAO,mCAAmC;AAC3E,SAAQC,wBAAwB,QAAO,kCAAkC;AAEzE,MAAM,EACJC,iBAAiB,EACjBC,8BAA8B,EAC9BC,mBAAmB,EACnBC,YAAY,EACZC,wBAAwB,EACzB,GAAGlB,gBAAkE;IACpEmB,qBAAqB;AACvB;AAEA,MAAM,EAACC,yBAAyB,EAAC,GAAGpB,gBAElC;IACAmB,qBAAqB;IACrBE,OAAO;QAACC,iBAAiB;IAAC;AAC5B;AAEA,OAAO,MAAMC,kBAAkB3B,aAAiBa,mBAAmB;AAQnE,OAAO,eAAee,iBAAiBC,KAA4B;IACjE,IAAIC,kBAAkB;IACtBzB,WAAWsB,iBAAiB;QAC1BG,kBAAkB;IACpB;IAEA,MAAMC,MAAM,MAAMb,kBAAkBW,MAAMG,YAAY;IACtD,MAAMC,cAAcC,KAAKC,GAAG,KAAKJ,IAAIK,YAAY;IAEjD,IAAIC,aAAaN,IAAIM,UAAU;IAC/B,MAAM,EAACC,UAAU,EAAEC,MAAM,EAAC,GAAG,MAAMnB,oBAAoB;QACrDY,cAAcH,MAAMG,YAAY;QAChCO,QAAQ;QACRC,SAASH;IACX;IACAA,aAAaC;IACb,IAAI,CAAC3B,yBAAyB4B,SAAS;IAEvC,MAAME,WAAWnC,yBAAyByB,IAAIW,IAAI;IAClD,MAAMC,WAAW,IAAIC;IAErB,MAAO,KAAM;QACX,IAAId,iBAAiB;YACnB,MAAMe,6BAA6Bd,KAAKU;YACxC;QACF;QACA,IAAI3B,gBAAgBmB,cAAc;YAChC,MAAMT,0BAA0B;gBAACQ,cAAcH,MAAMG,YAAY;YAAA;YACjE,MAAMa,6BAA6Bd,KAAKU;YACxC;QACF;QAEA,MAAMK,WAAWlC,mBAAmB;YAClC8B,MAAMX,IAAIW,IAAI;YACdK,WAAWN,SAASM,SAAS;YAC7BC,SAAS,IAAIC,IAAIN,SAASO,IAAI;QAChC;QACA,MAAMC,cAAcL,SAASM,IAAI,CAAC,CAACC,UAAYA,QAAQC,IAAI,KAAK;QAChE,MAAMC,YAAY,IAAIX;QAEtB,KAAK,MAAMS,WAAWP,SAAU;YAC9B,IAAIO,QAAQC,IAAI,KAAK,YAAY;YACjC,MAAME,QAAQH,QAAQI,GAAG,EAAEhB,UAAUY,QAAQK,YAAY;QAC3D;QAEA,KAAK,MAAML,WAAWP,SAAU;YAC9B,IAAIO,QAAQC,IAAI,KAAK,2BAA2B;YAChD,MAAMK,qBAAqB,IAAIf,IAAIS,QAAQX,IAAI,CAACkB,GAAG,CAAC,CAACH,MAAQ;oBAACA,IAAII,EAAE;oBAAEJ;iBAAI;YAC1E,MAAMK,YAAY,MAAM3C,+BAA+B;gBACrDa,cAAcH,MAAMG,YAAY;gBAChCU,MAAMW,QAAQX,IAAI,CAACkB,GAAG,CAAC,CAACH,MAAS,CAAA;wBAC/BM,OAAON,IAAII,EAAE;wBACbG,iBAAiBtD,kBAAkB+C,KAAKhB;oBAC1C,CAAA;YACF;YACA,KAAK,MAAMwB,YAAYH,UAAW;gBAChC,MAAML,MAAME,mBAAmBO,GAAG,CAACD,SAASF,KAAK;gBACjD,IAAI,CAACN,KAAK;gBACV,IAAIQ,SAASX,IAAI,KAAK,aAAa;oBACjCC,UAAUY,GAAG,CAACV,IAAIW,GAAG,EAAEX;oBACvB;gBACF;gBACAjD,uBAAuBiD,KAAKhB,UAAU;oBACpCF,QAAQ0B,SAAS1B,MAAM;oBACvB8B,YAAYJ,SAASI,UAAU;gBACjC;YACF;QACF;QAEA,KAAK,MAAMhB,WAAWP,SAAU;YAC9B,IAAIO,QAAQC,IAAI,KAAK,aAAa;YAClCC,UAAUY,GAAG,CAACd,QAAQI,GAAG,CAACW,GAAG,EAAEf,QAAQI,GAAG;QAC5C;QAEA,KAAK,MAAMA,OAAOF,UAAUe,MAAM,GAAI;YACpC,IAAI,CAAC3B,SAAS4B,GAAG,CAACd,IAAIW,GAAG,GAAG;gBAC1BzB,SAASwB,GAAG,CAACV,IAAIW,GAAG,EAAEI,UAAUf,KAAK1B,KAAKU;YAC5C;QACF;QAEA,IAAIU,aAAa;YACf,MAAM/B,oBAAoB;gBACxBY,cAAcH,MAAMG,YAAY;gBAChCO,QAAQY,YAAYZ,MAAM;gBAC1BC,SAASH;YACX;YACA;QACF;QAEA,IAAIkB,UAAUkB,IAAI,KAAK,KAAK9B,SAAS8B,IAAI,KAAK,GAAG;QAEjD,MAAMC,UAAU,MAAMC,sBAAsBhC,UAAU,IAAMb,iBAAiBG;QAC7E,IAAIyC,QAAQpB,IAAI,KAAK,aAAa;YAChC,MAAMT,6BAA6Bd,KAAKU;YACxC;QACF;QACA,IAAIiC,QAAQpB,IAAI,KAAK,aAAa;YAChC,MAAM9B,0BAA0B;gBAACQ,cAAcH,MAAMG,YAAY;YAAA;YACjE,MAAMa,6BAA6Bd,KAAKU;YACxC;QACF;QACAE,SAASiC,MAAM,CAACF,QAAQjB,GAAG,CAACW,GAAG;QAC/B5D,uBAAuBkE,QAAQjB,GAAG,EAAEhB,UAAUiC,QAAQG,MAAM;IAC9D;AACF;AAEA,eAAerB,QACbC,GAAW,EACXhB,QAA4B,EAC5BiB,YAAqC;IAErC,MAAMlB,UAAU9B,kBAAkB+C,KAAKhB;IACvC,MAAM,EAACH,UAAU,EAAC,GAAG,MAAMjB,aAAa;QACtC0C,OAAON,IAAII,EAAE;QACbtB,QAAQ;QACRC;QACAkB;IACF;IACAjD,wBAAwBgD,KAAKhB,UAAUH;AACzC;AAOA,SAASkC,UACPf,GAAW,EACXqB,GAAW,EACXrC,QAA4B;IAE5B,IAAIgB,IAAIsB,IAAI,KAAK,aAAa;QAC5B,OAAO9E,aAAagB,0BAA0B;YAC5C+D,YAAY,CAAC,aAAa,EAAEvB,IAAII,EAAE,EAAE;YACpCoB,MAAM;gBACJ;oBACEC,aAAaJ,IAAII,WAAW;oBAC5BC,eAAeL,IAAIK,aAAa;oBAChCC,WAAWN,IAAIM,SAAS;oBACxBrB,OAAON,IAAII,EAAE;oBACb7B,cAAc8C,IAAI9C,YAAY;oBAC9BqC,YAAY3D,kBAAkB+C,KAAKhB;oBACnC,GAAIgB,IAAI4B,kBAAkB,KAAKC,YAC3B,CAAC,IACD;wBAACD,oBAAoB5B,IAAI4B,kBAAkB;oBAAA,CAAC;oBAChD,GAAI5B,IAAI8B,kBAAkB,KAAKD,YAC3B,CAAC,IACD;wBAACC,oBAAoB9B,IAAI8B,kBAAkB;oBAAA,CAAC;oBAChD,GAAI9B,IAAI+B,aAAa,KAAKF,YAAY,CAAC,IAAI;wBAACE,eAAe/B,IAAI+B,aAAa;oBAAA,CAAC;oBAC7E,GAAI/B,IAAIgC,SAAS,KAAKH,YAAY,CAAC,IAAI;wBAACG,WAAWhC,IAAIgC,SAAS;oBAAA,CAAC;oBACjE,GAAIhC,IAAIiC,eAAe,KAAKJ,YAAY,CAAC,IAAI;wBAACI,iBAAiBjC,IAAIiC,eAAe;oBAAA,CAAC;oBACnF,GAAIjC,IAAIkC,YAAY,KAAKL,YAAY,CAAC,IAAI;wBAACK,cAAclC,IAAIkC,YAAY;oBAAA,CAAC;oBAC1E,GAAIlC,IAAImC,cAAc,KAAKN,YAAY,CAAC,IAAI;wBAACM,gBAAgBnC,IAAImC,cAAc;oBAAA,CAAC;oBAChFC,gBAAgBpC,IAAIqC,MAAM;gBAC5B;aACD;YACDC,mBAAmB5F,kBAAkB6F,SAAS;QAChD,GACGC,IAAI,CAAC,CAACpB,SAAY,CAAA;gBAACpB;gBAAKoB;YAAM,CAAA,GAC9BqB,KAAK,CAAC,OAAOC;YACZjG,IAAIkG,IAAI,CAAC,qDAAqD;gBAC5DrC,OAAON,IAAII,EAAE;gBACbsC,OAAOE,OAAOF;YAChB;YACA,MAAMG,SAAS,MAAMjF,aAAa;gBAChC0C,OAAON,IAAII,EAAE;gBACbtB,QAAQ;gBACRC,SAAS9B,kBAAkB+C,KAAKhB;gBAChCiB,cAAc;YAChB;YACA,OAAO;gBACLD;gBACAoB,QAAQ;oBAACtC,QAAQ;oBAAmB8B,YAAYiC,OAAOhE,UAAU;gBAAA;YACnE;QACF;IACJ;IAEA,IAAImB,IAAI8C,cAAc,KAAKjB,WAAW;QACpC,MAAM,IAAIkB,MAAM,CAAC,uCAAuC,EAAE/C,IAAII,EAAE,EAAE;IACpE;IACA,OAAO5D,aAAae,2BAA2B;QAC7CgE,YAAY,CAAC,IAAI,EAAEvB,IAAII,EAAE,EAAE;QAC3BoB,MAAM;YACJ;gBACEC,aAAaJ,IAAII,WAAW;gBAC5BC,eAAeL,IAAIK,aAAa;gBAChCC,WAAWN,IAAIM,SAAS;gBACxBrB,OAAON,IAAII,EAAE;gBACb0C,gBAAgB9C,IAAI8C,cAAc;gBAClCvE,cAAc8C,IAAI9C,YAAY;gBAC9BqC,YAAY3D,kBAAkB+C,KAAKhB;gBACnCgE,kBAAkBhD,IAAIgD,gBAAgB,IAAI/F,kBAAkB+C,KAAKhB;gBACjE,GAAIgB,IAAI4B,kBAAkB,KAAKC,YAC3B,CAAC,IACD;oBAACD,oBAAoB5B,IAAI4B,kBAAkB;gBAAA,CAAC;gBAChDQ,gBAAgBpC,IAAIqC,MAAM;YAC5B;SACD;QACDC,mBAAmB5F,kBAAkB6F,SAAS;IAChD,GAAGC,IAAI,CAAC,CAACpB,SAAY,CAAA;YACnBpB;YACAoB,QAAQ;gBAACtC,QAAQsC,OAAOtC,MAAM;gBAAE8B,YAAYQ,OAAOR,UAAU;YAAA;QAC/D,CAAA;AACF;AAEA,eAAeM,sBACbhC,QAA2E,EAC3E+D,iBAAgC,EAChCzE,WAAmB;IAInB,MAAM0E,YAAY5F,YAAYkB,gBAAgB;IAC9C,MAAM2E,eAAeC,QAAQC,IAAI,CAAC;WAAInE,SAAS2B,MAAM;KAAG,EAAE2B,IAAI,CAAC,CAACvB,UAAa,CAAA;YAC3EpB,MAAM;YACN,GAAGoB,OAAO;QACZ,CAAA;IACA,MAAMqC,SAAShH,UAAU2G,mBAAmBC,WAAWV,IAAI,CAAC,CAACe,OAC3DA,OAAO;YAAC1D,MAAM;QAAoB,IAAI;YAACA,MAAM;QAAoB;IAEnE,OAAO,MAAMuD,QAAQC,IAAI,CAAC;QAACF;QAAcG;KAAO;AAClD;AAEA,eAAelE,6BACbd,GAAW,EACXU,QAA4B;IAE5B,MAAMwE,SAAS1G,0BAA0BwB,IAAIW,IAAI,EAAED;IACnD,MAAMnB,yBAAyB;QAAC2F;IAAM;AACxC"}

package/dist/temporal/workflows/test-env.d.ts

@@ -0,0 +1,114 @@
+import { TestWorkflowEnvironment } from '@temporalio/testing';
+import { Worker } from '@temporalio/worker';
+import type { RuntimeCompletionStatus } from '#core/workflow-scheduling/runtime-dag.js';
+import type { JobActivationDecision } from '#db/index.js';
+import type { ActivateJobListenerResult, DrainListenerEventsResult, ListenerBufferPeek } from '#db/job-listeners.js';
+import type { RunDag } from '../activities/orchestration-activities.js';
+declare const TASK_QUEUE = "test-orchestration";
+export { TASK_QUEUE };
+export interface TestConfig {
+    dag: RunDag;
+    /** Map of jobId → status the mock runner should signal back */
+    jobResults: Map<string, RuntimeCompletionStatus>;
+    /** If set, enqueueJobExecutionForRunner will throw with this message instead of signaling */
+    enqueueError?: string;
+    /** If true, enqueueJobExecutionForRunner sends two job-finished signals (for dedup testing) */
+    duplicateSignal?: boolean;
+    /** If true, enqueueJobExecutionForRunner does nothing (no signal — for timeout testing) */
+    skipSignal?: boolean;
+    /** If true, signal job-lease-expired instead of job-finished */
+    signalLeaseExpired?: boolean;
+    /** If true, signal BOTH job-finished and job-lease-expired (precedence testing) */
+    signalBoth?: boolean;
+    /** Status resolveLeaseExpiredJobExecutionActivity returns (defaults to 'failed') */
+    leaseExpiredStatus?: RuntimeCompletionStatus;
+    /** If set, resolveJobStatusFromJobExecutionsActivity throws with this message */
+    resolveJobStatusError?: string;
+    /** If set, releaseLeaseActivity throws (non-retryable) to prove the workflow still returns */
+    releaseLeaseError?: string;
+    /** Scripted job activation decisions keyed by job id; defaults to start */
+    activationDecisions?: Map<string, JobActivationDecision>;
+    /** If set, failJobExecutionAsTimedOutActivity throws (for timeout error-path testing) */
+    failJobExecutionAsTimedOutError?: string;
+    /** Effective status returned by the initial running setRunAttemptStatus call */
+    initialRunStatus?: string;
+    /** Effective status returned by a running setJobStatus call */
+    runningJobStatus?: string;
+    /** Result returned by activateJobListenerActivity (defaults to a running listener) */
+    listenerActivated?: ActivateJobListenerResult;
+    /** Scripted drain results consumed in order; once exhausted, drain returns {kind: 'empty'} */
+    drainResults?: DrainListenerEventsResult[];
+    /** Scripted listener buffer peeks consumed in order; once exhausted, returns an empty buffer */
+    peekResults?: ListenerBufferPeek[];
+    /** Result returned by resolveJobListenerActivity (defaults to succeeded) */
+    listenerResolved?: {
+        status: 'succeeded' | 'failed';
+        jobVersion: number;
+    };
+}
+export declare let cfg: TestConfig;
+export declare function setCfg(value: TestConfig): void;
+export interface ActivityCall {
+    name: string;
+    params: unknown;
+}
+export declare let calls: ActivityCall[];
+export declare function resetCalls(): void;
+export declare function callsNamed(name: string): ActivityCall[];
+export declare function setRunAttemptStatusCalls(): Array<{
+    name: string;
+    params: {
+        runAttemptId: string;
+        status: string;
+        version: number;
+    };
+}>;
+export declare function setJobStatusCalls(): Array<{
+    name: string;
+    params: {
+        jobId: string;
+        status: string;
+        version: number;
+        statusReason?: string | null;
+    };
+}>;
+export declare function setExecutionStatusCalls(): Array<{
+    name: string;
+    params: {
+        jobExecutionId: string;
+        status: string;
+        version: number;
+        statusReason?: string | null;
+    };
+}>;
+export declare function listenerFiringOutcomeCalls(): Array<{
+    name: string;
+    params: {
+        outcome: "succeeded" | "failed" | "cancelled";
+    };
+}>;
+export declare function resolveJobListenerCalls(): Array<{
+    name: string;
+    params: {
+        jobId: string;
+        reason: string;
+    };
+}>;
+export declare function settleListenerCalls(): Array<{
+    name: string;
+    params: {
+        jobExecutionId: string;
+        status: "failed" | "cancelled";
+    };
+}>;
+export declare function dagJob(id: string, key: string, deps?: string[], options?: {
+    mode?: RunDag['jobs'][number]['mode'] | undefined;
+    status?: RunDag['jobs'][number]['status'] | undefined;
+    hasActivationCondition?: boolean | undefined;
+}): RunDag['jobs'][number];
+export declare function makeDag(jobs: RunDag['jobs'], workflowRunId?: string): RunDag;
+export declare let testEnv: TestWorkflowEnvironment;
+export declare let worker: Worker;
+export declare function setupEnv(): Promise<void>;
+export declare function teardownEnv(): Promise<void>;
+//# sourceMappingURL=test-env.d.ts.map

package/dist/temporal/workflows/test-env.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-env.d.ts","sourceRoot":"","sources":["../../../src/temporal/workflows/test-env.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,uBAAuB,EAAC,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAC,MAAM,EAAC,MAAM,oBAAoB,CAAC;AAC1C,OAAO,KAAK,EAAC,uBAAuB,EAAC,MAAM,0CAA0C,CAAC;AACtF,OAAO,KAAK,EAAC,qBAAqB,EAAC,MAAM,cAAc,CAAC;AACxD,OAAO,KAAK,EACV,yBAAyB,EACzB,yBAAyB,EACzB,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,2CAA2C,CAAC;AAGtE,QAAA,MAAM,UAAU,uBAAuB,CAAC;AAGxC,OAAO,EAAC,UAAU,EAAC,CAAC;AAOpB,MAAM,WAAW,UAAU;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,+DAA+D;IAC/D,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC;IACjD,6FAA6F;IAC7F,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,+FAA+F;IAC/F,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,2FAA2F;IAC3F,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,gEAAgE;IAChE,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,mFAAmF;IACnF,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,oFAAoF;IACpF,kBAAkB,CAAC,EAAE,uBAAuB,CAAC;IAC7C,iFAAiF;IACjF,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,8FAA8F;IAC9F,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,2EAA2E;IAC3E,mBAAmB,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IACzD,yFAAyF;IACzF,+BAA+B,CAAC,EAAE,MAAM,CAAC;IACzC,gFAAgF;IAChF,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,+DAA+D;IAC/D,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sFAAsF;IACtF,iBAAiB,CAAC,EAAE,yBAAyB,CAAC;IAC9C,8FAA8F;IAC9F,YAAY,CAAC,EAAE,yBAAyB,EAAE,CAAC;IAC3C,gGAAgG;IAChG,WAAW,CAAC,EAAE,kBAAkB,EAAE,CAAC;IACnC,4EAA4E;IAC5E,gBAAgB,CAAC,EAAE;QAAC,MAAM,EAAE,WAAW,GAAG,QAAQ,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAC,CAAC;CACzE;AAED,eAAO,IAAI,GAAG,EAAE,UAAU,CAAC;AAE3B,wBAAgB,MAAM,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI,CAE9C;AAGD,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,eAAO,IAAI,KAAK,EAAE,YAAY,EAAE,CAAC;AAOjC,wBAAgB,UAAU,IAAI,IAAI,CAGjC;AAED,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,YAAY,EAAE,CAEvD;AAED,wBAAgB,wBAAwB,IACM,KAAK,CAAC;IAChD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAC,CAAC;CACjE,CAAC,CACH;AAED,wBAAgB,iBAAiB,IACM,KAAK,CAAC;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;KAAC,CAAC;CACxF,CAAC,CACH;AAED,wBAAgB,uBAAuB,IACS,KAAK,CAAC;IAClD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,cAAc,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;KAAC,CAAC;CACjG,CAAC,CACH;AAED,wBAAgB,0BAA0B,IACoB,KAAK,CAAC;IAChE,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,OAAO,EAAE,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAA;KAAC,CAAC;CACzD,CAAC,CACH;AAED,wBAAgB,uBAAuB,IACc,KAAK,CAAC;IACvD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAC,CAAC;CACzC,CAAC,CACH;AAED,wBAAgB,mBAAmB,IAC0B,KAAK,CAAC;IAC/D,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE;QAAC,cAAc,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,QAAQ,GAAG,WAAW,CAAA;KAAC,CAAC;CAClE,CAAC,CACH;AAMD,wBAAgB,MAAM,CACpB,EAAE,EAAE,MAAM,EACV,GAAG,EAAE,MAAM,EACX,IAAI,GAAE,MAAM,EAAO,EACnB,OAAO,GAAE;IACP,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;IAClD,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,SAAS,CAAC;IACtD,sBAAsB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACzC,GACL,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAiBxB;AAED,wBAAgB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,aAAa,SAAU,GAAG,MAAM,CAU7E;AAMD,eAAO,IAAI,OAAO,EAAE,uBAAuB,CAAC;AAC5C,eAAO,IAAI,MAAM,EAAE,MAAM,CAAC;AAI1B,wBAAsB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAe9C;AAED,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAIjD"}