Contents

require 'ddtrace/ext/app_types'
require 'ddtrace/contrib/analytics'
require 'ddtrace/contrib/grpc/ext'

module Datadog
  module Contrib
    module GRPC
      # :nodoc:
      module DatadogInterceptor
        # :nodoc:
        class Base < ::GRPC::Interceptor
          attr_accessor :datadog_pin

          def initialize(options = {})
            add_datadog_pin! { |c| yield(c) if block_given? }
          end

          def request_response(**keywords, &block)
            trace(keywords, &block)
          end

          def client_streamer(**keywords, &block)
            trace(keywords, &block)
          end

          def server_streamer(**keywords, &block)
            trace(keywords, &block)
          end

          def bidi_streamer(**keywords, &block)
            trace(keywords, &block)
          end

          private

          def add_datadog_pin!
            Pin.new(
              service_name,
              app: Ext::APP,
              app_type: Datadog::Ext::AppTypes::WEB,
              tracer: -> { datadog_configuration[:tracer] }
            ).tap do |pin|
              yield(pin) if block_given?
              pin.onto(self)
            end
          end

          def datadog_configuration
            Datadog.configuration[:grpc]
          end

          def tracer
            (datadog_pin && datadog_pin.tracer) || datadog_configuration[:tracer]
          end

          def service_name
            (datadog_pin && datadog_pin.service_name) || datadog_configuration[:service_name]
          end

          def analytics_enabled?
            Contrib::Analytics.enabled?(datadog_configuration[:analytics_enabled])
          end

          def analytics_sample_rate
            datadog_configuration[:analytics_sample_rate]
          end
        end

        require_relative 'datadog_interceptor/client'
        require_relative 'datadog_interceptor/server'
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
ddtrace-0.51.1	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.51.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.50.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.49.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.48.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.47.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb
ddtrace-0.46.0	lib/ddtrace/contrib/grpc/datadog_interceptor.rb