@workglow/storage 0.2.25 → 0.2.27

package/dist/browser.js

@@ -45,6 +45,21 @@ class StorageUnsupportedError extends StorageError {
   }
 }
 
+// src/tabular/CoveringIndexMissingError.ts
+class CoveringIndexMissingError extends StorageError {
+  static type = "CoveringIndexMissingError";
+  table;
+  requiredColumns;
+  registeredIndexes;
+  constructor(table, requiredColumns, registeredIndexes) {
+    const indexList = registeredIndexes.map((cols) => `[${cols.join(", ")}]`).join(", ");
+    super(`No covering index for table "${table}". ` + `Required columns: [${requiredColumns.join(", ")}]. ` + `Registered indexes: ${indexList || "(none)"}.`);
+    this.table = table;
+    this.requiredColumns = requiredColumns;
+    this.registeredIndexes = registeredIndexes;
+  }
+}
+
 // src/tabular/BaseTabularStorage.ts
 var TABULAR_REPOSITORY = createServiceToken("storage.tabularRepository");
 
@@ -168,6 +183,15 @@ class BaseTabularStorage {
     const entities = await this.query(criteria);
     return entities?.length ?? 0;
   }
+  queryIndex(criteria, options) {
+    this.validateSelect(options);
+    const requiredColumns = [
+      ...Object.keys(criteria),
+      ...(options.orderBy ?? []).map((o) => String(o.column)),
+      ...options.select.map(String)
+    ];
+    throw new CoveringIndexMissingError(this.constructor.name, requiredColumns, []);
+  }
   async* records(pageSize = 100) {
     if (pageSize <= 0) {
       throw new RangeError(`pageSize must be greater than 0, got ${pageSize}`);
@@ -263,6 +287,18 @@ class BaseTabularStorage {
       }
     }
   }
+  validateSelect(options) {
+    if (!options.select || options.select.length === 0) {
+      throw new StorageValidationError("queryIndex requires a non-empty select array");
+    }
+    const schemaProps = Object.keys(this.schema.properties);
+    for (const col of options.select) {
+      const colStr = String(col);
+      if (!schemaProps.includes(colStr)) {
+        throw new StorageValidationError(`queryIndex select column "${colStr}" is not in schema`);
+      }
+    }
+  }
   primaryKeyColumns() {
     const columns = [];
     for (const key of Object.keys(this.primaryKeySchema.properties)) {
@@ -383,6 +419,61 @@ import { createServiceToken as createServiceToken3, getLogger } from "@workglow/
 
 // src/tabular/InMemoryTabularStorage.ts
 import { createServiceToken as createServiceToken2, makeFingerprint as makeFingerprint2, uuid4 } from "@workglow/util";
+
+// src/tabular/coveringIndexPicker.ts
+function pickCoveringIndex(input) {
+  const { table, indexes, criteriaColumns, orderByColumns, selectColumns, primaryKeyColumns } = input;
+  const required = uniqueColumns([
+    ...criteriaColumns,
+    ...orderByColumns.map((o) => o.column),
+    ...selectColumns
+  ]);
+  const pkSet = new Set(primaryKeyColumns);
+  for (const idx of indexes) {
+    const fit = tryFitIndex(idx, criteriaColumns, orderByColumns, selectColumns, pkSet);
+    if (fit !== undefined) {
+      return { name: idx.name, keyPath: idx.keyPath, reverseDirection: fit.reverseDirection };
+    }
+  }
+  throw new CoveringIndexMissingError(table, required, indexes.map((i) => i.keyPath));
+}
+function tryFitIndex(index, criteria, orderBy, select, pkSet) {
+  const keyPath = index.keyPath;
+  const criteriaSet = new Set(criteria);
+  if (criteria.length > keyPath.length)
+    return;
+  for (let i = 0;i < criteria.length; i++) {
+    if (!criteriaSet.has(keyPath[i]))
+      return;
+  }
+  let reverseDirection = false;
+  if (orderBy.length > 0) {
+    const start = criteria.length;
+    if (start + orderBy.length > keyPath.length)
+      return;
+    for (let i = 0;i < orderBy.length; i++) {
+      if (keyPath[start + i] !== orderBy[i].column)
+        return;
+    }
+    const allDesc = orderBy.every((o) => o.direction === "DESC");
+    const allAsc = orderBy.every((o) => o.direction === "ASC");
+    if (allDesc)
+      reverseDirection = true;
+    else if (!allAsc)
+      return;
+  }
+  const keyPathSet = new Set(keyPath);
+  for (const col of select) {
+    if (!keyPathSet.has(col) && !pkSet.has(col))
+      return;
+  }
+  return { reverseDirection };
+}
+function uniqueColumns(cols) {
+  return Array.from(new Set(cols));
+}
+
+// src/tabular/InMemoryTabularStorage.ts
 var MEMORY_TABULAR_REPOSITORY = createServiceToken2("storage.tabularRepository.inMemory");
 
 class InMemoryTabularStorage extends BaseTabularStorage {
@@ -625,6 +716,38 @@ class InMemoryTabularStorage extends BaseTabularStorage {
     this.events.emit("query", criteria, result);
     return result;
   }
+  async queryIndex(criteria, options) {
+    this.validateSelect(options);
+    this.validateQueryParams(criteria, options);
+    const registered = this.indexes.map((cols, i) => ({
+      name: `idx_${i}`,
+      keyPath: cols
+    }));
+    pickCoveringIndex({
+      table: "InMemoryTabularStorage",
+      indexes: registered,
+      criteriaColumns: Object.keys(criteria),
+      orderByColumns: (options.orderBy ?? []).map((o) => ({
+        column: String(o.column),
+        direction: o.direction
+      })),
+      selectColumns: options.select.map(String),
+      primaryKeyColumns: this.primaryKeyNames.map(String)
+    });
+    const full = await this.query(criteria, {
+      orderBy: options.orderBy,
+      limit: options.limit,
+      offset: options.offset
+    });
+    if (!full)
+      return [];
+    return full.map((row) => {
+      const out = {};
+      for (const k of options.select)
+        out[k] = row[k];
+      return out;
+    });
+  }
   subscribeToChanges(callback, options) {
     const handlePut = (entity) => {
       callback({ type: this._lastPutWasInsert ? "INSERT" : "UPDATE", new: entity });
@@ -767,6 +890,10 @@ class CachedTabularStorage extends BaseTabularStorage {
     await this.initializeCache();
     return await this.cache.query(criteria, options);
   }
+  async queryIndex(criteria, options) {
+    await this.initializeCache();
+    return await this.cache.queryIndex(criteria, options);
+  }
   async deleteSearch(criteria) {
     await this.initializeCache();
     await this.durable.deleteSearch(criteria);
@@ -1207,6 +1334,9 @@ class TelemetryTabularStorage {
   query(criteria, options) {
     return traced("workglow.storage.tabular.query", this.storageName, () => this.inner.query(criteria, options));
   }
+  queryIndex(criteria, options) {
+    return traced("workglow.storage.tabular.queryIndex", this.storageName, () => this.inner.queryIndex(criteria, options));
+  }
   records(pageSize) {
     return this.inner.records(pageSize);
   }
@@ -1444,6 +1574,7 @@ var IN_MEMORY_QUEUE_STORAGE = createServiceToken9("jobqueue.storage.inMemory");
 
 class InMemoryQueueStorage {
   queueName;
+  scope = "process";
   prefixValues;
   events = new EventEmitter3;
   constructor(queueName, options) {
@@ -1653,6 +1784,9 @@ class TelemetryQueueStorage {
     this.storageName = storageName;
     this.inner = inner;
   }
+  get scope() {
+    return this.inner.scope;
+  }
   add(job) {
     return traced("workglow.storage.queue.add", this.storageName, () => this.inner.add(job));
   }
@@ -1703,13 +1837,15 @@ class TelemetryQueueStorage {
   }
 }
 // src/queue-limiter/InMemoryRateLimiterStorage.ts
-import { createServiceToken as createServiceToken10, sleep as sleep2 } from "@workglow/util";
+import { createServiceToken as createServiceToken10, sleep as sleep2, uuid4 as uuid43 } from "@workglow/util";
 var IN_MEMORY_RATE_LIMITER_STORAGE = createServiceToken10("ratelimiter.storage.inMemory");
 
 class InMemoryRateLimiterStorage {
+  scope = "process";
   prefixValues;
   executions = new Map;
   nextAvailableTimes = new Map;
+  reserveChains = new Map;
   constructor(options) {
     this.prefixValues = options?.prefixValues ?? {};
   }
@@ -1718,11 +1854,67 @@ class InMemoryRateLimiterStorage {
     return prefixPart ? `${prefixPart}|${queueName}` : queueName;
   }
   async setupDatabase() {}
+  async withKeyLock(key, fn) {
+    const previous = this.reserveChains.get(key) ?? Promise.resolve();
+    let release;
+    const next = new Promise((resolve) => {
+      release = resolve;
+    });
+    this.reserveChains.set(key, next);
+    try {
+      await previous;
+      return await fn();
+    } finally {
+      release(undefined);
+      if (this.reserveChains.get(key) === next) {
+        this.reserveChains.delete(key);
+      }
+    }
+  }
+  async tryReserveExecution(queueName, maxExecutions, windowMs) {
+    const key = this.makeKey(queueName);
+    return this.withKeyLock(key, () => {
+      const now = Date.now();
+      const windowStart = new Date(now - windowMs);
+      const executions = this.executions.get(key) ?? [];
+      const live = executions.filter((e) => e.executedAt > windowStart);
+      if (live.length >= maxExecutions) {
+        if (live.length !== executions.length) {
+          this.executions.set(key, live);
+        }
+        return null;
+      }
+      const next = this.nextAvailableTimes.get(key);
+      if (next && next.getTime() > now) {
+        return null;
+      }
+      const id = uuid43();
+      live.push({ id, queueName, executedAt: new Date(now) });
+      this.executions.set(key, live);
+      return id;
+    });
+  }
+  async releaseExecution(queueName, token) {
+    if (token === null || token === undefined)
+      return;
+    const key = this.makeKey(queueName);
+    await this.withKeyLock(key, () => {
+      const executions = this.executions.get(key);
+      if (!executions || executions.length === 0)
+        return;
+      const idx = executions.findIndex((e) => e.id === token);
+      if (idx === -1)
+        return;
+      executions.splice(idx, 1);
+      this.executions.set(key, executions);
+    });
+  }
   async recordExecution(queueName) {
     await sleep2(0);
     const key = this.makeKey(queueName);
     const executions = this.executions.get(key) ?? [];
     executions.push({
+      id: uuid43(),
       queueName,
       executedAt: new Date
     });
@@ -2304,7 +2496,7 @@ class LazyEncryptedCredentialStore {
   }
 }
 // src/tabular/IndexedDbTabularStorage.ts
-import { createServiceToken as createServiceToken12, deepEqual as deepEqual2, makeFingerprint as makeFingerprint5, uuid4 as uuid43 } from "@workglow/util";
+import { createServiceToken as createServiceToken12, deepEqual as deepEqual2, makeFingerprint as makeFingerprint5, uuid4 as uuid44 } from "@workglow/util";
 
 // src/util/IndexedDbTable.ts
 import { deepEqual } from "@workglow/util";
@@ -2695,7 +2887,7 @@ class IndexedDbTabularStorage extends BaseTabularStorage {
   }
   generateKeyValue(columnName, strategy) {
     if (strategy === "uuid") {
-      return uuid43();
+      return uuid44();
     }
     throw new Error(`IndexedDB autoincrement keys are generated by the database, not client-side. Column: ${columnName}`);
   }
@@ -3196,6 +3388,107 @@ class IndexedDbTabularStorage extends BaseTabularStorage {
       request.onerror = () => reject(request.error);
     });
   }
+  async queryIndex(criteria, options) {
+    this.validateSelect(options);
+    this.validateQueryParams(criteria, options);
+    const registered = this.indexes.map((cols) => {
+      const cs = Array.isArray(cols) ? cols : [cols];
+      return { name: cs.join("_"), keyPath: cs };
+    });
+    const picked = pickCoveringIndex({
+      table: this.table,
+      indexes: registered,
+      criteriaColumns: Object.keys(criteria),
+      orderByColumns: (options.orderBy ?? []).map((o) => ({
+        column: String(o.column),
+        direction: o.direction
+      })),
+      selectColumns: options.select.map(String),
+      primaryKeyColumns: this.primaryKeyColumns().map(String)
+    });
+    const db = await this.getDb();
+    return new Promise((resolve, reject) => {
+      const tx = db.transaction(this.table, "readonly");
+      const store = tx.objectStore(this.table);
+      const idx = store.index(picked.name);
+      const prefix = [];
+      for (const col of picked.keyPath) {
+        const c = criteria[col];
+        if (c === undefined && !(col in criteria))
+          break;
+        if (isSearchCondition(c)) {
+          if (c.operator !== "=")
+            break;
+          prefix.push(c.value);
+        } else {
+          prefix.push(c);
+        }
+      }
+      const range = prefix.length === 0 ? undefined : prefix.length === picked.keyPath.length ? IDBKeyRange.only(prefix.length === 1 ? prefix[0] : prefix) : IDBKeyRange.bound(prefix, [...prefix, []]);
+      const direction = picked.reverseDirection ? "prev" : "next";
+      const request = idx.openKeyCursor(range, direction);
+      const out = [];
+      let toSkip = options.offset ?? 0;
+      const keyPathPositions = new Map;
+      picked.keyPath.forEach((col, i) => keyPathPositions.set(col, i));
+      const pkCols = this.primaryKeyColumns().map(String);
+      const pkPositions = new Map;
+      pkCols.forEach((col, i) => pkPositions.set(col, i));
+      request.onsuccess = () => {
+        const cursor = request.result;
+        if (!cursor) {
+          resolve(out);
+          return;
+        }
+        const key = cursor.key;
+        const row = {};
+        for (const col of options.select) {
+          const colStr = String(col);
+          const pos = keyPathPositions.get(colStr);
+          if (pos !== undefined) {
+            row[colStr] = Array.isArray(key) ? key[pos] : key;
+          } else {
+            if (pkCols.length === 1 && colStr === pkCols[0]) {
+              row[colStr] = cursor.primaryKey;
+            } else {
+              const pkPos = pkPositions.get(colStr);
+              if (pkPos !== undefined) {
+                row[colStr] = Array.isArray(cursor.primaryKey) ? cursor.primaryKey[pkPos] : cursor.primaryKey;
+              }
+            }
+          }
+        }
+        let matches = true;
+        for (const [col, crit] of Object.entries(criteria)) {
+          const pos = keyPathPositions.get(col);
+          if (pos === undefined)
+            continue;
+          if (pos < prefix.length)
+            continue;
+          const valFromKey = Array.isArray(key) ? key[pos] : key;
+          const op = isSearchCondition(crit) ? crit.operator : "=";
+          const val = isSearchCondition(crit) ? crit.value : crit;
+          if (!compareWithOperator(valFromKey, op, val)) {
+            matches = false;
+            break;
+          }
+        }
+        if (matches) {
+          if (toSkip > 0) {
+            toSkip -= 1;
+          } else {
+            out.push(row);
+            if (options.limit !== undefined && out.length >= options.limit) {
+              resolve(out);
+              return;
+            }
+          }
+        }
+        cursor.continue();
+      };
+      request.onerror = () => reject(request.error);
+    });
+  }
   getHybridManager() {
     if (!this.hybridManager) {
       const channelName = `indexeddb-tabular-${this.table}`;
@@ -3233,6 +3526,22 @@ class IndexedDbTabularStorage extends BaseTabularStorage {
     this.db?.close();
   }
 }
+function compareWithOperator(a, op, b) {
+  const av = a;
+  const bv = b;
+  switch (op) {
+    case "=":
+      return av === bv;
+    case "<":
+      return av !== null && av !== undefined && av < bv;
+    case "<=":
+      return av !== null && av !== undefined && av <= bv;
+    case ">":
+      return av !== null && av !== undefined && av > bv;
+    case ">=":
+      return av !== null && av !== undefined && av >= bv;
+  }
+}
 // src/tabular/SharedInMemoryTabularStorage.ts
 import { createServiceToken as createServiceToken13 } from "@workglow/util";
 var SHARED_IN_MEMORY_TABULAR_REPOSITORY = createServiceToken13("storage.tabularRepository.sharedInMemory");
@@ -3406,6 +3715,9 @@ class SharedInMemoryTabularStorage extends BaseTabularStorage {
   async query(criteria, options) {
     return await this.inMemoryRepo.query(criteria, options);
   }
+  async queryIndex(criteria, options) {
+    return await this.inMemoryRepo.queryIndex(criteria, options);
+  }
   async deleteSearch(criteria) {
     await this.inMemoryRepo.deleteSearch(criteria);
     this.broadcast({
@@ -4093,6 +4405,56 @@ class SupabaseTabularStorage extends BaseSqlTabularStorage {
     this.events.emit("query", criteria, undefined);
     return;
   }
+  async queryIndex(criteria, options) {
+    this.validateSelect(options);
+    this.validateQueryParams(criteria, options);
+    const registered = this.indexes.map((cols, i) => {
+      const cs = Array.isArray(cols) ? cols : [cols];
+      return { name: `idx_${i}`, keyPath: cs };
+    });
+    pickCoveringIndex({
+      table: this.table,
+      indexes: registered,
+      criteriaColumns: Object.keys(criteria),
+      orderByColumns: (options.orderBy ?? []).map((o) => ({
+        column: String(o.column),
+        direction: o.direction
+      })),
+      selectColumns: options.select.map(String),
+      primaryKeyColumns: this.primaryKeyNames.map(String)
+    });
+    const colList = options.select.map(String).join(",");
+    let q = this.applyCriteriaToFilter(this.client.from(this.table).select(colList), criteria);
+    if (options.orderBy) {
+      for (const { column, direction } of options.orderBy) {
+        q = q.order(String(column), { ascending: direction === "ASC" });
+      }
+    }
+    if (options.offset !== undefined && options.limit === undefined) {
+      throw new StorageValidationError("queryIndex with offset requires limit (no implicit cap)");
+    }
+    if (options.offset !== undefined || options.limit !== undefined) {
+      const start = options.offset ?? 0;
+      if (options.limit !== undefined) {
+        q = q.range(start, start + options.limit - 1);
+      }
+    }
+    const { data, error } = await q;
+    if (error)
+      throw error;
+    if (!data)
+      return [];
+    const rows = data;
+    const sel = new Set(options.select.map(String));
+    for (const row of rows) {
+      for (const key of Object.keys(row)) {
+        if (sel.has(key)) {
+          row[key] = this.sqlToJsValue(key, row[key]);
+        }
+      }
+    }
+    return rows;
+  }
   convertRealtimeRow(row) {
     const entity = { ...row };
     const record = entity;
@@ -4158,11 +4520,12 @@ class SupabaseKvStorage extends KvViaTabularStorage {
   }
 }
 // src/queue/IndexedDbQueueStorage.ts
-import { createServiceToken as createServiceToken17, deepEqual as deepEqual3, makeFingerprint as makeFingerprint6, uuid4 as uuid44 } from "@workglow/util";
+import { createServiceToken as createServiceToken17, deepEqual as deepEqual3, makeFingerprint as makeFingerprint6, uuid4 as uuid45 } from "@workglow/util";
 var INDEXED_DB_QUEUE_STORAGE = createServiceToken17("jobqueue.storage.indexedDb");
 
 class IndexedDbQueueStorage {
   queueName;
+  scope = "process";
   db;
   tableName;
   migrationOptions;
@@ -4239,8 +4602,8 @@ class IndexedDbQueueStorage {
     const db = await this.getDb();
     const now = new Date().toISOString();
     const jobWithPrefixes = job;
-    jobWithPrefixes.id = jobWithPrefixes.id ?? uuid44();
-    jobWithPrefixes.job_run_id = jobWithPrefixes.job_run_id ?? uuid44();
+    jobWithPrefixes.id = jobWithPrefixes.id ?? uuid45();
+    jobWithPrefixes.job_run_id = jobWithPrefixes.job_run_id ?? uuid45();
     jobWithPrefixes.queue = this.queueName;
     jobWithPrefixes.fingerprint = await makeFingerprint6(jobWithPrefixes.input);
     jobWithPrefixes.status = JobStatus.PENDING;
@@ -4738,11 +5101,12 @@ class IndexedDbQueueStorage {
   }
 }
 // src/queue/SupabaseQueueStorage.ts
-import { createServiceToken as createServiceToken18, deepEqual as deepEqual4, makeFingerprint as makeFingerprint7, uuid4 as uuid45 } from "@workglow/util";
+import { createServiceToken as createServiceToken18, deepEqual as deepEqual4, makeFingerprint as makeFingerprint7, uuid4 as uuid46 } from "@workglow/util";
 var SUPABASE_QUEUE_STORAGE = createServiceToken18("jobqueue.storage.supabase");
 
 class SupabaseQueueStorage {
   queueName;
+  scope = "cluster";
   client;
   prefixes;
   prefixValues;
@@ -4867,7 +5231,7 @@ class SupabaseQueueStorage {
   async add(job) {
     const now = new Date().toISOString();
     job.queue = this.queueName;
-    job.job_run_id = job.job_run_id ?? uuid45();
+    job.job_run_id = job.job_run_id ?? uuid46();
     job.fingerprint = await makeFingerprint7(job.input);
     job.status = JobStatus.PENDING;
     job.progress = 0;
@@ -5269,6 +5633,7 @@ import { createServiceToken as createServiceToken19 } from "@workglow/util";
 var INDEXED_DB_RATE_LIMITER_STORAGE = createServiceToken19("ratelimiter.storage.indexedDb");
 
 class IndexedDbRateLimiterStorage {
+  scope = "process";
   executionDb;
   nextAvailableDb;
   executionTableName;
@@ -5337,6 +5702,72 @@ class IndexedDbRateLimiterStorage {
     ];
     this.nextAvailableDb = await ensureIndexedDbTable(this.nextAvailableTableName, buildKeyPath(["queue_name"]).join("_"), nextAvailableIndexes, this.migrationOptions);
   }
+  async tryReserveExecution(queueName, maxExecutions, windowMs) {
+    const nextIso = await this.getNextAvailableTime(queueName);
+    if (nextIso && new Date(nextIso).getTime() > Date.now()) {
+      return null;
+    }
+    const execDb = await this.getExecutionDb();
+    const prefixKeyValues = this.getPrefixKeyValues();
+    const windowStartIso = new Date(Date.now() - windowMs).toISOString();
+    const execTx = execDb.transaction(this.executionTableName, "readwrite");
+    const execStore = execTx.objectStore(this.executionTableName);
+    const insertedId = crypto.randomUUID();
+    return new Promise((resolve, reject) => {
+      let liveCount = 0;
+      let didInsert = false;
+      const liveRange = IDBKeyRange.bound([...prefixKeyValues, queueName, windowStartIso], [...prefixKeyValues, queueName, "￿"], true, false);
+      const cursorReq = execStore.index("queue_executed_at").openCursor(liveRange);
+      cursorReq.onsuccess = (event) => {
+        const cursor = event.target.result;
+        if (cursor) {
+          const record2 = cursor.value;
+          if (this.matchesPrefixes(record2)) {
+            liveCount++;
+          }
+          cursor.continue();
+          return;
+        }
+        if (liveCount >= maxExecutions) {
+          execTx.abort();
+          return;
+        }
+        const record = {
+          id: insertedId,
+          queue_name: queueName,
+          executed_at: new Date().toISOString()
+        };
+        for (const [k, v] of Object.entries(this.prefixValues)) {
+          record[k] = v;
+        }
+        const addReq = execStore.add(record);
+        didInsert = true;
+        addReq.onerror = () => {
+          try {
+            execTx.abort();
+          } catch {}
+          reject(addReq.error);
+        };
+      };
+      cursorReq.onerror = () => reject(cursorReq.error);
+      execTx.oncomplete = () => resolve(didInsert ? insertedId : null);
+      execTx.onerror = () => reject(execTx.error);
+      execTx.onabort = () => resolve(null);
+    });
+  }
+  async releaseExecution(queueName, token) {
+    if (token === null || token === undefined)
+      return;
+    const db = await this.getExecutionDb();
+    const tx = db.transaction(this.executionTableName, "readwrite");
+    const store = tx.objectStore(this.executionTableName);
+    return new Promise((resolve, reject) => {
+      const req = store.delete(token);
+      req.onerror = () => reject(req.error);
+      tx.oncomplete = () => resolve();
+      tx.onerror = () => reject(tx.error);
+    });
+  }
   async recordExecution(queueName) {
     const db = await this.getExecutionDb();
     const tx = db.transaction(this.executionTableName, "readwrite");
@@ -5490,6 +5921,7 @@ import { createServiceToken as createServiceToken20 } from "@workglow/util";
 var SUPABASE_RATE_LIMITER_STORAGE = createServiceToken20("ratelimiter.storage.supabase");
 
 class SupabaseRateLimiterStorage {
+  scope = "cluster";
   client;
   prefixes;
   prefixValues;
@@ -5572,6 +6004,75 @@ class SupabaseRateLimiterStorage {
     if (nextTableError && nextTableError.code !== "42P07") {
       throw nextTableError;
     }
+    const fnName = this.atomicReserveFunctionName();
+    const prefixSig = this.prefixes.map((p) => `${p.name} ${this.getPrefixColumnType(p.type)}`).join(", ");
+    const prefixSigPrefix = prefixSig ? prefixSig + ", " : "";
+    const prefixWhere = this.prefixes.length > 0 ? " AND " + this.prefixes.map((p) => `${p.name} = _${p.name}`).join(" AND ") : "";
+    const prefixInsertCols = this.prefixes.length > 0 ? this.prefixes.map((p) => p.name).join(", ") + ", " : "";
+    const prefixInsertVals = this.prefixes.length > 0 ? this.prefixes.map((p) => `_${p.name}`).join(", ") + ", " : "";
+    const lockKeyParts = [`'${this.executionTableName}'`, ...this.prefixes.map((p) => `_${p.name}::text`), `_queue_name::text`];
+    const lockKeyExpr = `hashtextextended(${lockKeyParts.join(" || '|' || ")}, 0)`;
+    const createFnSql = `
+      CREATE OR REPLACE FUNCTION ${fnName}(
+        ${prefixSigPrefix}_queue_name TEXT, _window_start TIMESTAMPTZ, _max_exec INT
+      ) RETURNS BIGINT AS $fn$
+      DECLARE
+        _count INT;
+        _next TIMESTAMPTZ;
+        _new_id BIGINT;
+      BEGIN
+        PERFORM pg_advisory_xact_lock(${lockKeyExpr});
+        SELECT COUNT(*) INTO _count FROM ${this.executionTableName}
+          WHERE queue_name = _queue_name AND executed_at > _window_start${prefixWhere};
+        IF _count >= _max_exec THEN RETURN NULL; END IF;
+        SELECT next_available_at INTO _next FROM ${this.nextAvailableTableName}
+          WHERE queue_name = _queue_name${prefixWhere};
+        IF _next IS NOT NULL AND _next > NOW() THEN RETURN NULL; END IF;
+        INSERT INTO ${this.executionTableName} (${prefixInsertCols}queue_name)
+          VALUES (${prefixInsertVals}_queue_name)
+          RETURNING id INTO _new_id;
+        RETURN _new_id;
+      END;
+      $fn$ LANGUAGE plpgsql;
+    `;
+    const { error: fnError } = await this.client.rpc("exec_sql", { query: createFnSql });
+    if (fnError) {
+      throw fnError;
+    }
+  }
+  atomicReserveFunctionName() {
+    return `${this.executionTableName}_try_reserve`.slice(0, 63);
+  }
+  async tryReserveExecution(queueName, maxExecutions, windowMs) {
+    const args = {
+      _queue_name: queueName,
+      _window_start: new Date(Date.now() - windowMs).toISOString(),
+      _max_exec: maxExecutions
+    };
+    for (const p of this.prefixes) {
+      args[`_${p.name}`] = this.prefixValues[p.name];
+    }
+    const { data, error } = await this.client.rpc(this.atomicReserveFunctionName(), args);
+    if (error)
+      throw error;
+    if (data === null || data === undefined)
+      return null;
+    if (Array.isArray(data)) {
+      if (data.length === 0)
+        return null;
+      const first = Object.values(data[0])[0];
+      return first ?? null;
+    }
+    return data;
+  }
+  async releaseExecution(queueName, token) {
+    if (token === null || token === undefined)
+      return;
+    let del = this.client.from(this.executionTableName).delete().eq("id", token).eq("queue_name", queueName);
+    del = this.applyPrefixFilters(del);
+    const { error: delError } = await del;
+    if (delError)
+      throw delError;
   }
   async recordExecution(queueName) {
     const prefixInsertValues = this.getPrefixInsertValues();
@@ -5739,6 +6240,7 @@ class IndexedDbVectorStorage extends IndexedDbTabularStorage {
 export {
   traced,
   registerTabularRepository,
+  pickCoveringIndex,
   isSearchCondition,
   getVectorProperty,
   getTabularRepository,
@@ -5801,9 +6303,10 @@ export {
   EncryptedKvCredentialStore,
   DefaultKeyValueSchema,
   DefaultKeyValueKey,
+  CoveringIndexMissingError,
   CachedTabularStorage,
   CACHED_TABULAR_REPOSITORY,
   BaseTabularStorage
 };
 
-//# debugId=4C0D1FA7EA7C0F0564756E2164756E21
+//# debugId=E2D716C3197417F164756E2164756E21