@mastra/pg 1.17.0 → 1.17.1-alpha.0

package/dist/index.js

@@ -9,7 +9,7 @@ import xxhash from 'xxhash-wasm';
 import { parse } from 'pg-connection-string';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import { MastraBase } from '@mastra/core/base';
-import { randomUUID, createHash } from 'crypto';
+import { randomUUID } from 'crypto';
 import { MessageList } from '@mastra/core/agent';
 import { coreFeatures } from '@mastra/core/features';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
@@ -20226,10 +20226,6 @@ function serializeDefault(value) {
   if (typeof value === "string") return `'${value.replace(/'/g, "''")}'`;
   return String(value);
 }
-function hashAdvisoryLockKey(key) {
-  const digest = createHash("sha256").update(key).digest();
-  return [digest.readInt32BE(0), digest.readInt32BE(4)];
-}
 var PgFactoryStorageOps = class {
   #queryable;
   #transactionClient;
@@ -20524,21 +20520,36 @@ var PgFactoryStorage = class extends FactoryStorage {
   async initStorage() {
     await this.#pool.query("SELECT 1");
   }
-  async withTransaction(fn) {
-    const client = await this.#pool.connect();
-    try {
-      await client.query("BEGIN");
+  async withTransaction(fn, options) {
+    const maxAttempts = options?.isolationLevel === "serializable" ? 3 : 1;
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+      const client = await this.#pool.connect();
+      let transactionOpen = false;
+      let releaseError;
       try {
+        await client.query(options?.isolationLevel === "serializable" ? "BEGIN ISOLATION LEVEL SERIALIZABLE" : "BEGIN");
+        transactionOpen = true;
         const result = await fn(new PgFactoryStorageOps(client, this.#schemas, client));
         await client.query("COMMIT");
+        transactionOpen = false;
         return result;
       } catch (error) {
-        await client.query("ROLLBACK");
-        throw error;
+        if (transactionOpen) {
+          try {
+            await client.query("ROLLBACK");
+            transactionOpen = false;
+          } catch (rollbackError) {
+            releaseError = rollbackError instanceof Error ? rollbackError : new Error(String(rollbackError));
+            throw new AggregateError([error, rollbackError], "Factory transaction and rollback both failed");
+          }
+        }
+        const serializationFailure = typeof error === "object" && error !== null && error.code === "40001";
+        if (!serializationFailure || attempt === maxAttempts) throw error;
+      } finally {
+        client.release(releaseError);
       }
-    } finally {
-      client.release();
     }
+    throw new Error("PgFactoryStorage: serializable transaction retry limit exceeded");
   }
   async ensureCollections(schemas) {
     for (const schema of schemas) {
@@ -20556,30 +20567,6 @@ var PgFactoryStorage = class extends FactoryStorage {
   authDatabase() {
     return { dialect: "postgres", pool: this.#pool };
   }
-  /**
-   * Run `fn` while holding a Postgres transaction-scoped advisory lock for
-   * `key`, serializing callers across replicas. The lock releases when the
-   * transaction ends (commit, rollback, or connection loss), so a crashed
-   * replica can never hold it forever.
-   */
-  async withDistributedLock(key, fn) {
-    const [k1, k2] = hashAdvisoryLockKey(key);
-    const client = await this.#pool.connect();
-    try {
-      await client.query("BEGIN");
-      await client.query("SELECT pg_advisory_xact_lock($1, $2)", [k1, k2]);
-      try {
-        const result = await fn();
-        await client.query("COMMIT");
-        return result;
-      } catch (error) {
-        await client.query("ROLLBACK");
-        throw error;
-      }
-    } finally {
-      client.release();
-    }
-  }
   #columnDdl(name, spec) {
     assertIdentifier("column", name);
     let ddl = `"${name}" ${COLUMN_DDL[spec.type]}`;
@@ -21015,6 +21002,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, FavoritesPG, MCPClientsPG, MCPServersPG, MemoryPG, NotificationsPG, ObservabilityPG, ObservabilityStoragePostgresVNext, PGVECTOR_PROMPT, PgFactoryStorage, PgVector, PoolAdapter, PostgresStore, PostgresStoreVNext, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, ToolProviderConnectionsPG, WorkflowsPG, WorkspacesPG, exportSchemas, hashAdvisoryLockKey };
+export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, FavoritesPG, MCPClientsPG, MCPServersPG, MemoryPG, NotificationsPG, ObservabilityPG, ObservabilityStoragePostgresVNext, PGVECTOR_PROMPT, PgFactoryStorage, PgVector, PoolAdapter, PostgresStore, PostgresStoreVNext, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, ToolProviderConnectionsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map