Sha256: 1b55e32b652344ccc3ec04061184f07b8bc27babdad5daeb483ff1e92af1ea1d

Contents?: true

Size: 1.68 KB

Versions: 5

Compression:

Stored size: 1.68 KB

Contents

# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
require 'new_relic/agent/instrumentation/notifications_subscriber'

module NewRelic
  module Agent
    module Instrumentation
      class ActiveStorageSubscriber < NotificationsSubscriber
        def start name, id, payload
          return unless state.is_execution_traced?
          start_segment name, id, payload
        rescue => e
          log_notification_error e, name, 'start'
        end

        def finish name, id, payload
          return unless state.is_execution_traced?
          finish_segment id
        rescue => e
          log_notification_error e, name, 'finish'
        end

        def start_segment name, id, payload
          segment = Tracer.start_segment name: metric_name(name, payload)
          segment.params[:key] = payload[:key]
          segment.params[:exist] = payload[:exist] if payload.key? :exist
          push_segment id, segment
        end

        def finish_segment id
          segment = pop_segment id
          segment.finish if segment
        end

        def metric_name name, payload
          service = payload[:service]
          method = method_from_name name
          "Ruby/ActiveStorage/#{service}Service/#{method}"
        end

        PATTERN = /\Aservice_([^\.]*)\.active_storage\z/
        UNKNOWN = "unknown".freeze

        METHOD_NAME_MAPPING = Hash.new do |h, k|
          if PATTERN =~ k
            h[k] = $1
          else
            h[k] = UNKNOWN
          end
        end

        def method_from_name name
          METHOD_NAME_MAPPING[name]
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
newrelic_rpm-6.9.0.363 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.8.0.360 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.7.0.359 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.6.0.358 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.5.0.357 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb