lib/datadog/tracing/contrib/concurrent_ruby/patcher.rb


# frozen_string_literal: true

require_relative '../patcher'

module Datadog
  module Tracing
    module Contrib
      module ConcurrentRuby
        # Patcher enables patching of 'Future' class.
        module Patcher
          include Contrib::Patcher

          module_function

          def target_version
            Integration.version
          end

          def patch
            require_relative 'future_patch'
            patch_future
            require_relative 'promises_future_patch'
            patch_promises_future
          end

          # Propagate tracing context in Concurrent::Future
          def patch_future
            ::Concurrent::Future.prepend(FuturePatch) if defined?(::Concurrent::Future)
          end

          # Propagate tracing context in Concurrent::Promises::Future
          def patch_promises_future
            ::Concurrent::Promises.singleton_class.prepend(PromisesFuturePatch) if defined?(::Concurrent::Promises::Future)
          end
        end
      end
    end
  end
end