Sha256: ddbb14665f94cd391d987592048d698a4e2d6501778d98581b9403709d421100
Contents?: true
Size: 1.3 KB
Versions: 78
Compression:
Stored size: 1.3 KB
Contents
require_relative 'datadog_interceptor'
module Datadog
module Contrib
module GRPC
# :nodoc:
# The `#intercept!` method is implemented in gRPC; this module
# will be prepended to the original class, effectively injecting
# our tracing middleware into the head of the call chain.
module InterceptWithDatadog
def intercept!(type, args = {})
if should_prepend?
datadog_interceptor = choose_datadog_interceptor(args)
@interceptors.unshift(datadog_interceptor.new) if datadog_interceptor
@trace_started = true
end
super
end
private
def should_prepend?
!trace_started? && !already_prepended?
end
def trace_started?
defined?(@trace_started) && @trace_started
end
def already_prepended?
@interceptors.any? do |interceptor|
interceptor.class.ancestors.include?(Datadog::Contrib::GRPC::DatadogInterceptor::Base)
end
end
def choose_datadog_interceptor(args)
if args.key?(:metadata)
Datadog::Contrib::GRPC::DatadogInterceptor::Client
elsif args.key?(:call)
Datadog::Contrib::GRPC::DatadogInterceptor::Server
end
end
end
end
end
end
Version data entries
78 entries across 78 versions & 2 rubygems