@fedify/postgres 2.0.0-dev.323 → 2.0.0-dev.335

package/dist/mq.cjs

@@ -35,6 +35,7 @@ var PostgresMessageQueue = class {
 	#channelName;
 	#pollIntervalMs;
 	#initialized;
+	#initPromise;
 	#driverSerializesJson = false;
 	constructor(sql, options = {}) {
 		this.#sql = sql;
@@ -114,7 +115,7 @@ var PostgresMessageQueue = class {
 		const poll = async () => {
 			while (!signal?.aborted) {
 				let processed = false;
-				const query = this.#sql`
+				for (const row of await this.#sql`
           WITH candidate AS (
             SELECT id, ordering_key
             FROM ${this.#sql(this.#tableName)}
@@ -127,15 +128,11 @@ var PostgresMessageQueue = class {
           DELETE FROM ${this.#sql(this.#tableName)}
           WHERE id IN (SELECT id FROM candidate)
           RETURNING message, ordering_key;
-        `.execute();
-				const cancel = query.cancel.bind(query);
-				signal?.addEventListener("abort", cancel);
-				for (const row of await query) {
+        `) {
 					if (signal?.aborted) return;
 					await handler(row.message);
 					processed = true;
 				}
-				signal?.removeEventListener("abort", cancel);
 				if (processed) continue;
 				const attemptedOrderingKeys = /* @__PURE__ */ new Set();
 				while (!signal?.aborted) {
@@ -153,46 +150,57 @@ var PostgresMessageQueue = class {
 					const candidateId = candidate.id;
 					const orderingKey = candidate.ordering_key;
 					attemptedOrderingKeys.add(orderingKey);
-					const lockResult = await this.#sql`
-            SELECT pg_try_advisory_lock(
-              hashtext(${this.#tableName}),
-              hashtext(${orderingKey})
-            ) AS acquired
-          `;
-					if (lockResult[0].acquired) {
-						try {
-							const deleteResult = await this.#sql`
-                DELETE FROM ${this.#sql(this.#tableName)}
-                WHERE id = ${candidateId}
-                RETURNING message, ordering_key
-              `;
-							for (const row of deleteResult) {
-								if (signal?.aborted) return;
-								await handler(row.message);
-								processed = true;
+					const reserved = await this.#sql.reserve();
+					try {
+						const lockResult = await reserved`
+              SELECT pg_try_advisory_lock(
+                hashtext(${this.#tableName}),
+                hashtext(${orderingKey})
+              ) AS acquired
+            `;
+						if (lockResult[0].acquired) {
+							try {
+								const deleteResult = await reserved`
+                  DELETE FROM ${reserved(this.#tableName)}
+                  WHERE id = ${candidateId}
+                  RETURNING message, ordering_key
+                `;
+								for (const row of deleteResult) {
+									if (signal?.aborted) return;
+									await handler(row.message);
+									processed = true;
+								}
+							} finally {
+								await reserved`
+                  SELECT pg_advisory_unlock(
+                    hashtext(${this.#tableName}),
+                    hashtext(${orderingKey})
+                  )
+                `;
 							}
-						} finally {
-							await this.#sql`
-                SELECT pg_advisory_unlock(
-                  hashtext(${this.#tableName}),
-                  hashtext(${orderingKey})
-                )
-              `;
+							if (processed) break;
 						}
-						if (processed) break;
+					} finally {
+						reserved.release();
 					}
 				}
 				if (processed) continue;
 				break;
 			}
 		};
+		let pollLock = Promise.resolve();
+		const serializedPoll = () => {
+			const next = pollLock.then(poll);
+			pollLock = next.catch(() => {});
+			return next;
+		};
 		const timeouts = /* @__PURE__ */ new Set();
 		const listen = await this.#sql.listen(this.#channelName, async (delay) => {
 			const duration = Temporal.Duration.from(delay);
 			const durationMs = duration.total("millisecond");
-			if (durationMs < 1) await poll();
-			else timeouts.add(setTimeout(poll, durationMs));
-		}, poll);
+			if (durationMs < 1) await serializedPoll();
+			else timeouts.add(setTimeout(serializedPoll, durationMs));
+		}, serializedPoll);
 		signal?.addEventListener("abort", () => {
 			listen.unlisten();
 			for (const timeout of timeouts) clearTimeout(timeout);
@@ -208,7 +216,7 @@ var PostgresMessageQueue = class {
 				timeouts.add(timeout);
 			});
 			if (timeout != null) timeouts.delete(timeout);
-			await poll();
+			await serializedPoll();
 		}
 		await new Promise((resolve) => {
 			signal?.addEventListener("abort", () => resolve());
@@ -218,8 +226,11 @@ var PostgresMessageQueue = class {
 	/**
 	* Initializes the message queue table if it does not already exist.
 	*/
-	async initialize() {
-		if (this.#initialized) return;
+	initialize() {
+		if (this.#initialized) return Promise.resolve();
+		return this.#initPromise ??= this.#doInitialize();
+	}
+	async #doInitialize() {
 		logger.debug("Initializing the message queue table {tableName}...", { tableName: this.#tableName });
 		try {
 			await this.#sql`
@@ -236,7 +247,7 @@ var PostgresMessageQueue = class {
       ADD COLUMN IF NOT EXISTS ordering_key text;
     `;
 		} catch (error) {
-			if (!(error instanceof postgres.default.PostgresError && error.constraint_name === "pg_type_typname_nsp_index")) {
+			if (!(error instanceof postgres.default.PostgresError && (error.constraint_name === "pg_type_typname_nsp_index" || error.code === "42P07"))) {
 				logger.error("Failed to initialize the message queue table: {error}", { error });
 				throw error;
 			}

package/dist/mq.js

@@ -34,6 +34,7 @@ var PostgresMessageQueue = class {
 	#channelName;
 	#pollIntervalMs;
 	#initialized;
+	#initPromise;
 	#driverSerializesJson = false;
 	constructor(sql, options = {}) {
 		this.#sql = sql;
@@ -113,7 +114,7 @@ var PostgresMessageQueue = class {
 		const poll = async () => {
 			while (!signal?.aborted) {
 				let processed = false;
-				const query = this.#sql`
+				for (const row of await this.#sql`
           WITH candidate AS (
             SELECT id, ordering_key
             FROM ${this.#sql(this.#tableName)}
@@ -126,15 +127,11 @@ var PostgresMessageQueue = class {
           DELETE FROM ${this.#sql(this.#tableName)}
           WHERE id IN (SELECT id FROM candidate)
           RETURNING message, ordering_key;
-        `.execute();
-				const cancel = query.cancel.bind(query);
-				signal?.addEventListener("abort", cancel);
-				for (const row of await query) {
+        `) {
 					if (signal?.aborted) return;
 					await handler(row.message);
 					processed = true;
 				}
-				signal?.removeEventListener("abort", cancel);
 				if (processed) continue;
 				const attemptedOrderingKeys = /* @__PURE__ */ new Set();
 				while (!signal?.aborted) {
@@ -152,46 +149,57 @@ var PostgresMessageQueue = class {
 					const candidateId = candidate.id;
 					const orderingKey = candidate.ordering_key;
 					attemptedOrderingKeys.add(orderingKey);
-					const lockResult = await this.#sql`
-            SELECT pg_try_advisory_lock(
-              hashtext(${this.#tableName}),
-              hashtext(${orderingKey})
-            ) AS acquired
-          `;
-					if (lockResult[0].acquired) {
-						try {
-							const deleteResult = await this.#sql`
-                DELETE FROM ${this.#sql(this.#tableName)}
-                WHERE id = ${candidateId}
-                RETURNING message, ordering_key
-              `;
-							for (const row of deleteResult) {
-								if (signal?.aborted) return;
-								await handler(row.message);
-								processed = true;
+					const reserved = await this.#sql.reserve();
+					try {
+						const lockResult = await reserved`
+              SELECT pg_try_advisory_lock(
+                hashtext(${this.#tableName}),
+                hashtext(${orderingKey})
+              ) AS acquired
+            `;
+						if (lockResult[0].acquired) {
+							try {
+								const deleteResult = await reserved`
+                  DELETE FROM ${reserved(this.#tableName)}
+                  WHERE id = ${candidateId}
+                  RETURNING message, ordering_key
+                `;
+								for (const row of deleteResult) {
+									if (signal?.aborted) return;
+									await handler(row.message);
+									processed = true;
+								}
+							} finally {
+								await reserved`
+                  SELECT pg_advisory_unlock(
+                    hashtext(${this.#tableName}),
+                    hashtext(${orderingKey})
+                  )
+                `;
 							}
-						} finally {
-							await this.#sql`
-                SELECT pg_advisory_unlock(
-                  hashtext(${this.#tableName}),
-                  hashtext(${orderingKey})
-                )
-              `;
+							if (processed) break;
 						}
-						if (processed) break;
+					} finally {
+						reserved.release();
 					}
 				}
 				if (processed) continue;
 				break;
 			}
 		};
+		let pollLock = Promise.resolve();
+		const serializedPoll = () => {
+			const next = pollLock.then(poll);
+			pollLock = next.catch(() => {});
+			return next;
+		};
 		const timeouts = /* @__PURE__ */ new Set();
 		const listen = await this.#sql.listen(this.#channelName, async (delay) => {
 			const duration = Temporal.Duration.from(delay);
 			const durationMs = duration.total("millisecond");
-			if (durationMs < 1) await poll();
-			else timeouts.add(setTimeout(poll, durationMs));
-		}, poll);
+			if (durationMs < 1) await serializedPoll();
+			else timeouts.add(setTimeout(serializedPoll, durationMs));
+		}, serializedPoll);
 		signal?.addEventListener("abort", () => {
 			listen.unlisten();
 			for (const timeout of timeouts) clearTimeout(timeout);
@@ -207,7 +215,7 @@ var PostgresMessageQueue = class {
 				timeouts.add(timeout);
 			});
 			if (timeout != null) timeouts.delete(timeout);
-			await poll();
+			await serializedPoll();
 		}
 		await new Promise((resolve) => {
 			signal?.addEventListener("abort", () => resolve());
@@ -217,8 +225,11 @@ var PostgresMessageQueue = class {
 	/**
 	* Initializes the message queue table if it does not already exist.
 	*/
-	async initialize() {
-		if (this.#initialized) return;
+	initialize() {
+		if (this.#initialized) return Promise.resolve();
+		return this.#initPromise ??= this.#doInitialize();
+	}
+	async #doInitialize() {
 		logger.debug("Initializing the message queue table {tableName}...", { tableName: this.#tableName });
 		try {
 			await this.#sql`
@@ -235,7 +246,7 @@ var PostgresMessageQueue = class {
       ADD COLUMN IF NOT EXISTS ordering_key text;
     `;
 		} catch (error) {
-			if (!(error instanceof postgres.PostgresError && error.constraint_name === "pg_type_typname_nsp_index")) {
+			if (!(error instanceof postgres.PostgresError && (error.constraint_name === "pg_type_typname_nsp_index" || error.code === "42P07"))) {
 				logger.error("Failed to initialize the message queue table: {error}", { error });
 				throw error;
 			}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fedify/postgres",
-  "version": "2.0.0-dev.323+1d796545",
+  "version": "2.0.0-dev.335+6fe86704",
   "description": "PostgreSQL drivers for Fedify",
   "keywords": [
     "fedify",
@@ -74,14 +74,14 @@
   },
   "peerDependencies": {
     "postgres": "^3.4.7",
-    "@fedify/fedify": "^2.0.0-dev.323+1d796545"
+    "@fedify/fedify": "^2.0.0-dev.335+6fe86704"
   },
   "devDependencies": {
     "@std/async": "npm:@jsr/std__async@^1.0.13",
     "tsdown": "^0.12.9",
     "typescript": "^5.9.3",
-    "@fedify/fixture": "^2.0.0",
-    "@fedify/testing": "^2.0.0-dev.323+1d796545"
+    "@fedify/testing": "^2.0.0-dev.335+6fe86704",
+    "@fedify/fixture": "^2.0.0"
   },
   "scripts": {
     "build:self": "tsdown",
@@ -90,6 +90,6 @@
     "pretest": "pnpm build",
     "test": "node --experimental-transform-types --test",
     "pretest:bun": "pnpm build",
-    "test:bun": "bun test --timeout=10000"
+    "test:bun": "bun test --timeout=60000"
   }
 }