Add on_send_empty() to websockettpp.

vpndiksw · May 3, 2013 · 2629859 · 2629859
1 parent 2f33b36
commit 2629859
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/connection.hpp b/src/connection.hpp
@@ -1225,7 +1225,9 @@ class connection
                 alog()->at(log::alevel::DEBUG_CLOSE) 
                     << "Exit after inturrupt" << log::endl;
                 terminate(false);
-            }
+            } else {
+		m_handler->on_send_empty(type::shared_from_this());
+	    }
         }
     }
 
@@ -1267,7 +1269,7 @@ class connection
         if (m_write_queue.size() == 0) {
             alog()->at(log::alevel::DEBUG_CLOSE) 
                 << "handle_write called with empty queue" << log::endl;
-            return;
+	    return;
         }
 
         m_write_buffer -= m_write_queue.front()->get_payload().size();

diff --git a/src/roles/server.hpp b/src/roles/server.hpp
@@ -189,6 +189,8 @@ class server {
         virtual void on_pong(connection_ptr con,std::string) {}
         virtual void on_pong_timeout(connection_ptr con,std::string) {}
         virtual void http(connection_ptr con) {}
+
+        virtual void on_send_empty(connection_ptr con) {}
     };
 
     server(boost::asio::io_service& m)