def _translate_span(self, span: Span) -> model_pb2.Span:
ctx = span.get_span_context()
# pb2 span expects in byte format
trace_id = _trace_id_to_bytes(ctx.trace_id)
span_id = _span_id_to_bytes(ctx.span_id)
start_time = _proto_timestamp_from_epoch_nanos(span.start_time)
end_time = _proto_timestamp_from_epoch_nanos(span.end_time)
duration = _duration_from_two_time_stamps(start_time, end_time)
tags = self._extract_tags(span)
refs = self._extract_refs(span)
logs = self._extract_logs(span)
flags = int(ctx.trace_flags)
process = model_pb2.Process(service_name=self.svc_name,
tags=_extract_resource_tags(span))
jaeger_span = model_pb2.Span(
trace_id=trace_id,
span_id=span_id,
operation_name=span.name,
references=refs,
flags=flags,
start_time=start_time,
duration=duration,
tags=tags,
logs=logs,
process=process,
)
return jaeger_span
def _translate_span(self, span: Span) -> TCollector.Span:
ctx = span.get_span_context()
trace_id = ctx.trace_id
span_id = ctx.span_id
start_time_us = _nsec_to_usec_round(span.start_time)
duration_us = _nsec_to_usec_round(span.end_time - span.start_time)
parent_id = span.parent.span_id if span.parent else 0
tags = self._extract_tags(span)
refs = self._extract_refs(span)
logs = self._extract_logs(span)
flags = int(ctx.trace_flags)
jaeger_span = TCollector.Span(
traceIdHigh=_get_trace_id_high(trace_id),
traceIdLow=_get_trace_id_low(trace_id),
spanId=_convert_int_to_i64(span_id),
operationName=span.name,
startTime=start_time_us,
duration=duration_us,
tags=tags,
logs=logs,
references=refs,
flags=flags,
parentSpanId=_convert_int_to_i64(parent_id),
)
return jaeger_span
def on_end(self, span: Span) -> None:
if self.done:
logger.warning("Already shutdown, dropping span.")
return
ctx = span.get_span_context()
trace_id = ctx.trace_id
with self.traces_lock:
self.traces_spans_ended_count[trace_id] += 1
if self.is_trace_exportable(trace_id):
self.check_traces_queue.appendleft(trace_id)
def on_start(self, span: Span) -> None:
ctx = span.get_span_context()
trace_id = ctx.trace_id
with self.traces_lock:
# check upper bound on number of spans for trace before adding new
# span
if self.traces_spans_count[trace_id] == self.max_trace_size:
logger.warning("Max spans for trace, spans will be dropped.")
self._spans_dropped = True
return
# add span to end of list for a trace and update the counter
self.traces[trace_id].append(span)
self.traces_spans_count[trace_id] += 1
def _encode_span(sdk_span: SDKSpan) -> PB2SPan:
span_context = sdk_span.get_span_context()
return PB2SPan(
trace_id=_encode_trace_id(span_context.trace_id),
span_id=_encode_span_id(span_context.span_id),
trace_state=_encode_trace_state(span_context.trace_state),
parent_span_id=_encode_parent_id(sdk_span.parent),
name=sdk_span.name,
kind=_SPAN_KIND_MAP[sdk_span.kind],
start_time_unix_nano=sdk_span.start_time,
end_time_unix_nano=sdk_span.end_time,
attributes=_encode_attributes(sdk_span.attributes),
events=_encode_events(sdk_span.events),
links=_encode_links(sdk_span.links),
status=_encode_status(sdk_span.status),
)