37 lines
1.6 KiB
C++
37 lines
1.6 KiB
C++
#pragma once
|
|
|
|
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::RegisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler) {
|
|
auto found = std::find(m_Handlers[a_MessageType].begin(), m_Handlers[a_MessageType].end(), a_Handler);
|
|
if (found == m_Handlers[a_MessageType].end())
|
|
m_Handlers[a_MessageType].push_back(a_Handler);
|
|
}
|
|
|
|
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::UnregisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler) {
|
|
auto found = std::find(m_Handlers[a_MessageType].begin(), m_Handlers[a_MessageType].end(), a_Handler);
|
|
if (found != m_Handlers[a_MessageType].end())
|
|
m_Handlers[a_MessageType].erase(found);
|
|
}
|
|
|
|
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::UnregisterHandler(MessageHandler* a_Handler) {
|
|
for (auto& pair : m_Handlers) {
|
|
if (pair.second.empty())
|
|
continue;
|
|
|
|
MessageIdType type = pair.first;
|
|
|
|
pair.second.erase(std::remove(pair.second.begin(), pair.second.end(), a_Handler), pair.second.end());
|
|
}
|
|
}
|
|
|
|
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::Dispatch(const MessageBase& a_Message) {
|
|
MessageIdType type = a_Message.GetId();
|
|
for (auto& handler : m_Handlers[type]) {
|
|
if (handler)
|
|
a_Message.Dispatch(*handler);
|
|
}
|
|
}
|