Method DoAction

cpp/src/arrow/flight/flight_test.cc:1684–1716

class TracingTestServer : public FlightServerBase {
 public:
  Status DoAction(const ServerCallContext& call_context, const Action&,
                  std::unique_ptr<ResultStream>* result) override {
    std::vector<Result> results;
    auto* middleware =
        reinterpret_cast<TracingServerMiddleware*>(call_context.GetMiddleware("tracing"));
    if (!middleware) return Status::Invalid("Could not find middleware");
// When running with OTel, ASAN reports false-positives that can't be easily suppressed.
// Disable OTel for ASAN. See GH-46509.
#if defined(ARROW_WITH_OPENTELEMETRY) && !defined(ADDRESS_SANITIZER)
    // Ensure the trace context is present (but the value is random so
    // we cannot assert any particular value)
    EXPECT_FALSE(middleware->GetTraceContext().empty());
    auto span = arrow::internal::tracing::GetTracer()->GetCurrentSpan();
    const auto context = span->GetContext();
    {
      const auto& span_id = context.span_id();
      ARROW_ASSIGN_OR_RAISE(auto buffer, AllocateBuffer(span_id.Id().size()));
      std::memcpy(buffer->mutable_data(), span_id.Id().data(), span_id.Id().size());
      results.push_back({std::move(buffer)});
    }
    {
      const auto& trace_id = context.trace_id();
      ARROW_ASSIGN_OR_RAISE(auto buffer, AllocateBuffer(trace_id.Id().size()));
      std::memcpy(buffer->mutable_data(), trace_id.Id().data(), trace_id.Id().size());
      results.push_back({std::move(buffer)});
    }
#else
    // Ensure the trace context is not present (as OpenTelemetry is not enabled)
    EXPECT_TRUE(middleware->GetTraceContext().empty());
#endif
    *result = std::make_unique<SimpleResultStream>(std::move(results));
    return Status::OK();
  }
};

class TestTracing : public ::testing::Test {