Skip to content

Commit ada006a

Browse files
committed
Added span attributes to tracer middleware
1 parent 43e6df8 commit ada006a

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/workflows/transport/middleware/tracer.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,12 @@ def wrapped_callback(header, message):
7272
with self.tracer.start_as_current_span(self.service_name, context=ctx) as span:
7373
if ctx == {}:
7474
self._inject_trace_context(message)
75+
76+
# Insert header and message info:
77+
for key, value in header.items():
78+
span.set_attribute(str(key), str(value))
79+
for key, value in message.items():
80+
span.set_attribute(str(key), str(value))
7581
return callback(header, message)
7682

7783
return call_next(channel, wrapped_callback, **kwargs)

0 commit comments

Comments
 (0)