Class: ZipkinTracer::TraceWrapper

Inherits:
Object
  • Object
show all
Defined in:
lib/zipkin-tracer/trace_wrapper.rb

Constant Summary collapse

REQUIRED_KEYS = 
%i[trace_id span_id].freeze
KEYS = 
%i[trace_id parent_id span_id sampled].freeze

Class Method Summary collapse

Class Method Details

.initialize_tracer(app, config) ⇒ Object 



24
25
26
27
28
29
 
# File 'lib/zipkin-tracer/trace_wrapper.rb', line 24

def self.initialize_tracer(app, config)
  return if Trace.tracer

  zipkin_config = ZipkinTracer::Config.new(app, config).freeze
  ZipkinTracer::TracerFactory.new.tracer(zipkin_config)
end

.next_trace_id(trace_context) ⇒ Object 



31
32
33
34
35
36
37
 
# File 'lib/zipkin-tracer/trace_wrapper.rb', line 31

def self.next_trace_id(trace_context)
  if trace_context.is_a?(Hash) && REQUIRED_KEYS.all? { |key| trace_context.key?(key) }
    Trace::TraceId.new(*trace_context.values_at(*KEYS), Trace::Flags::EMPTY).next_id
  else
    ZipkinTracer::TraceGenerator.new.next_trace_id
  end
end

.wrap_in_custom_span(config, span_name, span_kind: Trace::Span::Kind::SERVER, app: nil, trace_context: nil) ⇒ Object

Raises:

  • (ArgumentError) 


6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/zipkin-tracer/trace_wrapper.rb', line 6

def self.wrap_in_custom_span(config, span_name, span_kind: Trace::Span::Kind::SERVER, app: nil, trace_context: nil)
  raise ArgumentError, "you must provide a block" unless block_given?

  initialize_tracer(app, config)
  trace_id = next_trace_id(trace_context)

  ZipkinTracer::TraceContainer.with_trace_id(trace_id) do
    if trace_id.sampled?
      Trace.tracer.with_new_span(trace_id, span_name) do |span|
        span.kind = span_kind
        yield(span)
      end
    else
      yield(ZipkinTracer::NullSpan.new)
    end
  end
end