# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.

module NewRelic::Agent::Instrumentation
  module Bunny
    def self.instrument!
      ::Bunny::Exchange.class_eval do
        include NewRelic::Agent::Instrumentation::Bunny::Exchange

        alias_method :publish_without_new_relic, :publish

        def publish payload, opts = {}
          publish_with_tracing(payload, opts) { publish_without_new_relic payload, opts }
        end
      end

      ::Bunny::Queue.class_eval do
        include NewRelic::Agent::Instrumentation::Bunny::Queue

        alias_method :pop_without_new_relic, :pop

        def pop(opts = {:manual_ack => false}, &block)
          pop_with_tracing { pop_without_new_relic opts, &block }
        end

        alias_method :purge_without_new_relic, :purge

        def purge *args
          purge_with_tracing { purge_without_new_relic(*args) }
        end
      end

      ::Bunny::Consumer.class_eval do
        include NewRelic::Agent::Instrumentation::Bunny::Consumer

        alias_method :call_without_new_relic, :call

        def call *args
          call_with_tracing(*args) { call_without_new_relic(*args) }
        end
      end
    end
  end
end