mirror of
https://github.com/XRPLF/rippled.git
synced 2026-04-29 15:37:57 +00:00
initial work on broadcast_server example
This commit is contained in:
100
examples/broadcast_server/broadcast_server.cpp
Normal file
100
examples/broadcast_server/broadcast_server.cpp
Normal file
@@ -0,0 +1,100 @@
|
||||
#include <websocketpp/config/asio_no_tls.hpp>
|
||||
|
||||
#include <websocketpp/server.hpp>
|
||||
|
||||
#include <iostream>
|
||||
|
||||
typedef websocketpp::server<websocketpp::config::asio> server;
|
||||
|
||||
using websocketpp::lib::placeholders::_1;
|
||||
using websocketpp::lib::placeholders::_2;
|
||||
using websocketpp::lib::bind;
|
||||
|
||||
/* on_open insert connection_hdl into channel
|
||||
* on_close remove connection_hdl from channel
|
||||
* on_message queue send to all channels
|
||||
*/
|
||||
|
||||
class broadcast_server {
|
||||
public:
|
||||
broadcast_server() {
|
||||
// Initialize Asio Transport
|
||||
m_server.init_asio();
|
||||
|
||||
// Register handler callbacks
|
||||
m_server.set_open_handler(bind(&on_open,this,::_1));
|
||||
m_server.set_close_handler(bind(&on_message,this,::_1));
|
||||
m_server.set_message_handler(bind(&on_message,this,::_1,::_2));
|
||||
}
|
||||
|
||||
void run(uint16_t port) {
|
||||
// listen on specified port
|
||||
m_server.listen(port);
|
||||
|
||||
// Start the server accept loop
|
||||
echo_server.start_accept();
|
||||
|
||||
// Start the ASIO io_service run loop
|
||||
try {
|
||||
echo_server.run();
|
||||
} catch (const std::exception & e) {
|
||||
std::cout << e.what() << std::endl;
|
||||
} catch (websocketpp::lib::error_code e) {
|
||||
std::cout << e.message() << std::endl;
|
||||
} catch (...) {
|
||||
std::cout << "other exception" << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
void on_open(connection_hdl hdl) {
|
||||
boost::unique_lock<boost::mutex> lock(m_lock);
|
||||
m_connections.insert(hdl);
|
||||
}
|
||||
|
||||
void on_close(connection_hdl hdl) {
|
||||
boost::unique_lock<boost::mutex> lock(m_lock);
|
||||
m_connections.remove(hdl);
|
||||
}
|
||||
|
||||
void on_message(connection_hdl hdl, server::message_ptr msg) {
|
||||
// queue message up for sending by processing thread
|
||||
boost::unique_lock<boost::mutex> lock(m_lock);
|
||||
m_message_queue.push(msg);
|
||||
lock.unlock();
|
||||
m_cond.notify_one();
|
||||
}
|
||||
|
||||
void process_messages() {
|
||||
while(1) {
|
||||
boost::unique_lock<boost::mutex> lock(m_lock);
|
||||
|
||||
while(m_message_queue.empty()) {
|
||||
m_cond.wait(m_lock);
|
||||
}
|
||||
|
||||
message_ptr msg = m_message_queue.front();
|
||||
m_message_queue.pop();
|
||||
|
||||
lock.unlock();
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
private:
|
||||
server m_server;
|
||||
std::set<connection_hdl> m_connections;
|
||||
std::queue<server::message_ptr> m_message_queue;
|
||||
|
||||
boost::mutex m_mutex;
|
||||
boost::condition_variable m_cond;
|
||||
}
|
||||
|
||||
int main() {
|
||||
broadcast_server server;
|
||||
|
||||
// Start a thread to run the processing loop
|
||||
boost::thread(bind(&broadcast_server::process_messages,&server));
|
||||
|
||||
// Run the asio loop with the main thread
|
||||
server.run(9002);
|
||||
}
|
||||
Reference in New Issue
Block a user