partition_client.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
#pragma once
#include "eventhubs_exception.hpp"
#include "models/event_data.hpp"
#include "models/partition_client_models.hpp"
 
#include <azure/core/amqp.hpp>
#include <azure/core/amqp/internal/message_receiver.hpp>
#include <azure/core/datetime.hpp>
#include <azure/core/http/policies/policy.hpp>
#include <azure/core/nullable.hpp>
 
namespace Azure { namespace Messaging { namespace EventHubs {
  namespace _detail {
    class PartitionClientFactory;
  }
  struct PartitionClientOptions final
  {
    Models::StartPosition StartPosition;
 
    Azure::Nullable<std::int64_t> OwnerLevel{};
 
    int32_t Prefetch = 300;
  };
 
  class PartitionClient final {
 
  public:
    PartitionClient(PartitionClient const& other) = delete;
    PartitionClient(PartitionClient&& other) = default;
 
    PartitionClient& operator=(PartitionClient const& other) = delete;
    PartitionClient& operator=(PartitionClient&& other) = default;
 
    virtual ~PartitionClient();
 
    std::vector<std::shared_ptr<const Models::ReceivedEventData>> ReceiveEvents(
        uint32_t maxMessages,
        Core::Context const& context = {});
 
    void Close(Core::Context const& context) { m_receiver.Close(context); }
 
  private:
    friend class _detail::PartitionClientFactory;
    Azure::Core::Amqp::_internal::MessageReceiver m_receiver;
 
    PartitionClientOptions m_partitionOptions;
 
    //    std::string m_partitionId;
 
    Azure::Core::Http::Policies::RetryOptions m_retryOptions{};
 
    PartitionClient(
        Azure::Core::Amqp::_internal::MessageReceiver const& messageReceiver,
        PartitionClientOptions options,
        Core::Http::Policies::RetryOptions retryOptions);
 
    std::string GetStartExpression(Models::StartPosition const& startPosition);
  };
}}} // namespace Azure::Messaging::EventHubs