Sha256: bfd991f2e6b78186547f26a811893fff537e0b13a7505cb9cec112a19d93eed6
Contents?: true
Size: 1.88 KB
Versions: 4
Compression:
Stored size: 1.88 KB
Contents
require 'ddtrace/contrib/analytics' require 'ddtrace/contrib/active_support/notifications/event' require 'ddtrace/contrib/action_cable/ext' module Datadog module Contrib module ActionCable # Defines basic behaviors for an event. module Event def self.included(base) base.send(:include, ActiveSupport::Notifications::Event) base.send(:extend, ClassMethods) end # Class methods for events. module ClassMethods def span_options { service: configuration[:service_name] } end def tracer -> { configuration[:tracer] } end def configuration Datadog.configuration[: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.send(:include, ActiveSupport::Notifications::Event) base.send(: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 configuration[:tracer].call_context.current_span configuration[:tracer].provider.context = Context.new end end end end end end
Version data entries
4 entries across 4 versions & 1 rubygems