def _extract_logs(self, span: Span) -> Optional[Sequence[model_pb2.Log]]:
if not span.events:
return None
logs = []
for event in span.events:
fields = []
for key, value in event.attributes.items():
tag = _translate_attribute(key, value)
if tag:
fields.append(tag)
fields.append(
_get_string_key_value(
key="message",
value=event.name,
))
event_ts = _proto_timestamp_from_epoch_nanos(event.timestamp)
logs.append(model_pb2.Log(timestamp=event_ts, fields=fields))
return logs
def test_translate_to_jaeger(self):
span_names = ("test1", "test2", "test3")
trace_id = 0x6E0C63257DE34C926F9EFCD03927272E
span_id = 0x34BF92DEEFC58C92
parent_id = 0x1111111111111111
other_id = 0x2222222222222222
base_time = 683647322 * 10 ** 9 # in ns
start_times = (
base_time,
base_time + 150 * 10 ** 6,
base_time + 300 * 10 ** 6,
)
durations = (50 * 10 ** 6, 100 * 10 ** 6, 200 * 10 ** 6)
end_times = (
start_times[0] + durations[0],
start_times[1] + durations[1],
start_times[2] + durations[2],
)
span_context = trace_api.SpanContext(
trace_id, span_id, is_remote=False
)
parent_span_context = trace_api.SpanContext(
trace_id, parent_id, is_remote=False
)
other_context = trace_api.SpanContext(
trace_id, other_id, is_remote=False
)
event_attributes = OrderedDict(
[
("annotation_bool", True),
("annotation_string", "annotation_test"),
("key_float", 0.3),
]
)
event_timestamp = base_time + 50 * 10 ** 6
# pylint:disable=protected-access
event_timestamp_proto = pb_translator._proto_timestamp_from_epoch_nanos(
event_timestamp
)
event = trace.Event(
name="event0",
timestamp=event_timestamp,
attributes=event_attributes,
)
link_attributes = {"key_bool": True}
link = trace_api.Link(
context=other_context, attributes=link_attributes
)
default_tags = [
model_pb2.KeyValue(
key="span.kind",
v_type=model_pb2.ValueType.STRING,
v_str="internal",
),
]
otel_spans = [
trace._Span(
name=span_names[0],
context=span_context,
parent=parent_span_context,
events=(event,),
links=(link,),
kind=trace_api.SpanKind.CLIENT,
),
trace._Span(
name=span_names[1], context=parent_span_context, parent=None
),
trace._Span(
name=span_names[2], context=other_context, parent=None
),
]
otel_spans[0].start(start_time=start_times[0])
# added here to preserve order
otel_spans[0].set_attribute("key_bool", False)
otel_spans[0].set_attribute("key_string", "hello_world")
otel_spans[0].set_attribute("key_float", 111.22)
otel_spans[0].set_attribute("key_tuple", ("tuple_element",))
otel_spans[0].resource = Resource(
attributes={"key_resource": "some_resource"}
)
otel_spans[0].set_status(
Status(StatusCode.ERROR, "Example description")
)
otel_spans[0].end(end_time=end_times[0])
otel_spans[1].start(start_time=start_times[1])
otel_spans[1].resource = Resource({})
otel_spans[1].end(end_time=end_times[1])
otel_spans[2].start(start_time=start_times[2])
otel_spans[2].resource = Resource({})
otel_spans[2].set_status(Status(StatusCode.OK, "Example description"))
otel_spans[2].end(end_time=end_times[2])
otel_spans[2].instrumentation_info = InstrumentationInfo(
name="name", version="version"
)
translate = Translate(otel_spans)
# pylint: disable=protected-access
spans = translate._translate(pb_translator.ProtobufTranslator("svc"))
span1_start_time = pb_translator._proto_timestamp_from_epoch_nanos(
start_times[0]
)
span2_start_time = pb_translator._proto_timestamp_from_epoch_nanos(
start_times[1]
)
span3_start_time = pb_translator._proto_timestamp_from_epoch_nanos(
start_times[2]
)
span1_end_time = pb_translator._proto_timestamp_from_epoch_nanos(
end_times[0]
)
span2_end_time = pb_translator._proto_timestamp_from_epoch_nanos(
end_times[1]
)
span3_end_time = pb_translator._proto_timestamp_from_epoch_nanos(
end_times[2]
)
span1_duration = pb_translator._duration_from_two_time_stamps(
span1_start_time, span1_end_time
)
span2_duration = pb_translator._duration_from_two_time_stamps(
span2_start_time, span2_end_time
)
span3_duration = pb_translator._duration_from_two_time_stamps(
span3_start_time, span3_end_time
)
expected_spans = [
model_pb2.Span(
operation_name=span_names[0],
trace_id=pb_translator._trace_id_to_bytes(trace_id),
span_id=pb_translator._span_id_to_bytes(span_id),
start_time=span1_start_time,
duration=span1_duration,
flags=0,
tags=[
model_pb2.KeyValue(
key="key_bool",
v_type=model_pb2.ValueType.BOOL,
v_bool=False,
),
model_pb2.KeyValue(
key="key_string",
v_type=model_pb2.ValueType.STRING,
v_str="hello_world",
),
model_pb2.KeyValue(
key="key_float",
v_type=model_pb2.ValueType.FLOAT64,
v_float64=111.22,
),
model_pb2.KeyValue(
key="key_tuple",
v_type=model_pb2.ValueType.STRING,
v_str="('tuple_element',)",
),
model_pb2.KeyValue(
key="key_resource",
v_type=model_pb2.ValueType.STRING,
v_str="some_resource",
),
model_pb2.KeyValue(
key="otel.status_code",
v_type=model_pb2.ValueType.STRING,
v_str="ERROR",
),
model_pb2.KeyValue(
key="otel.status_description",
v_type=model_pb2.ValueType.STRING,
v_str="Example description",
),
model_pb2.KeyValue(
key="span.kind",
v_type=model_pb2.ValueType.STRING,
v_str="client",
),
model_pb2.KeyValue(
key="error",
v_type=model_pb2.ValueType.BOOL,
v_bool=True,
),
],
references=[
model_pb2.SpanRef(
ref_type=model_pb2.SpanRefType.FOLLOWS_FROM,
trace_id=pb_translator._trace_id_to_bytes(trace_id),
span_id=pb_translator._span_id_to_bytes(other_id),
)
],
logs=[
model_pb2.Log(
timestamp=event_timestamp_proto,
fields=[
model_pb2.KeyValue(
key="annotation_bool",
v_type=model_pb2.ValueType.BOOL,
v_bool=True,
),
model_pb2.KeyValue(
key="annotation_string",
v_type=model_pb2.ValueType.STRING,
v_str="annotation_test",
),
model_pb2.KeyValue(
key="key_float",
v_type=model_pb2.ValueType.FLOAT64,
v_float64=0.3,
),
model_pb2.KeyValue(
key="message",
v_type=model_pb2.ValueType.STRING,
v_str="event0",
),
],
)
],
process=model_pb2.Process(
service_name="svc",
tags=[
model_pb2.KeyValue(
key="key_resource",
v_str="some_resource",
v_type=model_pb2.ValueType.STRING,
)
],
),
),
model_pb2.Span(
operation_name=span_names[1],
trace_id=pb_translator._trace_id_to_bytes(trace_id),
span_id=pb_translator._span_id_to_bytes(parent_id),
start_time=span2_start_time,
duration=span2_duration,
flags=0,
tags=default_tags,
process=model_pb2.Process(service_name="svc",),
),
model_pb2.Span(
operation_name=span_names[2],
trace_id=pb_translator._trace_id_to_bytes(trace_id),
span_id=pb_translator._span_id_to_bytes(other_id),
start_time=span3_start_time,
duration=span3_duration,
flags=0,
tags=[
model_pb2.KeyValue(
key="otel.status_code",
v_type=model_pb2.ValueType.STRING,
v_str="OK",
),
model_pb2.KeyValue(
key="otel.status_description",
v_type=model_pb2.ValueType.STRING,
v_str="Example description",
),
model_pb2.KeyValue(
key="span.kind",
v_type=model_pb2.ValueType.STRING,
v_str="internal",
),
model_pb2.KeyValue(
key="otel.instrumentation_library.name",
v_type=model_pb2.ValueType.STRING,
v_str="name",
),
model_pb2.KeyValue(
key="otel.instrumentation_library.version",
v_type=model_pb2.ValueType.STRING,
v_str="version",
),
],
process=model_pb2.Process(service_name="svc",),
),
]
# events are complicated to compare because order of fields
# (attributes) in otel is not important but in jeager it is
# pylint: disable=no-member
self.assertCountEqual(
spans[0].logs[0].fields, expected_spans[0].logs[0].fields,
)
self.assertEqual(spans, expected_spans)