@axiom-lattice/local-stores 3.1.23 → 3.2.0

package/dist/index.mjs

@@ -4506,6 +4506,9 @@ CREATE TABLE IF NOT EXISTS lt_thread_message_queue (
   custom_run_config TEXT,
   trusted_run_context TEXT,
   execution_mode TEXT CHECK (execution_mode IS NULL OR execution_mode = 'followup'),
+  locked_by TEXT,
+  locked_at TEXT,
+  lease_expires_at TEXT,
   created_at TEXT NOT NULL
 );
 CREATE INDEX IF NOT EXISTS idx_lt_tmq_thread ON lt_thread_message_queue(thread_id, status);
@@ -4524,6 +4527,16 @@ var LocalThreadMessageQueueStore = class {
     if (!names.has("execution_mode")) {
       db.run("ALTER TABLE lt_thread_message_queue ADD COLUMN execution_mode TEXT CHECK (execution_mode IS NULL OR execution_mode = 'followup')");
     }
+    if (!names.has("locked_by")) {
+      db.run("ALTER TABLE lt_thread_message_queue ADD COLUMN locked_by TEXT");
+    }
+    if (!names.has("locked_at")) {
+      db.run("ALTER TABLE lt_thread_message_queue ADD COLUMN locked_at TEXT");
+    }
+    if (!names.has("lease_expires_at")) {
+      db.run("ALTER TABLE lt_thread_message_queue ADD COLUMN lease_expires_at TEXT");
+    }
+    db.run("CREATE INDEX IF NOT EXISTS idx_lt_tmq_lease ON lt_thread_message_queue(status, lease_expires_at)");
   }
   async addMessage(params) {
     const trusted = validateQueueTrust(params);
@@ -4648,6 +4661,37 @@ var LocalThreadMessageQueueStore = class {
     ).all(threadId, ...filter.params);
     return rows2.map(rowToMessage);
   }
+  async claimPendingMessages(threadId, scope, options) {
+    const filter = scopeClause(scope);
+    const limitClause = options.limit === void 0 ? "" : ` LIMIT ${Math.max(0, Math.floor(options.limit))}`;
+    const claimedRows = transaction(this.db, () => {
+      const candidates = this.db.prepare(
+        `SELECT * FROM lt_thread_message_queue WHERE thread_id = ? AND status = 'pending'${filter.sql} ORDER BY priority DESC, sequence_order ASC${limitClause}`
+      ).all(threadId, ...filter.params);
+      const startedAt = Date.now();
+      const lockedAt = new Date(startedAt).toISOString();
+      const leaseExpiresAt = new Date(startedAt + options.ttlMs).toISOString();
+      const claimed = [];
+      for (const row of candidates) {
+        const changes = write(
+          this.db,
+          `UPDATE lt_thread_message_queue SET status = 'processing', locked_by = ?, locked_at = ?, lease_expires_at = ? WHERE id = ? AND status = 'pending'`,
+          [options.owner, lockedAt, leaseExpiresAt, row.id]
+        );
+        if (changes > 0) claimed.push(row);
+      }
+      return claimed;
+    });
+    return claimedRows.map(rowToMessage);
+  }
+  async reclaimExpiredProcessing(threadId, scope) {
+    const filter = scopeClause(scope);
+    return transaction(this.db, () => write(
+      this.db,
+      `UPDATE lt_thread_message_queue SET status = 'pending', locked_by = NULL, locked_at = NULL, lease_expires_at = NULL WHERE thread_id = ? AND status = 'processing' AND lease_expires_at IS NOT NULL AND lease_expires_at <= ?${filter.sql}`,
+      [threadId, nowISO(), ...filter.params]
+    ));
+  }
   async getQueueSize(threadId, scope) {
     const filter = scopeClause(scope);
     const row = this.db.prepare(
@@ -4689,7 +4733,7 @@ var LocalThreadMessageQueueStore = class {
   async resetProcessingToPending(threadId, scope) {
     const filter = scopeClause(scope);
     const result = this.db.prepare(
-      `UPDATE lt_thread_message_queue SET status = 'pending' WHERE thread_id = ? AND status = 'processing'${filter.sql}`
+      `UPDATE lt_thread_message_queue SET status = 'pending', locked_by = NULL, locked_at = NULL, lease_expires_at = NULL WHERE thread_id = ? AND status = 'processing'${filter.sql}`
     ).run(threadId, ...filter.params);
     return result.changes;
   }
@@ -5075,6 +5119,27 @@ var LocalScheduleStorage = class {
       `UPDATE lt_scheduled_tasks SET ${setClauses.join(", ")} WHERE task_id = ?`
     ).run(...values);
   }
+  async claimScheduledRun(taskId) {
+    const result = this.db.prepare(
+      `UPDATE lt_scheduled_tasks SET status = 'running', updated_at = ? WHERE task_id = ? AND status = 'pending'`
+    ).run((/* @__PURE__ */ new Date()).toISOString(), taskId);
+    return result.changes > 0;
+  }
+  /**
+   * Requeues `running` tasks whose `updated_at` is older than `olderThanMs` back
+   * to `pending` (recovery after a crash that left a claim unsettled).
+   *
+   * @param olderThanMs Age threshold in milliseconds.
+   * @returns The number of tasks requeued.
+   */
+  async requeueStaleRunning(olderThanMs) {
+    const now = /* @__PURE__ */ new Date();
+    const cutoff = new Date(now.getTime() - olderThanMs).toISOString();
+    const result = this.db.prepare(
+      `UPDATE lt_scheduled_tasks SET status = 'pending', updated_at = ? WHERE status = 'running' AND updated_at < ?`
+    ).run(now.toISOString(), cutoff);
+    return result.changes;
+  }
   async delete(taskId) {
     this.db.prepare(`DELETE FROM lt_scheduled_tasks WHERE task_id = ?`).run(taskId);
   }