newrelic/lib/subscribers/amqplib/send-message.js

New Relic agent

/*
 * Copyright 2025 New Relic Corporation. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

const MessageProducerSubscriber = require('../message-producer')
const { getParameters, TEMP_RE } = require('./utils')

class ChannelSubscriber extends MessageProducerSubscriber {
  constructor({ agent, logger }) {
    super({ agent, logger, packageName: 'amqplib', channelName: 'nr_sendMessage', system: 'RabbitMQ', type: 'Exchange' })
    this.events = ['end']
  }

  handler(data, ctx) {
    const [fields] = data.arguments

    if (!fields) {
      return ctx
    }

    this.destination = this.getExchange(fields)
    this.parameters = this.getParameters({ fields, channel: data.self })
    this.headers = fields?.headers
    return super.handler(data, ctx)
  }

  getParameters({ fields = {}, channel = {} }) {
    const params = {}
    getParameters.call(this, { fields, channel, params })
    return params
  }

  getExchange(fields = {}) {
    const isDefault = fields.exchange === ''
    let exchange = 'Default'
    if (!isDefault) {
      exchange = TEMP_RE.test(fields.exchange) ? null : fields.exchange
    }
    return exchange
  }
}

module.exports = ChannelSubscriber