Added Connection::send convenience function

eidheim · Jul 28, 2017 · 5a0c95a · 5a0c95a
1 parent 9a8d9e5
commit 5a0c95a
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/client_ws.hpp b/client_ws.hpp
@@ -181,6 +181,13 @@ namespace SimpleWeb {
         });
       }
 
+      /// Convenience function for sending an std::string message
+      void send(const std::string &message, const std::function<void(const error_code &)> &callback = nullptr, unsigned char fin_rsv_opcode = 129) {
+        auto send_stream = std::make_shared<SendStream>();
+        send_stream->write(message.c_str(), message.size());
+        send(send_stream, callback, fin_rsv_opcode);
+      }
+
       void send_close(int status, const std::string &reason = "", const std::function<void(const error_code &)> &callback = nullptr) {
         // Send close only once (in case close is initiated by client)
         if(closed)

diff --git a/server_ws.hpp b/server_ws.hpp
@@ -250,6 +250,13 @@ namespace SimpleWeb {
         });
       }
 
+      /// Convenience function for sending an std::string message
+      void send(const std::string &message, const std::function<void(const error_code &)> &callback = nullptr, unsigned char fin_rsv_opcode = 129) {
+        auto send_stream = std::make_shared<SendStream>();
+        send_stream->write(message.c_str(), message.size());
+        send(send_stream, callback, fin_rsv_opcode);
+      }
+
       void send_close(int status, const std::string &reason = "", const std::function<void(const error_code &)> &callback = nullptr) {
         // Send close only once (in case close is initiated by server)
         if(closed)

diff --git a/ws_examples.cpp b/ws_examples.cpp
@@ -40,6 +40,9 @@ int main() {
             "Error: " << ec << ", error message: " << ec.message() << endl;
       }
     });
+
+    // Alternatively, using a convenience function:
+    // connection->send(message_str, [](const SimpleWeb::error_code & /*ec*/) { /*handle error*/ });
   };
 
   echo.on_open = [](shared_ptr<WsServer::Connection> connection) {

diff --git a/wss_examples.cpp b/wss_examples.cpp
@@ -40,6 +40,9 @@ int main() {
             "Error: " << ec << ", error message: " << ec.message() << endl;
       }
     });
+
+    // Alternatively, using a convenience function:
+    // connection->send(message_str, [](const SimpleWeb::error_code & /*ec*/) { /*handle error*/ });
   };
 
   echo.on_open = [](shared_ptr<WssServer::Connection> connection) {