@vercel/queue 0.0.0-alpha.15 → 0.0.0-alpha.16

package/dist/index.js

@@ -632,207 +632,6 @@ var QueueClient = class {
   }
 };
 
-// src/consumer-group.ts
-var ConsumerGroup = class {
-  client;
-  topicName;
-  consumerGroupName;
-  visibilityTimeout;
-  refreshInterval;
-  transport;
-  /**
-   * Create a new ConsumerGroup instance
-   * @param client QueueClient instance to use for API calls
-   * @param topicName Name of the topic to consume from
-   * @param consumerGroupName Name of the consumer group
-   * @param options Optional configuration
-   */
-  constructor(client, topicName, consumerGroupName, options = {}) {
-    this.client = client;
-    this.topicName = topicName;
-    this.consumerGroupName = consumerGroupName;
-    this.visibilityTimeout = options.visibilityTimeoutSeconds || 30;
-    this.refreshInterval = options.refreshInterval || 10;
-    this.transport = options.transport || new JsonTransport();
-  }
-  /**
-   * Starts a background loop that periodically extends the visibility timeout for a message.
-   * This prevents the message from becoming visible to other consumers while it's being processed.
-   *
-   * The extension loop runs every `refreshInterval` seconds and updates the message's
-   * visibility timeout to `visibilityTimeout` seconds from the current time.
-   *
-   * @param messageId - The unique identifier of the message to extend visibility for
-   * @param ticket - The receipt ticket that proves ownership of the message
-   * @returns A function that when called will stop the extension loop
-   *
-   * @remarks
-   * - The first extension attempt occurs after `refreshInterval` seconds, not immediately
-   * - If an extension fails, the loop terminates with an error logged to console
-   * - The returned stop function is idempotent - calling it multiple times is safe
-   * - By default, the stop function returns immediately without waiting for in-flight
-   * - Pass `true` to the stop function to wait for any in-flight extension to complete
-   */
-  startVisibilityExtension(messageId, ticket) {
-    let isRunning = true;
-    let resolveLifecycle;
-    let timeoutId = null;
-    const lifecyclePromise = new Promise((resolve) => {
-      resolveLifecycle = resolve;
-    });
-    const extend = async () => {
-      if (!isRunning) {
-        resolveLifecycle();
-        return;
-      }
-      try {
-        await this.client.changeVisibility({
-          queueName: this.topicName,
-          consumerGroup: this.consumerGroupName,
-          messageId,
-          ticket,
-          visibilityTimeoutSeconds: this.visibilityTimeout
-        });
-        if (isRunning) {
-          timeoutId = setTimeout(() => extend(), this.refreshInterval * 1e3);
-        } else {
-          resolveLifecycle();
-        }
-      } catch (error) {
-        console.error(
-          `Failed to extend visibility for message ${messageId}:`,
-          error
-        );
-        resolveLifecycle();
-      }
-    };
-    timeoutId = setTimeout(() => extend(), this.refreshInterval * 1e3);
-    return async (waitForCompletion = false) => {
-      isRunning = false;
-      if (timeoutId) {
-        clearTimeout(timeoutId);
-        timeoutId = null;
-      }
-      if (waitForCompletion) {
-        await lifecyclePromise;
-      } else {
-        resolveLifecycle();
-      }
-    };
-  }
-  /**
-   * Process a single message with the given handler
-   * @param message The message to process
-   * @param handler Function to process the message
-   */
-  async processMessage(message, handler) {
-    const stopExtension = this.startVisibilityExtension(
-      message.messageId,
-      message.ticket
-    );
-    try {
-      const result = await handler(message.payload, {
-        messageId: message.messageId,
-        deliveryCount: message.deliveryCount,
-        createdAt: message.createdAt,
-        topicName: this.topicName,
-        consumerGroup: this.consumerGroupName
-      });
-      await stopExtension();
-      if (result && "timeoutSeconds" in result) {
-        await this.client.changeVisibility({
-          queueName: this.topicName,
-          consumerGroup: this.consumerGroupName,
-          messageId: message.messageId,
-          ticket: message.ticket,
-          visibilityTimeoutSeconds: result.timeoutSeconds
-        });
-      } else {
-        await this.client.deleteMessage({
-          queueName: this.topicName,
-          consumerGroup: this.consumerGroupName,
-          messageId: message.messageId,
-          ticket: message.ticket
-        });
-      }
-    } catch (error) {
-      await stopExtension();
-      if (this.transport.finalize && message.payload !== void 0 && message.payload !== null) {
-        try {
-          await this.transport.finalize(message.payload);
-        } catch (finalizeError) {
-          console.warn("Failed to finalize message payload:", finalizeError);
-        }
-      }
-      throw error;
-    }
-  }
-  async consume(handler, options) {
-    if (options?.messageId) {
-      if (options.skipPayload) {
-        const response = await this.client.receiveMessageById(
-          {
-            queueName: this.topicName,
-            consumerGroup: this.consumerGroupName,
-            messageId: options.messageId,
-            visibilityTimeoutSeconds: this.visibilityTimeout,
-            skipPayload: true
-          },
-          this.transport
-        );
-        await this.processMessage(
-          response.message,
-          handler
-        );
-      } else {
-        const response = await this.client.receiveMessageById(
-          {
-            queueName: this.topicName,
-            consumerGroup: this.consumerGroupName,
-            messageId: options.messageId,
-            visibilityTimeoutSeconds: this.visibilityTimeout
-          },
-          this.transport
-        );
-        await this.processMessage(
-          response.message,
-          handler
-        );
-      }
-    } else {
-      let messageFound = false;
-      for await (const message of this.client.receiveMessages(
-        {
-          queueName: this.topicName,
-          consumerGroup: this.consumerGroupName,
-          visibilityTimeoutSeconds: this.visibilityTimeout,
-          limit: 1
-        },
-        this.transport
-      )) {
-        messageFound = true;
-        await this.processMessage(message, handler);
-        break;
-      }
-      if (!messageFound) {
-        throw new Error("No messages available");
-      }
-    }
-  }
-  /**
-   * Get the consumer group name
-   */
-  get name() {
-    return this.consumerGroupName;
-  }
-  /**
-   * Get the topic name this consumer group is subscribed to
-   */
-  get topic() {
-    return this.topicName;
-  }
-};
-
 // src/callback.ts
 function validateWildcardPattern(pattern) {
   const firstIndex = pattern.indexOf("*");
@@ -1081,6 +880,17 @@ function createMockCloudEventRequest(topicName, consumerGroup, messageId) {
   });
 }
 var DEV_CALLBACK_DELAY = 1e3;
