@drej/postgres 0.1.2 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @drej/postgres
 
+## 0.2.0
+
+### Minor Changes
+
+- 2fd33e0: feat: run management API
+
+  Add `RunStatus` enum, `RunDetails` type, and `ListRunsOptions` for filtering. Replace `listRuns()` with `listRunDetails()`, `listAllRunDetails()`, `getRunDetails()`, and `deleteRun()` on both `IStorageAdapter` and `DrejClient`. Add `WorkflowRun.status` property that tracks execution state as events are consumed.
+
+### Patch Changes
+
+- Updated dependencies [22b8a32]
+- Updated dependencies [799b6dd]
+- Updated dependencies [8d9d8bb]
+- Updated dependencies [2fd33e0]
+- Updated dependencies [0d94c2a]
+  - @drej/core@0.3.0
+
 ## 0.1.2
 
 ### Patch Changes

package/package.json

@@ -1,14 +1,16 @@
 {
   "name": "@drej/postgres",
-  "version": "0.1.2",
-  "publishConfig": { "access": "public" },
+  "version": "0.2.0",
+  "publishConfig": {
+    "access": "public"
+  },
   "main": "./src/index.ts",
   "dependencies": {
-    "postgres": "^3.4.5",
+    "postgres": "3.4.9",
     "@drej/core": "workspace:*"
   },
   "devDependencies": {
-    "bun-types": "latest",
-    "typescript": "latest"
+    "bun-types": "1.3.14",
+    "typescript": "6.0.3"
   }
 }

package/src/adapter.ts

@@ -1,5 +1,6 @@
 import postgres from "postgres";
-import type { IStorageAdapter, LedgerEntry, LedgerEvent } from "@drej/core";
+import type { IStorageAdapter, LedgerEntry, LedgerEvent, RunDetails, ListRunsOptions } from "@drej/core";
+import { RunStatus } from "@drej/core";
 import { MIGRATION_SQL } from "./migrations";
 
 type Row = {
@@ -13,6 +14,42 @@ type Row = {
   ts: string;
 };
 
+type AggRow = {
+  wf_name: string;
+  run_id: string;
+  started_at: string | null;
+  completed_at: string | null;
+  terminal_event: string | null;
+  error_msg: string | null;
+  step_count: string;
+};
+
+function terminalToStatus(event: string | null): RunStatus {
+  if (event === "workflow_complete") return RunStatus.Completed;
+  if (event === "workflow_failed") return RunStatus.Failed;
+  return RunStatus.Running;
+}
+
+function aggRowToDetails(row: AggRow): RunDetails {
+  return {
+    workflowName: row.wf_name,
+    runId: row.run_id,
+    status: terminalToStatus(row.terminal_event),
+    startedAt: Number(row.started_at),
+    completedAt: row.completed_at != null ? Number(row.completed_at) : undefined,
+    stepCount: Number(row.step_count),
+    error: row.error_msg ?? undefined,
+  };
+}
+
+function applyOpts(details: RunDetails[], opts?: ListRunsOptions): RunDetails[] {
+  let result = details;
+  if (opts?.before != null) result = result.filter((d) => d.startedAt < opts.before!);
+  if (opts?.status != null) result = result.filter((d) => d.status === opts.status);
+  if (opts?.limit != null) result = result.slice(0, opts.limit);
+  return result;
+}
+
 function rowToEntry(row: Row): LedgerEntry {
   return {
     runId: row.run_id,
@@ -78,10 +115,42 @@ export class PostgresAdapter implements IStorageAdapter {
     return rows.length ? rowToEntry(rows[0]) : null;
   }
 
-  async listRuns(workflowName: string): Promise<string[]> {
-    const rows = await this.sql<{ run_id: string }[]>`
-      SELECT DISTINCT run_id FROM drej_events WHERE wf_name = ${workflowName}
-    `;
-    return rows.map((r) => r.run_id);
+  private async _aggQuery(whereClause: string, params: string[]): Promise<AggRow[]> {
+    return this.sql.unsafe<AggRow[]>(
+      `WITH agg AS (
+        SELECT
+          wf_name,
+          run_id,
+          MIN(CASE WHEN event = 'run_started' THEN ts END) AS started_at,
+          MAX(CASE WHEN event IN ('workflow_complete', 'workflow_failed') THEN ts END) AS completed_at,
+          MAX(CASE WHEN event IN ('workflow_complete', 'workflow_failed') THEN event END) AS terminal_event,
+          MAX(CASE WHEN event = 'workflow_failed' THEN error END) AS error_msg,
+          COUNT(CASE WHEN event = 'step_complete' THEN 1 END)::int AS step_count
+        FROM drej_events
+        ${whereClause}
+        GROUP BY wf_name, run_id
+      )
+      SELECT * FROM agg WHERE started_at IS NOT NULL ORDER BY started_at DESC`,
+      params,
+    );
+  }
+
+  async listRunDetails(workflowName: string, opts?: ListRunsOptions): Promise<RunDetails[]> {
+    const rows = await this._aggQuery("WHERE wf_name = $1", [workflowName]);
+    return applyOpts(rows.map(aggRowToDetails), opts);
+  }
+
+  async listAllRunDetails(opts?: ListRunsOptions): Promise<RunDetails[]> {
+    const rows = await this._aggQuery("", []);
+    return applyOpts(rows.map(aggRowToDetails), opts);
+  }
+
+  async getRunDetails(workflowName: string, runId: string): Promise<RunDetails | null> {
+    const rows = await this._aggQuery("WHERE wf_name = $1 AND run_id = $2", [workflowName, runId]);
+    return rows.length ? aggRowToDetails(rows[0]) : null;
+  }
+
+  async deleteRun(workflowName: string, runId: string): Promise<void> {
+    await this.sql`DELETE FROM drej_events WHERE wf_name = ${workflowName} AND run_id = ${runId}`;
   }
 }