diff options
-rw-r--r-- | include/erebos/message.h | 10 | ||||
-rw-r--r-- | include/erebos/network.h | 11 | ||||
-rw-r--r-- | src/message.cpp | 12 | ||||
-rw-r--r-- | src/message.h | 4 | ||||
-rw-r--r-- | src/network.cpp | 10 |
5 files changed, 36 insertions, 11 deletions
diff --git a/include/erebos/message.h b/include/erebos/message.h index c94e8c3..66d221c 100644 --- a/include/erebos/message.h +++ b/include/erebos/message.h @@ -4,11 +4,14 @@ #include <chrono> #include <functional> +#include <memory> #include <optional> #include <string> namespace erebos { +using std::unique_ptr; + class Identity; class DirectMessage @@ -78,13 +81,14 @@ public: void handle(Context &) const override; typedef std::function<void(const DirectMessageThread &, ssize_t, ssize_t)> ThreadWatcher; - static void onUpdate(ThreadWatcher); - static DirectMessageThread thread(const Identity &); + void onUpdate(ThreadWatcher); + DirectMessageThread thread(const Identity &); - static DirectMessage send(const Identity &, const Peer &, const std::string &); + DirectMessage send(const Identity &, const Peer &, const std::string &); private: struct Priv; + unique_ptr<Priv> p; }; } diff --git a/include/erebos/network.h b/include/erebos/network.h index d730fb5..a2f989e 100644 --- a/include/erebos/network.h +++ b/include/erebos/network.h @@ -4,6 +4,7 @@ #include <erebos/service.h> #include <functional> +#include <typeinfo> namespace erebos { @@ -13,14 +14,24 @@ public: Server(const Identity &, std::vector<std::unique_ptr<Service>> &&); ~Server(); + template<class S> S & svc(); + class PeerList & peerList() const; struct Peer; private: + Service & svcHelper(const std::type_info &); + struct Priv; const std::shared_ptr<Priv> p; }; +template<class S> +S & Server::svc() +{ + return dynamic_cast<S&>(svcHelper(typeid(S))); +} + class Peer { public: diff --git a/src/message.cpp b/src/message.cpp index 354703e..601bc39 100644 --- a/src/message.cpp +++ b/src/message.cpp @@ -202,10 +202,10 @@ const Identity & DirectMessageThread::peer() const } -vector<DirectMessageService::ThreadWatcher> DirectMessageService::Priv::watchers; -mutex DirectMessageService::Priv::watcherLock; +DirectMessageService::DirectMessageService(): + p(new Priv) +{} -DirectMessageService::DirectMessageService() = default; DirectMessageService::~DirectMessageService() = default; UUID DirectMessageService::uuid() const @@ -229,14 +229,14 @@ void DirectMessageService::handle(Context & ctx) const lock.unlock(); - for (const auto & w : Priv::watchers) + for (const auto & w : p->watchers) w(dmt, -1, -1); } void DirectMessageService::onUpdate(ThreadWatcher w) { - scoped_lock l(Priv::watcherLock); - Priv::watchers.push_back(w); + scoped_lock l(p->watcherLock); + p->watchers.push_back(w); } DirectMessageThread DirectMessageService::thread(const Identity & peer) diff --git a/src/message.h b/src/message.h index c4a3549..efdec8f 100644 --- a/src/message.h +++ b/src/message.h @@ -51,8 +51,8 @@ struct DirectMessageThread::Iterator::Priv struct DirectMessageService::Priv { - static vector<ThreadWatcher> watchers; - static mutex watcherLock; + vector<ThreadWatcher> watchers; + mutex watcherLock; }; } diff --git a/src/network.cpp b/src/network.cpp index 91f1923..4b79dcb 100644 --- a/src/network.cpp +++ b/src/network.cpp @@ -6,6 +6,7 @@ #include <algorithm> #include <cstring> #include <iostream> +#include <stdexcept> #include <arpa/inet.h> #include <ifaddrs.h> @@ -13,6 +14,7 @@ #include <unistd.h> using std::holds_alternative; +using std::runtime_error; using std::scoped_lock; using std::to_string; using std::unique_lock; @@ -26,6 +28,14 @@ Server::Server(const Identity & self, vector<unique_ptr<Service>> && svcs): Server::~Server() = default; +Service & Server::svcHelper(const std::type_info & tinfo) +{ + for (auto & s : p->services) + if (typeid(*s) == tinfo) + return *s; + throw runtime_error("service not found"); +} + PeerList & Server::peerList() const { return p->plist; |