@botbotgo/agent-harness 0.0.290 → 0.0.291

package/dist/persistence/sqlite-run-queue-store.js

@@ -1,120 +0,0 @@
-function asString(value) {
-    return typeof value === "string" ? value : String(value ?? "");
-}
-function asNullableString(value) {
-    return value == null ? null : asString(value);
-}
-function asBoolean(value) {
-    return value === true || value === 1 || value === "1";
-}
-function nowIso() {
-    return new Date(Date.now()).toISOString();
-}
-export class SqliteRunQueueStore {
-    db;
-    constructor(db) {
-        this.db = db;
-    }
-    mapQueuedRun(row) {
-        return {
-            runId: asString(row.run_id),
-            threadId: asString(row.thread_id),
-            priority: Number(row.priority ?? 0),
-            queueKey: asNullableString(row.queue_key),
-            enqueuedAt: asString(row.enqueued_at),
-            availableAt: asString(row.available_at),
-            claimedBy: asNullableString(row.claimed_by),
-            claimedAt: asNullableString(row.claimed_at),
-            leaseExpiresAt: asNullableString(row.lease_expires_at),
-            attemptCount: Number(row.attempt_count ?? 0),
-            lastError: asNullableString(row.last_error),
-        };
-    }
-    mapRunControl(row) {
-        return {
-            runId: asString(row.run_id),
-            cancelRequested: asBoolean(row.cancel_requested),
-            cancelReason: asNullableString(row.cancel_reason),
-            cancelRequestedAt: asNullableString(row.cancel_requested_at),
-            heartbeatAt: asNullableString(row.heartbeat_at),
-            workerId: asNullableString(row.worker_id),
-            workerStartedAt: asNullableString(row.worker_started_at),
-        };
-    }
-    async enqueueRun(input) {
-        const enqueuedAt = nowIso();
-        await this.db.execute(`INSERT OR REPLACE INTO run_queue
-       (run_id, thread_id, priority, queue_key, enqueued_at, available_at, claimed_by, claimed_at, lease_expires_at, attempt_count, last_error)
-       VALUES (?, ?, ?, ?, ?, ?, NULL, NULL, NULL, COALESCE((SELECT attempt_count FROM run_queue WHERE run_id = ?), 0), NULL)`, [
-            input.runId,
-            input.threadId,
-            input.priority ?? 0,
-            input.queueKey ?? null,
-            enqueuedAt,
-            input.availableAt ?? enqueuedAt,
-            input.runId,
-        ]);
-    }
-    async getQueuedRun(runId) {
-        const row = await this.db.selectOne("SELECT * FROM run_queue WHERE run_id = ?", [runId]);
-        return row ? this.mapQueuedRun(row) : null;
-    }
-    async claimQueuedRun(input) {
-        const claimedAt = input.claimedAt ?? nowIso();
-        await this.db.execute(`UPDATE run_queue
-       SET claimed_by = ?, claimed_at = ?, lease_expires_at = ?, attempt_count = attempt_count + 1
-       WHERE run_id = ? AND thread_id = ?`, [input.workerId, claimedAt, input.leaseExpiresAt, input.runId, input.threadId]);
-        await this.db.execute(`UPDATE run_control
-       SET heartbeat_at = ?, worker_id = ?, worker_started_at = COALESCE(worker_started_at, ?)
-       WHERE run_id = ?`, [claimedAt, input.workerId, claimedAt, input.runId]);
-        const claimed = await this.getQueuedRun(input.runId);
-        if (!claimed) {
-            throw new Error(`Missing queued run ${input.runId}`);
-        }
-        return claimed;
-    }
-    async renewRunLease(input) {
-        const heartbeatAt = input.heartbeatAt ?? nowIso();
-        await this.db.execute(`UPDATE run_queue
-       SET lease_expires_at = ?, claimed_by = COALESCE(claimed_by, ?), claimed_at = COALESCE(claimed_at, ?)
-       WHERE run_id = ?`, [input.leaseExpiresAt, input.workerId, heartbeatAt, input.runId]);
-        await this.db.execute(`UPDATE run_control
-       SET heartbeat_at = ?, worker_id = ?, worker_started_at = COALESCE(worker_started_at, ?)
-       WHERE run_id = ?`, [heartbeatAt, input.workerId, heartbeatAt, input.runId]);
-    }
-    async releaseRunClaim(runId) {
-        await this.db.execute("DELETE FROM run_queue WHERE run_id = ?", [runId]);
-        await this.db.execute(`UPDATE run_control
-       SET heartbeat_at = NULL, worker_id = NULL, worker_started_at = NULL
-       WHERE run_id = ?`, [runId]);
-    }
-    async listExpiredClaimedRuns(cutoffIso) {
-        const rows = await this.db.selectAll(`SELECT *
-       FROM run_queue
-       WHERE claimed_by IS NOT NULL
-         AND lease_expires_at IS NOT NULL
-         AND lease_expires_at <= ?
-       ORDER BY lease_expires_at ASC, run_id ASC`, [cutoffIso]);
-        return rows.map((row) => this.mapQueuedRun(row));
-    }
-    async getRunControl(runId) {
-        const row = await this.db.selectOne("SELECT * FROM run_control WHERE run_id = ?", [runId]);
-        return row ? this.mapRunControl(row) : null;
-    }
-    async requestRunCancel(runId, reason) {
-        const requestedAt = nowIso();
-        await this.db.execute(`UPDATE run_control
-       SET cancel_requested = 1, cancel_reason = ?, cancel_requested_at = ?
-       WHERE run_id = ?`, [reason ?? null, requestedAt, runId]);
-        const updated = await this.getRunControl(runId);
-        if (!updated) {
-            throw new Error(`Missing run control for ${runId}`);
-        }
-        return updated;
-    }
-    async clearRunCancel(runId) {
-        await this.db.execute(`UPDATE run_control
-       SET cancel_requested = 0, cancel_reason = NULL, cancel_requested_at = NULL
-       WHERE run_id = ?`, [runId]);
-    }
-}