@mastra/pg 1.31.1-alpha.1 → 1.31.2-alpha.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.31.1-alpha.1"
+  version: "1.31.2-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.31.1-alpha.1",
+  "version": "1.31.2-alpha.0",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -14563,6 +14563,7 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 };
 //#endregion
 //#region src/storage/domains/notifications/index.ts
+const COMPOSITE_PRIMARY_KEY$1 = _mastra_core_storage.TABLE_CONFIGS[_mastra_core_storage.TABLE_NOTIFICATIONS].compositePrimaryKey;
 const statusTimestamp = (status, now) => {
 	if (status === "delivered") return { deliveredAt: now };
 	if (status === "seen") return { seenAt: now };
@@ -14674,11 +14675,58 @@ var NotificationsPG = class NotificationsPG extends _mastra_core_storage.Notific
 	async init() {
 		await this.#db.createTable({
 			tableName: _mastra_core_storage.TABLE_NOTIFICATIONS,
-			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_NOTIFICATIONS]
+			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_NOTIFICATIONS],
+			compositePrimaryKey: COMPOSITE_PRIMARY_KEY$1
 		});
+		await this.ensurePrimaryKey();
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
+	async #hasPrimaryKey() {
+		return !!(await this.#db.client.oneOrNone(`SELECT EXISTS (
+        SELECT 1 FROM pg_index i
+        JOIN pg_class c ON c.oid = i.indrelid
+        JOIN pg_namespace n ON n.oid = c.relnamespace
+        WHERE c.relname = $1 AND n.nspname = $2 AND i.indisprimary
+      ) AS exists`, [_mastra_core_storage.TABLE_NOTIFICATIONS, this.#schema]))?.exists;
+	}
+	/**
+	* Tables created before the composite primary key was declared have no key at
+	* all, which allows duplicate ids and leaves the table without a replica
+	* identity (UPDATE/DELETE fail under logical replication). `createTable` is a
+	* no-op for existing tables, so add the key here. If duplicate rows already
+	* exist, or the role does not own the table, the key cannot be added; warn with
+	* remediation instead of blocking init.
+	*/
+	async ensurePrimaryKey() {
+		if (await this.#hasPrimaryKey()) return;
+		const tableName = getTableName$5({
+			indexName: _mastra_core_storage.TABLE_NOTIFICATIONS,
+			schemaName: getSchemaName$5(this.#schema)
+		});
+		const keyColumns = COMPOSITE_PRIMARY_KEY$1.map((column) => `"${(0, _mastra_core_utils.parseSqlIdentifier)(column, "column name")}"`).join(", ");
+		const remediation = `ALTER TABLE ${tableName} ADD PRIMARY KEY (${keyColumns});`;
+		if (await this.#db.client.oneOrNone(`SELECT 1 FROM ${tableName} GROUP BY ${keyColumns} HAVING count(*) > 1 LIMIT 1`)) {
+			this.logger?.warn?.(`${tableName} contains duplicate (${keyColumns}) rows, so its PRIMARY KEY could not be added. Remove the duplicates, then run: ${remediation}`);
+			return;
+		}
+		try {
+			await this.#db.client.none(`ALTER TABLE ${tableName} ADD PRIMARY KEY (${keyColumns})`);
+			this.logger?.info?.(`Added PRIMARY KEY (${keyColumns}) to ${tableName}`);
+		} catch (error) {
+			if (await this.#hasPrimaryKey()) return;
+			if (error?.code === "42501") {
+				this.logger?.warn?.(`Insufficient privilege to add PRIMARY KEY (${keyColumns}) to ${tableName}. Ask the table owner to run: ${remediation}`);
+				return;
+			}
+			throw new _mastra_core_error.MastraError({
+				id: (0, _mastra_core_storage.createStorageErrorId)("PG", "ADD_NOTIFICATIONS_PRIMARY_KEY", "FAILED"),
+				domain: _mastra_core_error.ErrorDomain.STORAGE,
+				category: _mastra_core_error.ErrorCategory.THIRD_PARTY,
+				details: { tableName }
+			}, error);
+		}
+	}
 	/**
 	* Lazily ensures a btree index exists on each configured policy's retention
 	* anchor column so age-based `prune()` deletes stay fast on large tables.
@@ -14762,6 +14810,7 @@ var NotificationsPG = class NotificationsPG extends _mastra_core_storage.Notific
 			tableName: _mastra_core_storage.TABLE_NOTIFICATIONS,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_NOTIFICATIONS],
 			schemaName,
+			compositePrimaryKey: COMPOSITE_PRIMARY_KEY$1,
 			includeAllConstraints: true
 		}));
 		for (const idx of NotificationsPG.getDefaultIndexDefs(schemaPrefix)) statements.push(generateIndexSQL(idx, schemaName));