Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions lib/ld-eventsource/client.rb
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,9 @@ def initialize(uri,

yield self if block_given?

trace_digest = capture_trace_digest
Thread.new do
continue_trace_context(trace_digest)
run_stream
end
end
Expand Down Expand Up @@ -379,5 +381,21 @@ def build_headers
h['Last-Event-Id'] = @last_id if !@last_id.nil? && @last_id != ""
h.merge(@headers)
end

def capture_trace_digest
return nil unless datadog_available?
active_trace = Datadog::Tracing.active_trace
return nil unless active_trace
active_trace.to_digest
end

def continue_trace_context(digest)
return unless datadog_available? && digest
Datadog::Tracing.continue_trace!(digest)
end

def datadog_available?
defined?(Datadog::Tracing) && defined?(Datadog::Tracing::TraceDigest)
end
end
end