text: // Copyright 2022 Google LLC
  //
  // Licensed under the Apache License, Version 2.0 (the "License");
  // you may not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //      https://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing, software
  // distributed under the License is distributed on an "AS IS" BASIS,
  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  // See the License for the specific language governing permissions and
  // limitations under the License.
  
  // Generated by the Codegen C++ plugin.
  // If you make any local changes, they will be lost.
  // source: google/cloud/bigquery/storage/v1/storage.proto
  
  #ifndef GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_INTERNAL_BIGQUERY_WRITE_STUB_H
  #define GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_INTERNAL_BIGQUERY_WRITE_STUB_H
  
  #include "google/cloud/async_streaming_read_write_rpc.h"
  #include "google/cloud/completion_queue.h"
  #include "google/cloud/status_or.h"
  #include "google/cloud/version.h"
  #include <google/cloud/bigquery/storage/v1/storage.grpc.pb.h>
  #include <memory>
  
  namespace google {
  namespace cloud {
  namespace bigquery_internal {
  GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
  
  class BigQueryWriteStub {
   public:
    virtual ~BigQueryWriteStub() = 0;
  
    virtual StatusOr<google::cloud::bigquery::storage::v1::WriteStream>
    CreateWriteStream(
        grpc::ClientContext& context,
        google::cloud::bigquery::storage::v1::CreateWriteStreamRequest const&
            request) = 0;
  
    virtual std::unique_ptr<::google::cloud::AsyncStreamingReadWriteRpc<
        google::cloud::bigquery::storage::v1::AppendRowsRequest,
        google::cloud::bigquery::storage::v1::AppendRowsResponse>>
    AsyncAppendRows(google::cloud::CompletionQueue const& cq,
                    std::unique_ptr<grpc::ClientContext> context) = 0;
  
    virtual StatusOr<google::cloud::bigquery::storage::v1::WriteStream>
    GetWriteStream(
        grpc::ClientContext& context,
        google::cloud::bigquery::storage::v1::GetWriteStreamRequest const&
            request) = 0;
  
    virtual StatusOr<
        google::cloud::bigquery::storage::v1::FinalizeWriteStreamResponse>
    FinalizeWriteStream(
        grpc::ClientContext& context,
        google::cloud::bigquery::storage::v1::FinalizeWriteStreamRequest const&
            request) = 0;
  
    virtual StatusOr<
        google::cloud::bigquery::storage::v1::BatchCommitWriteStreamsResponse>
    BatchCommitWriteStreams(
        grpc::ClientContext& context,
        google::cloud::bigquery::storage::v1::
            BatchCommitWriteStreamsRequest const& request) = 0;
  
    virtual StatusOr<google::cloud::bigquery::storage::v1::FlushRowsResponse>
    FlushRows(grpc::ClientContext& context,
              google::cloud::bigquery::storage::v1::FlushRowsRequest const&
                  request) = 0;
  };
  
  class DefaultBigQueryWriteStub : public BigQueryWriteStub {
   public:
    explicit DefaultBigQueryWriteStub(
        std::unique_ptr<
            google::cloud::bigquery::storage::v1::BigQueryWrite::StubInterface>
            grpc_stub)
        : grpc_stub_(std::move(grpc_stub)) {}
  
    StatusOr<google::cloud::bigquery::storage::v1::WriteStream> CreateWriteStream(
        grpc::ClientContext& client_context,
        google::cloud::bigquery::storage::v1::CreateWriteStreamRequest const&
            request) override;
  
    std::unique_ptr<::google::cloud::AsyncStreamingReadWriteRpc<
        google::cloud::bigquery::storage::v1::AppendRowsRequest,
        google::cloud::bigquery::storage::v1::AppendRowsResponse>>
    AsyncAppendRows(google::cloud::CompletionQueue const& cq,
                    std::unique_ptr<grpc::ClientContext> context) override;
  
    StatusOr<google::cloud::bigquery::storage::v1::WriteStream> GetWriteStream(
        grpc::ClientContext& client_context,
        google::cloud::bigquery::storage::v1::GetWriteStreamRequest const&
            request) override;
  
    StatusOr<google::cloud::bigquery::storage::v1::FinalizeWriteStreamResponse>
    FinalizeWriteStream(
        grpc::ClientContext& client_context,
        google::cloud::bigquery::storage::v1::FinalizeWriteStreamRequest const&
            request) override;
  
    StatusOr<
        google::cloud::bigquery::storage::v1::BatchCommitWriteStreamsResponse>
    BatchCommitWriteStreams(
        grpc::ClientContext& client_context,
        google::cloud::bigquery::storage::v1::
            BatchCommitWriteStreamsRequest const& request) override;
  
    StatusOr<google::cloud::bigquery::storage::v1::FlushRowsResponse> FlushRows(
        grpc::ClientContext& client_context,
        google::cloud::bigquery::storage::v1::FlushRowsRequest const& request)
        override;
  
   private:
    std::unique_ptr<
        google::cloud::bigquery::storage::v1::BigQueryWrite::StubInterface>
        grpc_stub_;
  };
  
  GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
  }  // namespace bigquery_internal
  }  // namespace cloud
  }  // namespace google
  
  #endif  // GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_INTERNAL_BIGQUERY_WRITE_STUB_H
  