gdax-orderbook-hpp/demo/dependencies/websocketpp-0.7.0/examples/simple_broadcast_server/simple_broadcast_server.cpp

   1 #include <set>
   2 #include <websocketpp/config/asio_no_tls.hpp>
   3 #include <websocketpp/server.hpp>
   4
   5 typedef websocketpp::server<websocketpp::config::asio> server;
   6
   7 using websocketpp::connection_hdl;
   8 using websocketpp::lib::placeholders::_1;
   9 using websocketpp::lib::placeholders::_2;
  10 using websocketpp::lib::bind;
  11
  12 class broadcast_server {
  13 public:
  14     broadcast_server() {
  15         m_server.init_asio();
  16
  17         m_server.set_open_handler(bind(&broadcast_server::on_open,this,::_1));
  18         m_server.set_close_handler(bind(&broadcast_server::on_close,this,::_1));
  19         m_server.set_message_handler(bind(&broadcast_server::on_message,this,::_1,::_2));
  20     }
  21
  22     void on_open(connection_hdl hdl) {
  23         m_connections.insert(hdl);
  24     }
  25
  26     void on_close(connection_hdl hdl) {
  27         m_connections.erase(hdl);
  28     }
  29
  30     void on_message(connection_hdl hdl, server::message_ptr msg) {
  31         for (auto it : m_connections) {
  32             m_server.send(it,msg);
  33         }
  34     }
  35
  36     void run(uint16_t port) {
  37         m_server.listen(port);
  38         m_server.start_accept();
  39         m_server.run();
  40     }
  41 private:
  42     typedef std::set<connection_hdl,std::owner_less<connection_hdl>> con_list;
  43
  44     server m_server;
  45     con_list m_connections;
  46 };
  47
  48 int main() {
  49     broadcast_server server;
  50     server.run(9002);
  51 }