Class: Datadog::Tracing::DefaultContextProvider

Inherits:
Object
  • Object
show all
Defined in:
lib/datadog/tracing/context_provider.rb

Overview

DefaultContextProvider is a default context provider that retrieves all contexts from the current fiber-local storage. It is suitable for synchronous programming.

See Also:

Instance Method Summary collapse

Constructor Details

#initialize ⇒ DefaultContextProvider

Initializes the default context provider with a fiber-bound context.



15
16
17
 
# File 'lib/datadog/tracing/context_provider.rb', line 15

def initialize
  @context = FiberLocalContext.new
end

Instance Method Details

#context(key = nil) ⇒ Object

Return the local context.



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/datadog/tracing/context_provider.rb', line 25

def context(key = nil)
  current_context = key.nil? ? @context.local : @context.local(key)

  # Rebuild/reset context after a fork
  #
  # We don't want forked processes to copy and retransmit spans
  # that were generated from the parent process. Reset it such
  # that it acts like a distributed trace.
  current_context.after_fork! do
    # TODO: Only assign to `self.context` when working on the current thread (`key == nil`)
    current_context = self.context = current_context.fork_clone
  end

  current_context
end

#context=(ctx) ⇒ Object

Sets the current context.



20
21
22
 
# File 'lib/datadog/tracing/context_provider.rb', line 20

def context=(ctx)
  @context.local = ctx
end