queen-mq 0.12.0 → 0.12.1

package/client-v2/builders/TransactionBuilder.js

@@ -13,10 +13,10 @@ export class TransactionBuilder {
     this.#httpClient = httpClient
   }
 
-  ack(messages, status = 'completed') {
+  ack(messages, status = 'completed', context = {}) {
     const msgs = Array.isArray(messages) ? messages : [messages]
     
-    logger.log('TransactionBuilder.ack', { count: msgs.length, status })
+    logger.log('TransactionBuilder.ack', { count: msgs.length, status, consumerGroup: context.consumerGroup })
 
     msgs.forEach(msg => {
       const transactionId = typeof msg === 'string' ? msg : (msg.transactionId || msg.id)
@@ -32,12 +32,19 @@ export class TransactionBuilder {
         throw new Error('Message must have partitionId property to ensure message uniqueness')
       }
 
-      this.#operations.push({
+      const operation = {
         type: 'ack',
         transactionId,
         partitionId,
         status
-      })
+      }
+
+      // Add consumerGroup if provided in context
+      if (context.consumerGroup) {
+        operation.consumerGroup = context.consumerGroup
+      }
+
+      this.#operations.push(operation)
 
       if (leaseId) {
         this.#requiredLeases.push(leaseId)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "queen-mq",
-  "version": "0.12.0",
+  "version": "0.12.1",
   "type": "module",
   "description": "High-performance C++ message queue backed by PostgreSQL",
   "main": "client-v2/index.js",

package/test-v2/transaction.js

@@ -503,3 +503,69 @@ export async function transactionRollback(client) {
     }
 }
 
+export async function transactionAckWithConsumerGroup(client) {
+    const queueA = await client.queue('test-queue-v2-txn-cg-a').create()
+    const queueB = await client.queue('test-queue-v2-txn-cg-b').create()
+    if (!queueA.configured || !queueB.configured) {
+        return { success: false, message: 'Queues not created' }
+    }
+
+    const consumerGroup = 'test-consumer-group-txn'
+
+    // Push messages to queue A
+    await client.queue('test-queue-v2-txn-cg-a').push([
+        { data: { value: 1 } },
+        { data: { value: 2 } }
+    ])
+
+    // Pop with consumer group
+    const messages = await client.queue('test-queue-v2-txn-cg-a')
+        .group(consumerGroup)
+        .batch(2)
+        .wait(false)
+        .pop()
+
+    if (messages.length !== 2) {
+        return { success: false, message: `Expected 2 messages, got ${messages.length}` }
+    }
+
+    // Transaction: ACK with consumer group context, push to B
+    await client
+        .transaction()
+        .ack(messages[0], 'completed', { consumerGroup })
+        .ack(messages[1], 'completed', { consumerGroup })
+        .queue('test-queue-v2-txn-cg-b')
+        .push([{ data: { sum: messages[0].data.value + messages[1].data.value } }])
+        .commit()
+
+    // Verify: Pop again with same consumer group should get no messages
+    // (consumer group cursor should have advanced)
+    const messagesAfterAck = await client.queue('test-queue-v2-txn-cg-a')
+        .group(consumerGroup)
+        .batch(2)
+        .wait(false)
+        .pop()
+
+    // Verify: Queue B has the result
+    const resultB = await client.queue('test-queue-v2-txn-cg-b').batch(1).wait(false).pop()
+
+    // Verify: Using a different consumer group should still see the messages
+    const messagesOtherGroup = await client.queue('test-queue-v2-txn-cg-a')
+        .group('other-consumer-group')
+        .batch(2)
+        .wait(false)
+        .pop()
+
+    const success = messagesAfterAck.length === 0 && 
+                    resultB.length === 1 && 
+                    resultB[0].data.sum === 3 &&
+                    messagesOtherGroup.length === 2
+
+    return { 
+        success,
+        message: success 
+            ? 'Transaction ack with consumer group completed - cursor advanced correctly'
+            : `Failed: afterAck=${messagesAfterAck.length} (expected 0), resultB=${resultB.length} (expected 1), otherGroup=${messagesOtherGroup.length} (expected 2)`
+    }
+}
+