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

package/dist/index.js

@@ -14539,6 +14539,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 };
 //#endregion
 //#region src/storage/domains/notifications/index.ts
+const COMPOSITE_PRIMARY_KEY$1 = TABLE_CONFIGS[TABLE_NOTIFICATIONS].compositePrimaryKey;
 const statusTimestamp = (status, now) => {
 	if (status === "delivered") return { deliveredAt: now };
 	if (status === "seen") return { seenAt: now };
@@ -14650,11 +14651,58 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 	async init() {
 		await this.#db.createTable({
 			tableName: TABLE_NOTIFICATIONS,
-			schema: TABLE_SCHEMAS[TABLE_NOTIFICATIONS]
+			schema: TABLE_SCHEMAS[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`, [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: TABLE_NOTIFICATIONS,
+			schemaName: getSchemaName$5(this.#schema)
+		});
+		const keyColumns = COMPOSITE_PRIMARY_KEY$1.map((column) => `"${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 MastraError({
+				id: createStorageErrorId("PG", "ADD_NOTIFICATIONS_PRIMARY_KEY", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: 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.
@@ -14738,6 +14786,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 			tableName: TABLE_NOTIFICATIONS,
 			schema: TABLE_SCHEMAS[TABLE_NOTIFICATIONS],
 			schemaName,
+			compositePrimaryKey: COMPOSITE_PRIMARY_KEY$1,
 			includeAllConstraints: true
 		}));
 		for (const idx of NotificationsPG.getDefaultIndexDefs(schemaPrefix)) statements.push(generateIndexSQL(idx, schemaName));