1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
/*
* Copyright © 2012 Canonical Ltd.
*
* This program is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* Authored by: Thomas Voß <thomas.voss@canonical.com>
*/
#ifndef CORE_DBUS_MESSAGE_ROUTER_H_
#define CORE_DBUS_MESSAGE_ROUTER_H_
#include <core/dbus/message.h>
#include <functional>
#include <iostream>
#include <memory>
#include <mutex>
#include <unordered_map>
namespace core
{
namespace dbus
{
/**
* @brief Takes a raw DBus message and routes it to a handler.
*/
template<typename Key>
class MessageRouter
{
public:
/**
* @brief Mapper takes a raw DBus Message and maps it to the Key type of the router.
*/
typedef std::function<Key(const Message::Ptr&)> Mapper;
/**
* @brief Handler is a function type that handles raw DBus messages.
*/
typedef std::function<void(const Message::Ptr&)> Handler;
/**
* @brief Constructs an empty router with the specified mapper instance.
* @param m An object of type Mapper.
*/
inline explicit MessageRouter(const Mapper& m) : mapper(m)
{
}
MessageRouter(const MessageRouter&) = delete;
MessageRouter& operator=(const MessageRouter&) = delete;
/**
* @brief Installs a route for a specific key in a thread-safe manner, replacing any previously installed route.
* @param key The key to install the route for.
* @param handler The handler to install, must not be empty.
*/
inline void install_route(const Key& key, Handler handler)
{
std::unique_lock<std::mutex> ul(guard);
router[key] = handler;
}
/**
* @brief Uninstalls a route for a specific key in a thread-safe manner.
* @param key The key to uninstall the route for.
*/
inline void uninstall_route(const Key& key)
{
std::unique_lock<std::mutex> ul(guard);
router.erase(key);
}
/**
* @brief Maps and routes a raw DBus message in a thread-safe manner.
* @param msg The message to map and route, must not be null.
* @return true if the message has been routes successfully, false otherwise.
*/
inline bool operator()(const Message::Ptr& msg)
{
std::unique_lock<std::mutex> ul(guard);
auto it = router.find(mapper(msg));
if (it != router.end()) {
// release the lock so that Handler can modify the Router
Handler handler = it->second;
ul.unlock();
handler(msg);
return true;
}
return false;
}
private:
std::mutex guard;
Mapper mapper;
std::unordered_map<Key,Handler> router;
};
}
}
#endif // CORE_DBUS_MESSAGE_ROUTER_H_
|
