mirror of https://github.com/SeanOMik/libki.git
protocol: Packet class definition
This commit is contained in:
parent
c08eb617bf
commit
c9e485fa86
|
@ -0,0 +1,56 @@
|
||||||
|
#pragma once
|
||||||
|
#include "../util/Serializable.h"
|
||||||
|
#include <cstdint>
|
||||||
|
#include <vector>
|
||||||
|
#include <sstream>
|
||||||
|
#include <type_traits>
|
||||||
|
|
||||||
|
namespace ki
|
||||||
|
{
|
||||||
|
namespace protocol
|
||||||
|
{
|
||||||
|
class Packet : public util::Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
Packet(bool control = false, uint8_t opcode = 0);
|
||||||
|
virtual ~Packet() = default;
|
||||||
|
|
||||||
|
bool is_control() const;
|
||||||
|
void set_control(bool control);
|
||||||
|
|
||||||
|
uint8_t get_opcode() const;
|
||||||
|
void set_opcode(uint8_t opcode);
|
||||||
|
|
||||||
|
template <typename DataT>
|
||||||
|
void set_payload_data(const DataT &data)
|
||||||
|
{
|
||||||
|
static_assert(std::is_base_of<Serializable, DataT>::value,
|
||||||
|
"DataT must derive from Serializable.");
|
||||||
|
|
||||||
|
std::ostringstream oss;
|
||||||
|
data.write_to(oss);
|
||||||
|
m_payload = std::vector<char>(oss.str().c_str(), data.get_size());
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename DataT>
|
||||||
|
DataT *get_payload_data() const
|
||||||
|
{
|
||||||
|
static_assert(std::is_base_of<Serializable, DataT>::value,
|
||||||
|
"DataT must derive from Serializable.");
|
||||||
|
|
||||||
|
std::istringstream iss(m_payload.data());
|
||||||
|
DataT *data = new DataT();
|
||||||
|
data->read_from(iss);
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual void write_to(std::ostream &ostream) const final;
|
||||||
|
virtual void read_from(std::istream &istream) final;
|
||||||
|
virtual size_t get_size() const final;
|
||||||
|
private:
|
||||||
|
bool m_control;
|
||||||
|
uint8_t m_opcode;
|
||||||
|
std::vector<char> m_payload;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue