lidavidm commented on code in PR #11920:
URL: https://github.com/apache/arrow/pull/11920#discussion_r887303083
##########
python/pyarrow/tests/test_flight.py:
##########
@@ -2177,3 +2177,32 @@ def test_interpreter_shutdown():
See https://issues.apache.org/jira/browse/ARROW-16597.
"""
util.invoke_script("arrow_16597.py")
+
+
+class TracingFlightServer(FlightServerBase):
+ """A server that echoes back trace context values."""
+
+ def do_action(self, context, action):
+ trace_context = context.get_middleware("tracing").trace_context
+ # Don't turn this method into a generator since then it'll be
+ # lazily evaluated, and the trace context will be lost
Review Comment:
I've updated the comment to be clearer about what goes on.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
