websocket

split broadcast out of message function

tools/main.cpp

@@ -38,9 +38,11 @@ int main(void)
     g_web_socket = &web_socket;
 
     std::thread message_processor_thread(std::bind(&Websocket::process_messages, &web_socket));
+    std::thread broadcast_processor_thred(std::bind(&Websocket::process_broadcast, &web_socket));
     
     // Run the asio loop with the main thread
     web_socket.run();
 
     message_processor_thread.join();
+    broadcast_processor_thred.join();
 }

tools/websocket.cpp

@@ -1,7 +1,8 @@
 #include "websocket.hpp"
 
-
-Websocket::Websocket(uint16_t port, std::function<std::string(std::string)> on_message, std::function<std::string()> on_update)
+Websocket::Websocket(uint16_t port,
+                     std::function<std::string(std::string)> on_message,
+                     std::function<std::string()> on_update)
     : m_on_message(on_message), m_on_update(on_update), m_port(port)
 {
     // Set logging settings
@@ -60,7 +61,8 @@ void Websocket::on_close(websocketpp::connection_hdl hdl)
     m_action_cond.notify_one();
 }
 
-void Websocket::on_message(websocketpp::connection_hdl hdl, Server::message_ptr msg) {
+void Websocket::on_message(websocketpp::connection_hdl hdl, Server::message_ptr msg)
+{
     // queue message up for sending by processing thread
     {
         std::lock_guard<std::mutex> guard(m_action_lock);
@@ -70,11 +72,14 @@ void Websocket::on_message(websocketpp::connection_hdl hdl, Server::message_ptr
     m_action_cond.notify_one();
 }
 
-void Websocket::process_messages() {
-    while(1) {
+void Websocket::process_messages()
+{
+    while (1)
+    {
         std::unique_lock<std::mutex> lock(m_action_lock);
 
-        while(m_actions.empty()) {
+        while (m_actions.empty())
+        {
             m_action_cond.wait(lock);
         }
 
@@ -83,8 +88,7 @@ void Websocket::process_messages() {
 
         lock.unlock();
 
-        switch (a.type)
-        {
+        switch (a.type) {
             case SUBSCRIBE:
             {
                 std::lock_guard<std::mutex> guard(m_connection_lock);
@@ -108,13 +112,25 @@ void Websocket::process_messages() {
                 // undefined.
                 break;
         }
+    }
+}
 
-        std::string update = m_on_update();
-        if (!update.empty())
+void Websocket::process_broadcast()
+{
+    while (1)
+    {
         {
             std::lock_guard<std::mutex> guard(m_connection_lock);
-            for (auto it = m_connections.begin(); it != m_connections.end(); ++it)
+            if (m_connections.empty())
             {
+                continue;
+            }
+        }
+        std::string update = m_on_update();
+        if (!update.empty() && update != "{}")
+        {
+            std::lock_guard<std::mutex> guard(m_connection_lock);
+            for (auto it = m_connections.begin(); it != m_connections.end(); ++it) {
                 m_server.send(*it, update, websocketpp::frame::opcode::TEXT);
             }
         }

tools/websocket.hpp

@@ -39,6 +39,7 @@ public:
     void run();
     void stop();
     void process_messages();
+    void process_broadcast();
 private:
     void on_open(websocketpp::connection_hdl hdl);
     void on_close(websocketpp::connection_hdl hdl);