Method DoPut

cpp/src/arrow/flight/test_definitions.cc:649–691

class DoPutTestServer : public FlightServerBase {
 public:
  Status DoPut(const ServerCallContext& context,
               std::unique_ptr<FlightMessageReader> reader,
               std::unique_ptr<FlightMetadataWriter> writer) override {
    descriptor_ = reader->descriptor();

    if (descriptor_.type == FlightDescriptor::DescriptorType::CMD) {
      if (descriptor_.cmd == "TestUndrained") {
        // Don't read all the messages
        return Status::OK();
      }
    }

    int counter = 0;
    FlightStreamChunk chunk;
    while (true) {
      ARROW_ASSIGN_OR_RAISE(chunk, reader->Next());
      if (!chunk.data) break;
      if (counter % 2 == 1) {
        if (!chunk.app_metadata) {
          return Status::Invalid("Expected app_metadata");
        } else if (chunk.app_metadata->ToString() != std::to_string(counter)) {
          return Status::Invalid("Expected app_metadata to be ", counter, " but got ",
                                 chunk.app_metadata->ToString());
        }
      } else if (chunk.app_metadata) {
        return Status::Invalid("Expected no app_metadata");
      }
      batches_.push_back(std::move(chunk.data));
      auto buffer = Buffer::FromString(std::to_string(counter));
      RETURN_NOT_OK(writer->WriteMetadata(*buffer));
      counter++;
    }

    // Expect a metadata-only message
    if (!chunk.app_metadata) {
      return Status::Invalid("Expected app_metadata at end of stream (#1)");
    } else if (chunk.app_metadata->ToString() != kExpectedMetadata) {
      return Status::Invalid("Expected app_metadata to be ", kExpectedMetadata,
                             " but got ", chunk.app_metadata->ToString());
    }

    return Status::OK();
  }

 protected:
  FlightDescriptor descriptor_;