packets_handler.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <unordered_map>
 
#include "network/tarcap/packets_handlers/latest/common/packet_handler.hpp"
 
namespace taraxa::network::tarcap {
 
class PacketsHandler {
 public:
  const std::shared_ptr<PacketHandler>& getSpecificHandler(SubprotocolPacketType packet_type) const;
 
  template <typename PacketHandlerType>
  std::shared_ptr<PacketHandlerType> getSpecificHandler();
 
  template <typename PacketHandlerType, typename... Args>
  void registerHandler(Args&&... args);
 
 private:
  // Map of all packets handlers, factory method selects specific packet handler for processing based on packet type
  std::unordered_map<SubprotocolPacketType, std::shared_ptr<PacketHandler>> packets_handlers_;
};
 
template <typename PacketHandlerType>
std::shared_ptr<PacketHandlerType> PacketsHandler::getSpecificHandler() {
  return std::static_pointer_cast<PacketHandlerType>(getSpecificHandler(PacketHandlerType::kPacketType_));
}
 
template <typename PacketHandlerType, typename... Args>
void PacketsHandler::registerHandler(Args&&... args) {
  assert(packets_handlers_.find(PacketHandlerType::kPacketType_) == packets_handlers_.end());
  packets_handlers_.emplace(PacketHandlerType::kPacketType_,
                            std::make_shared<PacketHandlerType>(std::forward<Args>(args)...));
}
 
}  // namespace taraxa::network::tarcap