@tenetkit/pg 0.37.0 → 0.38.1

package/dist/postgres/store-ops.d.ts

@@ -17,5 +17,5 @@ export declare const postgresOperations: (input: {
     readonly requireRun: (runId: string) => Effect.Effect<DecodedRun, RunNotFound | RuntimeUnavailable | SqlError, SqlR>;
     readonly requireClaim: (claim: import("tenetkit/runtime/driver/run-store").ExecutionClaim) => Effect.Effect<void, import("tenetkit/runtime/driver/sql/errors").StaleClaim | RunNotFound | RuntimeUnavailable | SqlError, SqlR>;
     readonly nextId: (prefix: string) => Effect.Effect<string>;
-}) => Pick<RunStoreInterface, "recordOperation" | "startOperation" | "completeOperation" | "commitModelResponse" | "commitInterruptedModelResponse" | "expireRunningOperation" | "getOperation" | "getOperationByKey" | "resolveOperation">;
+}) => Pick<RunStoreInterface, "recordOperation" | "startOperation" | "completeOperation" | "commitModelResponse" | "commitInterruptedModelResponse" | "expireRunningOperation" | "recoverRunningOperations" | "getOperation" | "getOperationByKey" | "resolveOperation">;
 export {};

package/dist/postgres/store-ops.js

