mirror of https://github.com/PurpleI2P/i2pd.git
orignal
8 years ago
committed by
GitHub
23 changed files with 408 additions and 40 deletions
@ -0,0 +1,30 @@ |
|||||||
|
#include "Event.h" |
||||||
|
#include "Log.h" |
||||||
|
|
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace event |
||||||
|
{ |
||||||
|
EventCore core; |
||||||
|
|
||||||
|
void EventCore::SetListener(EventListener * l) |
||||||
|
{ |
||||||
|
m_listener = l; |
||||||
|
LogPrint(eLogInfo, "Event: listener set"); |
||||||
|
} |
||||||
|
|
||||||
|
void EventCore::QueueEvent(const EventType & ev) |
||||||
|
{ |
||||||
|
if(m_listener) |
||||||
|
m_listener->HandleEvent(ev); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void EmitEvent(const EventType & e) |
||||||
|
{ |
||||||
|
#ifdef WITH_EVENTS |
||||||
|
i2p::event::core.QueueEvent(e); |
||||||
|
#endif |
||||||
|
} |
||||||
|
|
||||||
@ -0,0 +1,36 @@ |
|||||||
|
#ifndef EVENT_H__ |
||||||
|
#define EVENT_H__ |
||||||
|
#include <map> |
||||||
|
#include <string> |
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <boost/asio.hpp> |
||||||
|
|
||||||
|
typedef std::map<std::string, std::string> EventType; |
||||||
|
|
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace event |
||||||
|
{ |
||||||
|
class EventListener { |
||||||
|
public: |
||||||
|
virtual ~EventListener() {}; |
||||||
|
virtual void HandleEvent(const EventType & ev) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
class EventCore |
||||||
|
{ |
||||||
|
public: |
||||||
|
void QueueEvent(const EventType & ev); |
||||||
|
void SetListener(EventListener * l); |
||||||
|
|
||||||
|
private: |
||||||
|
EventListener * m_listener = nullptr; |
||||||
|
}; |
||||||
|
|
||||||
|
extern EventCore core; |
||||||
|
} |
||||||
|
} |
||||||
|
void EmitEvent(const EventType & ev); |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,137 @@ |
|||||||
|
#include "Websocket.h" |
||||||
|
#include "Log.h" |
||||||
|
|
||||||
|
#include <set> |
||||||
|
|
||||||
|
#include <websocketpp/config/asio_no_tls.hpp> |
||||||
|
#include <websocketpp/server.hpp> |
||||||
|
#include <boost/property_tree/ini_parser.hpp> |
||||||
|
#define GCC47_BOOST149 ((BOOST_VERSION == 104900) && (__GNUC__ == 4) && (__GNUC_MINOR__ >= 7)) |
||||||
|
#if !GCC47_BOOST149 |
||||||
|
#include <boost/property_tree/json_parser.hpp> |
||||||
|
#endif |
||||||
|
|
||||||
|
#include <stdexcept> |
||||||
|
|
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace event |
||||||
|
{ |
||||||
|
|
||||||
|
typedef websocketpp::server<websocketpp::config::asio> ServerImpl; |
||||||
|
typedef websocketpp::connection_hdl ServerConn; |
||||||
|
|
||||||
|
class WebsocketServerImpl : public EventListener |
||||||
|
{ |
||||||
|
private: |
||||||
|
typedef ServerImpl::message_ptr MessagePtr; |
||||||
|
public: |
||||||
|
|
||||||
|
WebsocketServerImpl(const std::string & addr, int port) : m_run(false), m_thread(nullptr) |
||||||
|
{ |
||||||
|
m_server.init_asio(); |
||||||
|
m_server.set_open_handler(std::bind(&WebsocketServerImpl::ConnOpened, this, std::placeholders::_1)); |
||||||
|
m_server.set_close_handler(std::bind(&WebsocketServerImpl::ConnClosed, this, std::placeholders::_1)); |
||||||
|
m_server.set_message_handler(std::bind(&WebsocketServerImpl::OnConnMessage, this, std::placeholders::_1, std::placeholders::_2)); |
||||||
|
|
||||||
|
m_server.listen(boost::asio::ip::address::from_string(addr), port); |
||||||
|
} |
||||||
|
|
||||||
|
~WebsocketServerImpl() |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
void Start() { |
||||||
|
m_run = true; |
||||||
|
m_server.start_accept(); |
||||||
|
m_thread = new std::thread([&] () { |
||||||
|
while(m_run) { |
||||||
|
try { |
||||||
|
m_server.run(); |
||||||
|
} catch (std::exception & e ) { |
||||||
|
LogPrint(eLogError, "Websocket server: ", e.what()); |
||||||
|
} |
||||||
|
} |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
void Stop() { |
||||||
|
m_run = false; |
||||||
|
m_server.stop(); |
||||||
|
if(m_thread) { |
||||||
|
m_thread->join(); |
||||||
|
delete m_thread; |
||||||
|
} |
||||||
|
m_thread = nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
void ConnOpened(ServerConn c) |
||||||
|
{ |
||||||
|
std::lock_guard<std::mutex> lock(m_connsMutex); |
||||||
|
m_conns.insert(c); |
||||||
|
} |
||||||
|
|
||||||
|
void ConnClosed(ServerConn c) |
||||||
|
{ |
||||||
|
std::lock_guard<std::mutex> lock(m_connsMutex); |
||||||
|
m_conns.erase(c); |
||||||
|
} |
||||||
|
|
||||||
|
void OnConnMessage(ServerConn conn, ServerImpl::message_ptr msg) |
||||||
|
{ |
||||||
|
(void) conn; |
||||||
|
(void) msg; |
||||||
|
} |
||||||
|
|
||||||
|
void HandleEvent(const EventType & ev) |
||||||
|
{ |
||||||
|
std::lock_guard<std::mutex> lock(m_connsMutex); |
||||||
|
LogPrint(eLogDebug, "websocket event"); |
||||||
|
boost::property_tree::ptree event; |
||||||
|
for (const auto & item : ev) { |
||||||
|
event.put(item.first, item.second); |
||||||
|
} |
||||||
|
std::ostringstream ss; |
||||||
|
write_json(ss, event); |
||||||
|
std::string s = ss.str(); |
||||||
|
|
||||||
|
ConnList::iterator it; |
||||||
|
for (it = m_conns.begin(); it != m_conns.end(); ++it) { |
||||||
|
ServerImpl::connection_ptr con = m_server.get_con_from_hdl(*it); |
||||||
|
con->send(s); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
typedef std::set<ServerConn, std::owner_less<ServerConn> > ConnList; |
||||||
|
bool m_run; |
||||||
|
std::thread * m_thread; |
||||||
|
std::mutex m_connsMutex; |
||||||
|
ConnList m_conns; |
||||||
|
ServerImpl m_server; |
||||||
|
}; |
||||||
|
|
||||||
|
|
||||||
|
WebsocketServer::WebsocketServer(const std::string & addr, int port) : m_impl(new WebsocketServerImpl(addr, port)) {} |
||||||
|
WebsocketServer::~WebsocketServer() |
||||||
|
{ |
||||||
|
delete m_impl; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
void WebsocketServer::Start() |
||||||
|
{ |
||||||
|
m_impl->Start(); |
||||||
|
} |
||||||
|
|
||||||
|
void WebsocketServer::Stop() |
||||||
|
{ |
||||||
|
m_impl->Stop(); |
||||||
|
} |
||||||
|
|
||||||
|
EventListener * WebsocketServer::ToListener() |
||||||
|
{ |
||||||
|
return m_impl; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,28 @@ |
|||||||
|
#ifndef WEBSOCKET_H__ |
||||||
|
#define WEBSOCKET_H__ |
||||||
|
#include "Event.h" |
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace event |
||||||
|
{ |
||||||
|
|
||||||
|
class WebsocketServerImpl; |
||||||
|
|
||||||
|
class WebsocketServer |
||||||
|
{ |
||||||
|
public: |
||||||
|
WebsocketServer(const std::string & addr, int port); |
||||||
|
~WebsocketServer(); |
||||||
|
|
||||||
|
void Start(); |
||||||
|
void Stop(); |
||||||
|
|
||||||
|
EventListener * ToListener(); |
||||||
|
|
||||||
|
private: |
||||||
|
WebsocketServerImpl * m_impl; |
||||||
|
}; |
||||||
|
|
||||||
|
} |
||||||
|
} |
||||||
|
#endif |
||||||
Loading…
Reference in new issue