37 lines
1.3 KiB
TypeScript
37 lines
1.3 KiB
TypeScript
/**
|
|
* Temporal client management — connection lifecycle and workflow operations.
|
|
* Uses @temporalio/client (not worker) since the API server only submits and queries workflows.
|
|
*/
|
|
|
|
import { Client, Connection } from '@temporalio/client';
|
|
import type { PipelineProgress } from '@trebuchet/worker/pipeline';
|
|
|
|
export interface TemporalClients {
|
|
readonly client: Client;
|
|
readonly connection: Connection;
|
|
}
|
|
|
|
export async function connectTemporal(address: string): Promise<TemporalClients> {
|
|
console.log(`Connecting to Temporal at ${address}...`);
|
|
const connection = await Connection.connect({ address });
|
|
const client = new Client({ connection });
|
|
console.log('Temporal connected.');
|
|
return { client, connection };
|
|
}
|
|
|
|
export async function disconnectTemporal(clients: TemporalClients): Promise<void> {
|
|
await clients.connection.close();
|
|
}
|
|
|
|
/** Query a workflow's progress via the getProgress query. */
|
|
export async function queryProgress(client: Client, workflowId: string): Promise<PipelineProgress> {
|
|
const handle = client.workflow.getHandle(workflowId);
|
|
return handle.query<PipelineProgress>('getProgress');
|
|
}
|
|
|
|
/** Cancel a running workflow. */
|
|
export async function cancelWorkflow(client: Client, workflowId: string): Promise<void> {
|
|
const handle = client.workflow.getHandle(workflowId);
|
|
await handle.cancel();
|
|
}
|