Sha256: dbd9acd56ffc29027a4330aca2a17bd203dba201a9e0767eaab67321051c4b1e

Contents?: true

Size: 1.67 KB

Versions: 4

Compression:

Stored size: 1.67 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/evented_subscriber'

module NewRelic
  module Agent
    module Instrumentation
      class ActiveStorageSubscriber < EventedSubscriber
        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
          event_stack[id].push segment
        end

        def finish_segment id
          segment = event_stack[id].pop
          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

4 entries across 4 versions & 1 rubygems

Version Path
newrelic_rpm-6.3.0.355 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.2.0.354 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.1.0.352 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb
newrelic_rpm-6.0.0.351 lib/new_relic/agent/instrumentation/active_storage_subscriber.rb