amqp_message.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include "amqp_header.hpp"
#include "amqp_properties.hpp"
#include "amqp_value.hpp"
 
#include <azure/core/nullable.hpp>
 
#include <map>
#include <vector>
 
namespace Azure { namespace Core { namespace Amqp { namespace Models { namespace _detail {
  class AmqpMessageFactory;
}}}}} // namespace Azure::Core::Amqp::Models::_detail
 
namespace Azure { namespace Core { namespace Amqp { namespace Models {
  enum class MessageBodyType
  {
    Invalid,
    None,
    Data,
    Sequence,
    Value,
  };
 
  constexpr int AmqpDefaultMessageFormatValue
      = 0; // Specifies the message format for an AMQP message.
 
  class AmqpMessage final {
  public:
    AmqpMessage() = default;
 
    ~AmqpMessage() = default;
 
    AmqpMessage(AmqpMessage&&) = default;
 
    AmqpMessage(AmqpMessage const&) = default;
 
    AmqpMessage& operator=(AmqpMessage const&) = default;
 
    AmqpMessage& operator=(AmqpMessage&&) = default;
 
    bool operator==(AmqpMessage const& that) const noexcept;
 
    bool operator!=(AmqpMessage const& that) const noexcept { return !(*this == that); }
 
    AmqpMessage(std::nullptr_t) : m_hasValue{false} {}
 
    operator bool() const noexcept { return m_hasValue; }
 
    uint32_t MessageFormat = AmqpDefaultMessageFormatValue;
 
    MessageHeader Header;
 
    AmqpMap DeliveryAnnotations;
 
    AmqpMap MessageAnnotations;
 
    MessageProperties Properties;
 
    std::map<std::string, AmqpValue> ApplicationProperties;
 
    AmqpValue DeliveryTag;
 
    AmqpMap Footer;
 
    MessageBodyType BodyType{MessageBodyType::None};
 
    void SetBody(std::vector<AmqpList> const& bodySequence);
 
    void SetBody(AmqpList const& bodySequence);
 
    void SetBody(std::vector<AmqpBinaryData> const& bodyBinarySequence);
 
    void SetBody(AmqpBinaryData const& bodyBinary);
 
    void SetBody(AmqpValue const& bodyValue);
 
    std::vector<AmqpList> const& GetBodyAsAmqpList() const;
 
    AmqpValue const& GetBodyAsAmqpValue() const;
 
    std::vector<AmqpBinaryData> const& GetBodyAsBinary() const;
 
    static std::vector<uint8_t> Serialize(AmqpMessage const& message);
 
    static AmqpMessage Deserialize(std::uint8_t const* buffer, size_t size);
 
    friend class _detail::AmqpMessageFactory;
 
  private:
    std::vector<AmqpBinaryData> m_binaryDataBody;
    std::vector<AmqpList> m_amqpSequenceBody;
    AmqpValue m_amqpValueBody;
    bool m_hasValue{true}; // By default, an AmqpMessage has a value.
  };
  std::ostream& operator<<(std::ostream&, AmqpMessage const&);
}}}} // namespace Azure::Core::Amqp::Models