Sha256: 12f7025e5827d370ccfb85c3ad87474a947cfb2acd0b36a6ff8f1621186efcea
Contents?: true
Size: 2 KB
Versions: 12
Compression:
Stored size: 2 KB
Contents
# typed: true
require_relative '../../context'
require_relative '../analytics'
require_relative '../active_support/notifications/event'
require_relative 'ext'
module Datadog
module Tracing
module Contrib
module ActionCable
# Defines basic behaviors for an event.
module Event
def self.included(base)
base.include(ActiveSupport::Notifications::Event)
base.extend(ClassMethods)
end
# Class methods for events.
module ClassMethods
def span_options
if configuration[:service_name]
{ service: configuration[:service_name] }
else
{}
end
end
def configuration
Datadog.configuration.tracing[:action_cable]
end
end
end
# Defines behavior for the first event of a thread execution.
#
# This event is not expected to be nested with other event,
# but to start a fresh tracing context.
module RootContextEvent
def self.included(base)
base.include(ActiveSupport::Notifications::Event)
base.extend(ClassMethods)
end
# Class methods for events.
module ClassMethods
include Contrib::ActionCable::Event::ClassMethods
def subscription(*args)
super.tap do |subscription|
subscription.before_trace { ensure_clean_context! }
end
end
private
# Context objects are thread-bound.
# If an integration re-uses threads, context from a previous trace
# could leak into the new trace. This "cleans" current context,
# preventing such a leak.
def ensure_clean_context!
return unless Tracing.active_span
Tracing.send(:tracer).provider.context = Tracing::Context.new
end
end
end
end
end
end
end
Version data entries
12 entries across 12 versions & 1 rubygems