67 lines
1.5 KiB
C++
67 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <functional>
|
|
#include <iostream>
|
|
#include <memory>
|
|
|
|
namespace sp {
|
|
|
|
namespace details {
|
|
|
|
template <typename TBase>
|
|
using ArrayType = std::vector<std::function<std::unique_ptr<TBase>(void)>>;
|
|
|
|
|
|
|
|
template <typename TBase, typename... TMessages>
|
|
struct ArrayFiller {};
|
|
|
|
template <typename TBase, typename... TMessages>
|
|
struct ArrayFiller<TBase, std::tuple<TMessages...>> {
|
|
static ArrayType<TBase> ArrayCreate() {
|
|
ArrayType<TBase> array;
|
|
array.reserve(sizeof...(TMessages));
|
|
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
|
return array;
|
|
}
|
|
};
|
|
|
|
template <typename TBase, typename TMessage, typename... TMessages>
|
|
struct ArrayFiller<TBase, TMessage, TMessages...> {
|
|
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
|
ArrayFiller<TBase, TMessage>::ArrayAppend(array);
|
|
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
|
}
|
|
};
|
|
|
|
template <typename TBase, typename TMessage>
|
|
struct ArrayFiller<TBase, TMessage> {
|
|
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
|
array.push_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
|
|
}
|
|
};
|
|
|
|
} // namespace details
|
|
|
|
template <typename TBase, typename TTMessages>
|
|
class MessageFactory {
|
|
public:
|
|
using IdType = typename TBase::MsgIdType;
|
|
|
|
MessageFactory() : m_Factory(details::ArrayFiller<TBase, TTMessages>::ArrayCreate()) {}
|
|
|
|
std::unique_ptr<TBase> CreateMessage(IdType id) {
|
|
if (id >= m_Factory.size())
|
|
return nullptr;
|
|
return m_Factory.at(id)();
|
|
}
|
|
|
|
private:
|
|
details::ArrayType<TBase> m_Factory;
|
|
};
|
|
|
|
|
|
|
|
} // namespace sp
|