@@ -18,6 +18,38 @@ import { lockRunHierarchy } from "./locks.js";
 export const postgresOperations = (input) => {
     const { sql, hub, run, runNoTxn, requireRun, requireClaim, nextId } = input;
     const fenced = (claim, effect) => run(sql `SELECT run_id FROM tenetkit_runs WHERE run_id = ${claim.runId} FOR UPDATE`.pipe(Effect.andThen(requireClaim(claim)), Effect.andThen(effect)));
+    const expire = (op) => Effect.gen(function* () {
+        const loaded = yield* requireRun(op.runId);
+        const rows = yield* sql `
+        SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
+      `;
+        const row = rows[0];
+        if (row === undefined)
+            return yield* RuntimeUnavailable.make({ message: "operation missing" });
+        if (row.status !== "running") {
+            return { record: toOperationRecord(row), outcome: row.status };
+        }
+        if (canBlindRetry(row.replay_policy)) {
+            yield* sql `
+          UPDATE tenetkit_run_operations
+          SET status = 'requested', started_at = NULL, finished_at = NULL, owner_worker_id = NULL, lease_expires_at = NULL
+          WHERE run_id = ${op.runId} AND operation_id = ${op.operationId} AND status = 'running'
+        `;
+            const next = yield* sql `
+          SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
+        `;
+            return { record: toOperationRecord(next[0]), outcome: "retried" };
+        }
+        yield* sql `
+        UPDATE tenetkit_run_operations SET status = 'unknown', finished_at = NOW()
+        WHERE run_id = ${op.runId} AND operation_id = ${op.operationId} AND status = 'running'
+      `;
+        yield* appendEvent(hub, loaded, { _tag: "OperationUnknown", operationId: op.operationId }, loaded.cancellationRequested ? "cancelling" : "needs-resolution");
+        const next = yield* sql `
+        SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
+      `;
+        return { record: toOperationRecord(next[0]), outcome: "unknown" };
+    });
     return {
         recordOperation: (op) => fenced(op, Effect.gen(function* () {
             const loaded = yield* requireRun(op.runId);
@@ -212,37 +244,18 @@ export const postgresOperations = (input) => {
             return toOperationRecord(rows[0]);
         })),
         ...postgresModelResponseOperations(input),
-        expireRunningOperation: (op) => fenced(op, Effect.gen(function* () {
-            const loaded = yield* requireRun(op.runId);
-            const rows = yield* sql `
-            SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
+        expireRunningOperation: (op) => fenced(op, expire(op)),
+        recoverRunningOperations: (claim) => fenced(claim, Effect.gen(function* () {
+            const operations = yield* sql `
+            SELECT operation_id FROM tenetkit_run_operations
+            WHERE run_id = ${claim.runId} AND status = 'running'
+            ORDER BY operation_id
           `;
-            const row = rows[0];
-            if (row === undefined)
-                return yield* RuntimeUnavailable.make({ message: "operation missing" });
-            if (row.status !== "running") {
-                return { record: toOperationRecord(row), outcome: row.status };
-            }
-            if (canBlindRetry(row.replay_policy)) {
-                yield* sql `
-              UPDATE tenetkit_run_operations
-              SET status = 'requested', started_at = NULL, finished_at = NULL, owner_worker_id = NULL, lease_expires_at = NULL
-              WHERE run_id = ${op.runId} AND operation_id = ${op.operationId} AND status = 'running'
-            `;
-                const next = yield* sql `
-              SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
-            `;
-                return { record: toOperationRecord(next[0]), outcome: "retried" };
+            for (const operation of operations) {
+                yield* expire({ runId: claim.runId, operationId: operation.operation_id });
             }
-            yield* sql `
-            UPDATE tenetkit_run_operations SET status = 'unknown', finished_at = NOW()
-            WHERE run_id = ${op.runId} AND operation_id = ${op.operationId} AND status = 'running'
-          `;
-            yield* appendEvent(hub, loaded, { _tag: "OperationUnknown", operationId: op.operationId }, loaded.cancellationRequested ? "cancelling" : "needs-resolution");
-            const next = yield* sql `
-            SELECT * FROM tenetkit_run_operations WHERE run_id = ${op.runId} AND operation_id = ${op.operationId}
-          `;
-            return { record: toOperationRecord(next[0]), outcome: "unknown" };
+            const loaded = yield* requireRun(claim.runId);
+            return loaded.status === "needs-resolution" ? "blocked" : "ready";
         })),
         getOperation: (op) => runNoTxn(Effect.gen(function* () {
             yield* requireRun(op.runId);

package/dist/postgres/store.js

@@ -35,6 +35,7 @@ import { lockMailbox, lockRun, lockRunHierarchy, lockSpawnParent } from "./locks
 import { inspectionStoreMethods } from "./store-inspection.js";
 import { programStoreMethods } from "./store-program.js";
 import { admitStart as admitExactStart } from "tenetkit/runtime/driver/sql/store-admit";
+import { activateRoot } from "tenetkit/runtime/driver/sql/store-activate";
 import { admitSend } from "./store-admit.js";
 import { associateRegistrations, loadRegistrations } from "tenetkit/runtime/driver/sql/executable-registrations";
 import { narrow } from "tenetkit/runtime/driver/executable-registration";
@@ -71,7 +72,8 @@ export const makePostgresServices = (options) => Effect.gen(function* () {
         sessionStore: (sessionId) => Effect.succeed(Option.some(makePostgresSessionStore({ sessionId, run, runNoTxn }))),
         hasAdmission: (input) => runNoTxn(hasAdmission(input)),
         admitSend: (input) => run(admitSend(transactionHub, addressBindings, nextId, input)),
-        admitStart: (input) => run(sql `SELECT pg_advisory_xact_lock(hashtext('tenetkit:executable-registrations'))`.pipe(Effect.andThen(admitExactStart(transactionHub, input)))),
+        admitStart: (input, startOptions) => run(sql `SELECT pg_advisory_xact_lock(hashtext('tenetkit:executable-registrations'))`.pipe(Effect.andThen(admitExactStart(transactionHub, input, startOptions)))),
+        activate: (input) => run(lockRun(input.runId).pipe(Effect.andThen(activateRoot(transactionHub, input.runId)))),
         admitSpawn: (input) => run(Effect.gen(function* () {
             const parent = yield* lockSpawnParent(input.parentRunId);
             const executableRef = resolveChild(parent.executableRef, parent.executableManifest, input.selection);

package/package.json

@@ -1,6 +1,6 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
-  "version": "0.37.0",
+  "version": "0.38.1",
   "private": false,
   "license": "MIT",
   "type": "module",
@@ -39,7 +39,7 @@
   },
   "dependencies": {
     "@effect/sql-pg": "4.0.0-rc.111",
-    "tenetkit": "0.37.0"
+    "tenetkit": "0.38.1"
   },
   "devDependencies": {
     "@effect/vitest": "4.0.0-rc.111",