@smithers-orchestrator/db 0.21.0 → 0.22.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@smithers-orchestrator/db",
-  "version": "0.21.0",
+  "version": "0.22.0",
   "description": "SQLite and Drizzle persistence adapter for Smithers workflows",
   "type": "module",
   "sideEffects": false,
@@ -26,10 +26,10 @@
     "drizzle-zod": "^0.8.3",
     "effect": "^3.21.1",
     "zod": "^4.3.6",
-    "@smithers-orchestrator/errors": "0.21.0",
-    "@smithers-orchestrator/graph": "0.21.0",
-    "@smithers-orchestrator/scheduler": "0.21.0",
-    "@smithers-orchestrator/observability": "0.21.0"
+    "@smithers-orchestrator/errors": "0.22.0",
+    "@smithers-orchestrator/observability": "0.22.0",
+    "@smithers-orchestrator/scheduler": "0.22.0",
+    "@smithers-orchestrator/graph": "0.22.0"
   },
   "devDependencies": {
     "@types/bun": "latest",

package/src/adapter.js

@@ -10,7 +10,7 @@
 /** @typedef {import("./adapter/StaleRunRecord.ts").StaleRunRecord} StaleRunRecord */
 // @smithers-type-exports-end
 
-import { getTableName, sql } from "drizzle-orm";
+import { getTableName } from "drizzle-orm";
 import { Effect, Exit, FiberId, Metric } from "effect";
 import { toSmithersError } from "@smithers-orchestrator/errors/toSmithersError";
 import { getSqlMessageStorage } from "./sql-message-storage.js";
@@ -1111,9 +1111,11 @@ export class SmithersDb {
    */
     getRawNodeOutput(tableName, runId, nodeId) {
         return runnableEffect(this.read(`get raw node output ${tableName}`, () => {
-            const query = sql.raw(`SELECT * FROM "${tableName}" WHERE run_id = '${runId}' AND node_id = '${nodeId}' ORDER BY iteration DESC LIMIT 1`);
-            const res = this.db.get(query);
-            return Promise.resolve(res ?? null);
+            const escaped = tableName.replaceAll(`"`, `""`);
+            const client = this.db.session.client;
+            const stmt = client.query(`SELECT * FROM "${escaped}" WHERE run_id = ? AND node_id = ? ORDER BY iteration DESC LIMIT 1`);
+            const row = stmt.get(runId, nodeId);
+            return Promise.resolve(row ?? null);
         }).pipe(Effect.catchAll(() => Effect.succeed(null))));
     }
     /**

package/src/loadInputEffect.js

@@ -13,16 +13,18 @@ import { SmithersError } from "@smithers-orchestrator/errors/SmithersError";
  * @returns {Effect.Effect<Record<string, unknown> | undefined, SmithersError>}
  */
 export function loadInput(db, inputTable, runId) {
-    const cols = getTableColumns(inputTable);
-    const runIdCol = cols.runId;
-    if (!runIdCol) {
-        throw new SmithersError("DB_MISSING_COLUMNS", "schema.input must include runId column");
-    }
-    return Effect.tryPromise({
-        try: () => db.select().from(inputTable).where(eq(runIdCol, runId)).limit(1),
-        catch: (cause) => toSmithersError(cause, "load input", {
-            code: "DB_QUERY_FAILED",
-            details: { runId },
-        }),
-    }).pipe(Effect.map((rows) => rows[0]), Effect.annotateLogs({ runId }), Effect.withLogSpan("db:load-input"));
+    return Effect.suspend(() => {
+        const cols = getTableColumns(inputTable);
+        const runIdCol = cols.runId;
+        if (!runIdCol) {
+            return Effect.fail(new SmithersError("DB_MISSING_COLUMNS", "schema.input must include runId column"));
+        }
+        return Effect.tryPromise({
+            try: () => db.select().from(inputTable).where(eq(runIdCol, runId)).limit(1),
+            catch: (cause) => toSmithersError(cause, "load input", {
+                code: "DB_QUERY_FAILED",
+                details: { runId },
+            }),
+        }).pipe(Effect.map((rows) => rows[0]));
+    }).pipe(Effect.annotateLogs({ runId }), Effect.withLogSpan("db:load-input"));
 }

package/src/snapshot.js

@@ -53,18 +53,20 @@ function coerceBooleanColumns(rows, boolKeys) {
  * @returns {Effect.Effect<Record<string, unknown> | undefined, SmithersError>}
  */
 export function loadInputEffect(db, inputTable, runId) {
-    const cols = getTableColumns(inputTable);
-    const runIdCol = cols.runId;
-    if (!runIdCol) {
-        throw new SmithersError("DB_MISSING_COLUMNS", "schema.input must include runId column");
-    }
-    return Effect.tryPromise({
-        try: () => db.select().from(inputTable).where(eq(runIdCol, runId)).limit(1),
-        catch: (cause) => toSmithersError(cause, "load input", {
-            code: "DB_QUERY_FAILED",
-            details: { runId },
-        }),
-    }).pipe(Effect.map((rows) => rows[0]), Effect.annotateLogs({ runId }), Effect.withLogSpan("db:load-input"));
+    return Effect.suspend(() => {
+        const cols = getTableColumns(inputTable);
+        const runIdCol = cols.runId;
+        if (!runIdCol) {
+            return Effect.fail(new SmithersError("DB_MISSING_COLUMNS", "schema.input must include runId column"));
+        }
+        return Effect.tryPromise({
+            try: () => db.select().from(inputTable).where(eq(runIdCol, runId)).limit(1),
+            catch: (cause) => toSmithersError(cause, "load input", {
+                code: "DB_QUERY_FAILED",
+                details: { runId },
+            }),
+        }).pipe(Effect.map((rows) => rows[0]));
+    }).pipe(Effect.annotateLogs({ runId }), Effect.withLogSpan("db:load-input"));
 }
 /**
  * @param {BunSQLiteDatabase<Record<string, unknown>>} db

package/src/sql-message-storage.js

@@ -3,7 +3,7 @@ import * as SqlClient from "@effect/sql/SqlClient";
 import { SqlError } from "@effect/sql/SqlError";
 import * as Statement from "@effect/sql/Statement";
 import { Database } from "bun:sqlite";
-import { Context, Effect, Layer, ManagedRuntime, Scope } from "effect";
+import { Context, Effect, Layer, ManagedRuntime, Scope, Stream } from "effect";
 import { runSmithersSchemaMigrations } from "./schema-migrations.js";
 import { camelToSnake } from "./utils/camelToSnake.js";
 /** @typedef {import("drizzle-orm/bun-sqlite").BunSQLiteDatabase} BunSQLiteDatabase */
@@ -494,7 +494,7 @@ function createConnection(sqlite) {
             }
         }),
         executeUnprepared: (statement, params, transformRows) => execute(statement, params, transformRows),
-        executeStream: () => Effect.dieMessage("executeStream not implemented"),
+        executeStream: (statement, params, transformRows) => Stream.fromIterableEffect(execute(statement, params, transformRows)),
     };
 }
 /**