PacketHandlers.h

Go to the documentation of this file.

 
#pragma once
#include "IoTypes.h"
#include "PacketPayload.h"
#include "catapult/utils/NonCopyable.h"
#include "catapult/functions.h"
#include "catapult/types.h"
#include <vector>
 
namespace catapult { namespace ionet {
 
    struct ServerPacketHandlerContext : utils::MoveOnly {
    public:
        explicit ServerPacketHandlerContext(const Key& key, const std::string& host);
 
    public:
        const Key& key() const;
 
        const std::string& host() const;
 
        bool hasResponse() const;
 
        const PacketPayload& response() const;
 
    public:
        void response(PacketPayload&& payload);
 
    private:
        const Key& m_key;
        const std::string& m_host;
        bool m_hasResponse;
        PacketPayload m_payload;
    };
 
    class ServerPacketHandlers {
    public:
        using ContextType = ServerPacketHandlerContext;
 
        using PacketHandler = consumer<const Packet&, ContextType&>;
 
    public:
        explicit ServerPacketHandlers(uint32_t maxPacketDataSize = std::numeric_limits<uint32_t>::max());
 
    public:
        size_t size() const;
 
        uint32_t maxPacketDataSize() const;
 
        bool canProcess(PacketType type) const;
 
        bool canProcess(const Packet& packet) const;
 
        bool process(const Packet& packet, ContextType& context) const;
 
    public:
        void registerHandler(PacketType type, const PacketHandler& handler);
 
    private:
        const PacketHandler* findHandler(const Packet& packet) const;
 
    private:
        uint32_t m_maxPacketDataSize;
        std::vector<PacketHandler> m_handlers;
    };
}}