EncryptionContext.h

#pragma once
 
#include <cstdint>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "CompressionType.h"
#include "defines.h"
 
namespace pulsar {
 
namespace proto {
class MessageMetadata;
}
 
struct PULSAR_PUBLIC EncryptionKey {
    std::string key;
    std::string value;
    std::unordered_map<std::string, std::string> metadata;
 
    EncryptionKey(const std::string& key, const std::string& value,
                  const decltype(EncryptionKey::metadata)& metadata)
        : key(key), value(value), metadata(metadata) {}
};

class PULSAR_PUBLIC EncryptionContext {
   public:
    using KeysType = std::vector<EncryptionKey>;

    const KeysType& keys() const noexcept { return keys_; }

    const std::string& param() const noexcept { return param_; }

    const std::string& algorithm() const noexcept { return algorithm_; }

    CompressionType compressionType() const noexcept { return compressionType_; }

    uint32_t uncompressedMessageSize() const noexcept { return uncompressedMessageSize_; }

    int32_t batchSize() const noexcept { return batchSize_; }

    bool isDecryptionFailed() const noexcept { return isDecryptionFailed_; }

    EncryptionContext(const proto::MessageMetadata&, bool);
 
   private:
    KeysType keys_;
    std::string param_;
    std::string algorithm_;
    CompressionType compressionType_{CompressionNone};
    uint32_t uncompressedMessageSize_{0};
    int32_t batchSize_{-1};
    bool isDecryptionFailed_{false};
 
    void setDecryptionFailed(bool failed) noexcept { isDecryptionFailed_ = failed; }
 
    friend class ConsumerImpl;
};
 
}  // namespace pulsar