Sha256: d48453303b9928a4d428fe9198009290fb9fc10417b98a7b5f75a6b36f2cda45
Contents?: true
Size: 1.8 KB
Versions: 12
Compression:
Stored size: 1.8 KB
Contents
require 'thread'
module Datadog
# Buffer used to store active spans
class SpanBuffer
# ensure that a new SpanBuffer clears the thread spans
def initialize
Thread.current[:datadog_span] = nil
end
# Set the current active span.
def set(span)
Thread.current[:datadog_span] = span
end
# Return the current active span or nil.
def get
Thread.current[:datadog_span]
end
# Pop the current active span.
def pop
span = get()
set(nil)
span
end
end
# Trace buffer that stores application traces. The buffer has a maximum size and when
# the buffer is full, a random trace is discarded. This class is thread-safe and is used
# automatically by the ``Tracer`` instance when a ``Span`` is finished.
class TraceBuffer
def initialize(max_size)
@max_size = max_size
@mutex = Mutex.new()
@traces = []
end
# Add a new ``trace`` in the local queue. This method doesn't block the execution
# even if the buffer is full. In that case, a random trace is discarded.
def push(trace)
@mutex.synchronize do
len = @traces.length
if len < @max_size || @max_size <= 0
@traces << trace
else
# we should replace a random trace with the new one
@traces[rand(len)] = trace
end
end
end
# Return the current number of stored traces.
def length
@mutex.synchronize do
return @traces.length
end
end
# Return if the buffer is empty.
def empty?
@mutex.synchronize do
return @traces.empty?
end
end
# Stored traces are returned and the local buffer is reset
def pop
@mutex.synchronize do
traces = @traces
@traces = []
return traces
end
end
end
end
Version data entries
12 entries across 12 versions & 1 rubygems