Method RunExchangeEcho

cpp/src/arrow/flight/test_flight_server.cc:314–336

Echo the client's messages back.

// Echo the client's messages back.
Status TestFlightServer::RunExchangeEcho(std::unique_ptr<FlightMessageReader> reader,
                                         std::unique_ptr<FlightMessageWriter> writer) {
  FlightStreamChunk chunk;
  bool begun = false;
  while (true) {
    ARROW_ASSIGN_OR_RAISE(chunk, reader->Next());
    if (!chunk.data && !chunk.app_metadata) {
      break;
    }
    if (!begun && chunk.data) {
      begun = true;
      RETURN_NOT_OK(writer->Begin(chunk.data->schema()));
    }
    if (chunk.data && chunk.app_metadata) {
      RETURN_NOT_OK(writer->WriteWithMetadata(*chunk.data, chunk.app_metadata));
    } else if (chunk.data) {
      RETURN_NOT_OK(writer->WriteRecordBatch(*chunk.data));
    } else if (chunk.app_metadata) {
      RETURN_NOT_OK(writer->WriteMetadata(chunk.app_metadata));
    }
  }
  return Status::OK();
}

// Regression test for ARROW-13253
Status TestFlightServer::RunExchangeLargeBatch(