Add websocket-chat-multi example:

This is the multi-threaded io_context version of the CppCon2018 WebSocket chat server and client example.
2019-01-21 12:13:04 -08:00
parent 9a8e22950f
commit 3081f52ad5
18 changed files with 1112 additions and 1 deletions
@@ -0,0 +1,132 @@
+//
+// Copyright (c) 2018 Vinnie Falco (vinnie dot falco at gmail dot com)
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+// Official repository: https://github.com/vinniefalco/CppCon2018
+//
+
+#include "websocket_session.hpp"
+#include <iostream>
+
+websocket_session::
+websocket_session(
+    tcp::socket socket,
+    boost::shared_ptr<shared_state> const& state)
+    : ws_(std::move(socket))
+    , state_(state)
+    , strand_(ws_.get_executor())
+{
+}
+
+websocket_session::
+~websocket_session()
+{
+    // Remove this session from the list of active sessions
+    state_->leave(this);
+}
+
+void
+websocket_session::
+fail(beast::error_code ec, char const* what)
+{
+    // Don't report these
+    if( ec == net::error::operation_aborted ||
+        ec == websocket::error::closed)
+        return;
+
+    std::cerr << what << ": " << ec.message() << "\n";
+}
+
+void
+websocket_session::
+on_accept(beast::error_code ec)
+{
+    // Handle the error, if any
+    if(ec)
+        return fail(ec, "accept");
+
+    // Add this session to the list of active sessions
+    state_->join(this);
+
+    // Read a message
+    ws_.async_read(
+        buffer_,
+        net::bind_executor(strand_,
+            std::bind(
+                &websocket_session::on_read,
+                shared_from_this(),
+                std::placeholders::_1,
+                std::placeholders::_2)));
+}
+
+void
+websocket_session::
+on_read(beast::error_code ec, std::size_t)
+{
+    // Handle the error, if any
+    if(ec)
+        return fail(ec, "read");
+
+    // Send to all connections
+    state_->send(beast::buffers_to_string(buffer_.data()));
+
+    // Clear the buffer
+    buffer_.consume(buffer_.size());
+
+    // Read another message
+    ws_.async_read(
+        buffer_,
+        net::bind_executor(strand_,
+            std::bind(
+                &websocket_session::on_read,
+                shared_from_this(),
+                std::placeholders::_1,
+                std::placeholders::_2)));
+}
+
+void
+websocket_session::
+send(boost::shared_ptr<std::string const> const& ss)
+{
+    // Always add to queue
+    queue_.push_back(ss);
+
+    // Are we already writing?
+    if(queue_.size() > 1)
+        return;
+
+    // We are not currently writing, so send this immediately
+    ws_.async_write(
+        net::buffer(*queue_.front()),
+        net::bind_executor(strand_,
+            std::bind(
+                &websocket_session::on_write,
+                shared_from_this(),
+                std::placeholders::_1,
+                std::placeholders::_2)));
+}
+
+void
+websocket_session::
+on_write(beast::error_code ec, std::size_t)
+{
+    // Handle the error, if any
+    if(ec)
+        return fail(ec, "write");
+
+    // Remove the string from the queue
+    queue_.erase(queue_.begin());
+
+    // Send the next message if any
+    if(! queue_.empty())
+        ws_.async_write(
+            net::buffer(*queue_.front()),
+            net::bind_executor(strand_,
+                std::bind(
+                    &websocket_session::on_write,
+                    shared_from_this(),
+                    std::placeholders::_1,
+                    std::placeholders::_2)));
+}