Sha256: 5ca31d0f0cc3b3e22fc460a03bb1a4a9f02ff815b91d2ae74da7bf5dc25e5db0

Contents?: true

Size: 1.33 KB

Versions: 7

Compression:

Stored size: 1.33 KB

Contents

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

module NewRelic::Agent::Instrumentation
  module ConcurrentRuby::Chain
    def self.instrument!
      ::Concurrent::ThreadPoolExecutor.class_eval do
        include NewRelic::Agent::Instrumentation::ConcurrentRuby

        alias_method(:post_without_new_relic, :post)

        def post(*args, &task)
          return post_without_new_relic(*args, &task) unless NewRelic::Agent::Tracer.tracing_enabled?

          traced_task = add_task_tracing(*args, &task)
          post_without_new_relic(*args, &traced_task)
        end
      end

      [::Concurrent::Promises.const_get(:'InternalStates')::Rejected,
        ::Concurrent::Promises.const_get(:'InternalStates')::PartiallyRejected].each do |klass|
        klass.class_eval do
          alias_method(:initialize_without_new_relic, :initialize)

          # Uses args.last to record the error becuase the methods that this will monkey patch
          # look like: initialize(reason) & initialize(value, reason)
          def initialize(*args)
            NewRelic::Agent.notice_error(args.last) if args.last.is_a?(Exception)
            initialize_without_new_relic(*args)
          end
        end
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
newrelic_rpm-9.2.2 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-9.2.1 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-9.2.0 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-9.1.0 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-9.0.0 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-8.16.0 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb
newrelic_rpm-8.15.0 lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb