event_data.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
#pragma once
 
#include <azure/core/amqp/models/amqp_message.hpp>
#include <azure/core/amqp/models/amqp_value.hpp>
#include <azure/core/datetime.hpp>
#include <azure/core/nullable.hpp>
 
#include <initializer_list>
#include <iostream>
#include <map>
#include <vector>
 
namespace Azure { namespace Messaging { namespace EventHubs { namespace Models {
 
  class EventData {
  public:
    std::vector<uint8_t> Body;
 
    Azure::Nullable<std::string> ContentType;
 
    Azure::Nullable<Azure::Core::Amqp::Models::AmqpValue> CorrelationId;
 
    Azure::Nullable<Azure::Core::Amqp::Models::AmqpValue> MessageId;
 
    std::map<std::string, Azure::Core::Amqp::Models::AmqpValue> Properties;
 
    EventData() : m_message{nullptr} {};
 
    EventData(std::shared_ptr<Azure::Core::Amqp::Models::AmqpMessage const> const& message);
 
    EventData(std::initializer_list<uint8_t> const& body) : Body(body), m_message{nullptr} {}
 
    EventData(std::vector<uint8_t> const& body) : Body(body), m_message{nullptr} {}
 
    EventData(std::string const& body) : Body(body.begin(), body.end()), m_message{nullptr} {}
 
    virtual ~EventData() = default;
 
    EventData(EventData const&) = default;
 
    EventData& operator=(EventData const&) = default;
 
    EventData(EventData&&) = default;
 
    EventData& operator=(EventData&&) = default;
 
    virtual std::shared_ptr<Azure::Core::Amqp::Models::AmqpMessage const> GetRawAmqpMessage() const;
 
  protected:
    std::shared_ptr<Azure::Core::Amqp::Models::AmqpMessage const> m_message;
  };
  std::ostream& operator<<(std::ostream&, EventData const&);
 
  class ReceivedEventData final : public EventData {
  public:
    Azure::Nullable<Azure::DateTime> EnqueuedTime;
 
    Azure::Nullable<std::uint64_t> Offset;
 
    Azure::Nullable<std::string> PartitionKey;
 
    Azure::Nullable<std::int64_t> SequenceNumber;
 
    std::map<std::string, Azure::Core::Amqp::Models::AmqpValue> SystemProperties;
 
    ReceivedEventData(std::shared_ptr<Azure::Core::Amqp::Models::AmqpMessage const> const& message);
 
    // Destructor
    ~ReceivedEventData() = default;
 
    ReceivedEventData(ReceivedEventData const&) = default;
 
    ReceivedEventData& operator=(ReceivedEventData const&) = default;
 
    ReceivedEventData(ReceivedEventData&&) = default;
 
    ReceivedEventData& operator=(ReceivedEventData&&) = default;
 
    std::shared_ptr<Azure::Core::Amqp::Models::AmqpMessage const> GetRawAmqpMessage() const override
    {
      return m_message;
    }
  };
  std::ostream& operator<<(std::ostream&, ReceivedEventData const&);
}}}} // namespace Azure::Messaging::EventHubs::Models