fix: Updated MessageConsumerSubscriber to end transaction created for a given message consumption (#3503)

Author: bizob2828

lib/subscribers/message-consumer.js

@@ -65,6 +65,14 @@ class MessageConsumerSubscriber extends Subscriber {
     }
   }
 
+  /**
+   * Ends the transaction created for the consumption callback.
+   */
+  asyncEnd() {
+    const ctx = this.agent.tracer.getContext()
+    ctx?.transaction?.end()
+  }
+
   enable() {
     super.enable()
     this.channel.asyncStart.bindStore(this.store, (data) => {
@@ -83,9 +91,8 @@ class MessageConsumerSubscriber extends Subscriber {
   /**
    * Used to create a transaction for every consumption callback.
    *
-   * @param {object} data event passed to asyncStart handler
    */
-  asyncStart(data) {
+  asyncStart() {
     const ctx = this.agent.tracer.getContext()
     const tx = ctx?.transaction
     tx.setPartialName(this.name)

test/versioned/amqplib/callback.test.js

@@ -323,7 +323,6 @@ test('amqplib callback instrumentation', async function (t) {
 
             channel.ack(msg)
             produceTx.end()
-            consumeTx.end()
             resolve()
           })
           helper.runInTransaction(agent, function (tx) {
@@ -373,7 +372,6 @@ test('amqplib callback instrumentation', async function (t) {
 
             channel.ack(msg)
             produceTx.end()
-            consumeTx.end()
             resolve()
           })