MessageAssembler.hpp

#pragma once
 
#include <map>
#include <vector>
#include <string>
#include <optional>
#include "message/MessageFrame.hpp"
 
struct CompleteMessage
{
    MessageHeader header;
    std::string payload;
};
 
class MessageAssembler {
private:
    std::map<std::string, std::vector<MessageFrame>> incompleteMessages;
    
public:
    std::optional<std::string> addFragment(const MessageFrame& frame);
    
    bool isMessageComplete(const std::string& messageId) const;
    
    std::optional<std::string> getAssembledMessage(const std::string& messageId);
    
    std::optional<MessageType> getMessageType(const std::string& messageId) const;
    
    void cleanup(const std::string& messageId);
    
    size_t getIncompleteMessageCount() const;
};