83 lines
2.3 KiB
C++
83 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include <sp/common/DataBuffer.h>
|
|
|
|
namespace sp {
|
|
|
|
template <typename TMessageFactory>
|
|
class SerializableMessage {
|
|
using MessageBaseType = typename TMessageFactory::MessageBaseType;
|
|
using HandlerType = typename MessageBaseType::HandlerType;
|
|
using MessageIdType = typename MessageBaseType::MessageIdType;
|
|
|
|
private:
|
|
std::unique_ptr<MessageBaseType> m_Message;
|
|
|
|
public:
|
|
SerializableMessage(std::unique_ptr<MessageBaseType>&& a_MessagePtr) : m_Message(std::move(a_MessagePtr)) {}
|
|
|
|
operator MessageBaseType&() {
|
|
return *m_Message;
|
|
}
|
|
|
|
SerializableMessage& operator=(std::unique_ptr<MessageBaseType>&& a_MessagePtr) {
|
|
m_Message = std::move(a_MessagePtr);
|
|
return *this;
|
|
}
|
|
|
|
MessageIdType GetId() const {
|
|
return m_Message->GetId();
|
|
}
|
|
|
|
void Dispatch(HandlerType& handler) const {
|
|
m_Message->Dispatch(handler);
|
|
}
|
|
|
|
void Read(DataBuffer& a_Buffer) {
|
|
m_Message->Read(a_Buffer);
|
|
}
|
|
|
|
DataBuffer Write() const {
|
|
return m_Message->Write();
|
|
}
|
|
};
|
|
|
|
template <typename TMessageFactory>
|
|
DataBuffer& operator<<(DataBuffer& a_Buffer, const std::unique_ptr<SerializableMessage<TMessageFactory>>& a_Message) {
|
|
return a_Buffer << VarInt{static_cast<std::uint64_t>(a_Message->GetId())} << a_Message->Write();
|
|
}
|
|
|
|
template <typename TMessageFactory>
|
|
DataBuffer& operator<<(DataBuffer& a_Buffer, const std::shared_ptr<SerializableMessage<TMessageFactory>>& a_Message) {
|
|
return a_Buffer << VarInt{static_cast<std::uint64_t>(a_Message->GetId())} << a_Message->Write();
|
|
}
|
|
|
|
template <typename TMessageFactory>
|
|
DataBuffer& operator>>(DataBuffer& a_Buffer, std::unique_ptr<SerializableMessage<TMessageFactory>>& a_Message) {
|
|
using MsgId = typename TMessageFactory::IdType;
|
|
static TMessageFactory factory;
|
|
|
|
VarInt msgId;
|
|
a_Buffer >> msgId;
|
|
|
|
a_Message = std::make_unique<SerializableMessage<TMessageFactory>>(std::move(factory.CreateMessage(MsgId(msgId.GetValue()))));
|
|
a_Message->Read(a_Buffer);
|
|
|
|
return a_Buffer;
|
|
}
|
|
|
|
template <typename TMessageFactory>
|
|
DataBuffer& operator>>(DataBuffer& a_Buffer, std::shared_ptr<SerializableMessage<TMessageFactory>>& a_Message) {
|
|
using MsgId = typename TMessageFactory::IdType;
|
|
static TMessageFactory factory;
|
|
|
|
VarInt msgId;
|
|
a_Buffer >> msgId;
|
|
|
|
a_Message = std::make_shared<SerializableMessage<TMessageFactory>>(std::move(factory.CreateMessage(MsgId(msgId.GetValue()))));
|
|
a_Message->Read(a_Buffer);
|
|
|
|
return a_Buffer;
|
|
}
|
|
|
|
} // namespace sp
|