A server that echoes back trace context values.
| 2617 | |
| 2618 | |
| 2619 | class TracingFlightServer(FlightServerBase): |
| 2620 | """A server that echoes back trace context values.""" |
| 2621 | |
| 2622 | def do_action(self, context, action): |
| 2623 | trace_context = context.get_middleware("tracing").trace_context |
| 2624 | # Don't turn this method into a generator since then |
| 2625 | # trace_context will be evaluated after we've exited the scope |
| 2626 | # of the OTel span (and so the value we want won't be present) |
| 2627 | return ((f"{key}: {value}").encode("utf-8") |
| 2628 | for (key, value) in trace_context.items()) |
| 2629 | |
| 2630 | |
| 2631 | def test_tracing(): |