73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import { runTaskProcessor, type TaskExecutionOutcome } from '@/lib/server/task-processors';
|
|
import {
|
|
completeTask,
|
|
getTaskById,
|
|
markTaskFailure,
|
|
markTaskRunning
|
|
} from '@/lib/server/repos/tasks';
|
|
import type { Task } from '@/lib/types';
|
|
|
|
export async function runTaskWorkflow(taskId: string) {
|
|
'use workflow';
|
|
|
|
const task = await loadTaskStep(taskId);
|
|
if (!task) {
|
|
return;
|
|
}
|
|
|
|
await markTaskRunningStep(task.id);
|
|
|
|
try {
|
|
const refreshedTask = await loadTaskStep(task.id);
|
|
if (!refreshedTask) {
|
|
return;
|
|
}
|
|
|
|
const outcome = await processTaskStep(refreshedTask);
|
|
await completeTaskStep(task.id, outcome);
|
|
} catch (error) {
|
|
const reason = error instanceof Error
|
|
? error.message
|
|
: 'Task failed unexpectedly';
|
|
const latestTask = await loadTaskStep(task.id);
|
|
await markTaskFailureStep(task.id, reason, latestTask);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
async function loadTaskStep(taskId: string) {
|
|
'use step';
|
|
return await getTaskById(taskId);
|
|
}
|
|
|
|
async function markTaskRunningStep(taskId: string) {
|
|
'use step';
|
|
await markTaskRunning(taskId);
|
|
}
|
|
|
|
async function processTaskStep(task: Task) {
|
|
'use step';
|
|
return await runTaskProcessor(task);
|
|
}
|
|
|
|
// Keep retries at the projection workflow level to avoid duplicate side effects.
|
|
(
|
|
processTaskStep as ((task: Task) => Promise<TaskExecutionOutcome>) & { maxRetries?: number }
|
|
).maxRetries = 0;
|
|
|
|
async function completeTaskStep(taskId: string, outcome: TaskExecutionOutcome) {
|
|
'use step';
|
|
await completeTask(taskId, outcome.result, {
|
|
detail: outcome.completionDetail,
|
|
context: outcome.completionContext ?? null
|
|
});
|
|
}
|
|
|
|
async function markTaskFailureStep(taskId: string, reason: string, latestTask: Task | null) {
|
|
'use step';
|
|
await markTaskFailure(taskId, reason, 'failed', {
|
|
detail: reason,
|
|
context: latestTask?.stage_context ?? null
|
|
});
|
|
}
|