+function scheduleDevTimeout(topicName, messageId, timeoutSeconds) {
+  console.log(
+    `[Dev Mode] Message ${messageId} timed out for ${timeoutSeconds}s, will re-trigger`
+  );
+  setTimeout(() => {
+    console.log(
+      `[Dev Mode] Re-triggering callback for timed-out message ${messageId}`
+    );
+    triggerDevCallbacks(topicName, messageId);
+  }, timeoutSeconds * 1e3);
+}
 function triggerDevCallbacks(topicName, messageId) {
   const handlersMap = findRouteHandlersForTopic(topicName);
   if (handlersMap.size === 0) {
@@ -1150,6 +960,214 @@ if (process.env.NODE_ENV === "test" || process.env.VITEST) {
   globalThis.__clearDevHandlers = clearDevHandlers;
 }
 
+// src/consumer-group.ts
+var ConsumerGroup = class {
+  client;
+  topicName;
+  consumerGroupName;
+  visibilityTimeout;
+  refreshInterval;
+  transport;
+  /**
+   * Create a new ConsumerGroup instance
+   * @param client QueueClient instance to use for API calls
+   * @param topicName Name of the topic to consume from
+   * @param consumerGroupName Name of the consumer group
+   * @param options Optional configuration
+   */
+  constructor(client, topicName, consumerGroupName, options = {}) {
+    this.client = client;
+    this.topicName = topicName;
+    this.consumerGroupName = consumerGroupName;
+    this.visibilityTimeout = options.visibilityTimeoutSeconds || 30;
+    this.refreshInterval = options.refreshInterval || 10;
+    this.transport = options.transport || new JsonTransport();
+  }
+  /**
+   * Starts a background loop that periodically extends the visibility timeout for a message.
+   * This prevents the message from becoming visible to other consumers while it's being processed.
+   *
+   * The extension loop runs every `refreshInterval` seconds and updates the message's
+   * visibility timeout to `visibilityTimeout` seconds from the current time.
+   *
+   * @param messageId - The unique identifier of the message to extend visibility for
+   * @param ticket - The receipt ticket that proves ownership of the message
+   * @returns A function that when called will stop the extension loop
+   *
+   * @remarks
+   * - The first extension attempt occurs after `refreshInterval` seconds, not immediately
+   * - If an extension fails, the loop terminates with an error logged to console
+   * - The returned stop function is idempotent - calling it multiple times is safe
+   * - By default, the stop function returns immediately without waiting for in-flight
+   * - Pass `true` to the stop function to wait for any in-flight extension to complete
+   */
+  startVisibilityExtension(messageId, ticket) {
+    let isRunning = true;
+    let resolveLifecycle;
+    let timeoutId = null;
+    const lifecyclePromise = new Promise((resolve) => {
+      resolveLifecycle = resolve;
+    });
+    const extend = async () => {
+      if (!isRunning) {
+        resolveLifecycle();
+        return;
+      }
+      try {
+        await this.client.changeVisibility({
+          queueName: this.topicName,
+          consumerGroup: this.consumerGroupName,
+          messageId,
+          ticket,
+          visibilityTimeoutSeconds: this.visibilityTimeout
+        });
+        if (isRunning) {
+          timeoutId = setTimeout(() => extend(), this.refreshInterval * 1e3);
+        } else {
+          resolveLifecycle();
+        }
+      } catch (error) {
+        console.error(
+          `Failed to extend visibility for message ${messageId}:`,
+          error
+        );
+        resolveLifecycle();
+      }
+    };
+    timeoutId = setTimeout(() => extend(), this.refreshInterval * 1e3);
+    return async (waitForCompletion = false) => {
+      isRunning = false;
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+        timeoutId = null;
+      }
+      if (waitForCompletion) {
+        await lifecyclePromise;
+      } else {
+        resolveLifecycle();
+      }
+    };
+  }
+  /**
+   * Process a single message with the given handler
+   * @param message The message to process
+   * @param handler Function to process the message
+   */
+  async processMessage(message, handler) {
+    const stopExtension = this.startVisibilityExtension(
+      message.messageId,
+      message.ticket
+    );
+    try {
+      const result = await handler(message.payload, {
+        messageId: message.messageId,
+        deliveryCount: message.deliveryCount,
+        createdAt: message.createdAt,
+        topicName: this.topicName,
+        consumerGroup: this.consumerGroupName
+      });
+      await stopExtension();
+      if (result && "timeoutSeconds" in result) {
+        await this.client.changeVisibility({
+          queueName: this.topicName,
+          consumerGroup: this.consumerGroupName,
+          messageId: message.messageId,
+          ticket: message.ticket,
+          visibilityTimeoutSeconds: result.timeoutSeconds
+        });
+        if (isDevMode()) {
+          scheduleDevTimeout(
+            this.topicName,
+            message.messageId,
+            result.timeoutSeconds
+          );
+        }
+      } else {
+        await this.client.deleteMessage({
+          queueName: this.topicName,
+          consumerGroup: this.consumerGroupName,
+          messageId: message.messageId,
+          ticket: message.ticket
+        });
+      }
+    } catch (error) {
+      await stopExtension();
+      if (this.transport.finalize && message.payload !== void 0 && message.payload !== null) {
+        try {
+          await this.transport.finalize(message.payload);
+        } catch (finalizeError) {
+          console.warn("Failed to finalize message payload:", finalizeError);
+        }
+      }
+      throw error;
+    }
+  }
+  async consume(handler, options) {
+    if (options?.messageId) {
+      if (options.skipPayload) {
+        const response = await this.client.receiveMessageById(
+          {
+            queueName: this.topicName,
+            consumerGroup: this.consumerGroupName,
+            messageId: options.messageId,
+            visibilityTimeoutSeconds: this.visibilityTimeout,
+            skipPayload: true
+          },
+          this.transport
+        );
+        await this.processMessage(
+          response.message,
+          handler
+        );
+      } else {
+        const response = await this.client.receiveMessageById(
+          {
+            queueName: this.topicName,
+            consumerGroup: this.consumerGroupName,
+            messageId: options.messageId,
+            visibilityTimeoutSeconds: this.visibilityTimeout
+          },
+          this.transport
+        );
+        await this.processMessage(
+          response.message,
+          handler
+        );
+      }
+    } else {
+      let messageFound = false;
+      for await (const message of this.client.receiveMessages(
+        {
+          queueName: this.topicName,
+          consumerGroup: this.consumerGroupName,
+          visibilityTimeoutSeconds: this.visibilityTimeout,
+          limit: 1
+        },
+        this.transport
+      )) {
+        messageFound = true;
+        await this.processMessage(message, handler);
+        break;
+      }
+      if (!messageFound) {
+        throw new Error("No messages available");
+      }
+    }
+  }
+  /**
+   * Get the consumer group name
+   */
+  get name() {
+    return this.consumerGroupName;
+  }
+  /**
+   * Get the topic name this consumer group is subscribed to
+   */
+  get topic() {
+    return this.topicName;
+  }
+};
+
 // src/topic.ts
 var Topic = class {
   client;