Added new projects of new RPC core: MyClient and MtServivce

2026-07-28 02:57:56 +00:00 · 2024-01-04 18:35:15 +02:00
parent 2a13699900
commit feb0192b11
24 changed files with 2078 additions and 0 deletions
@@ -0,0 +1,164 @@
+#include "pch.h"
+#include "MtConnection.h"
+
+#include <iostream>
+
+MtConnection::MtConnection(WebsocketStream&& ws)
+    : ws_(std::move(ws))
+    , resolver_(ws_.get_executor())
+{
+    log_.Debug("%s: connection %p created.", __FUNCTION__, this);
+}
+
+MtConnection::~MtConnection()
+{
+    log_.Debug("%s: connection %p destroyed.", __FUNCTION__, this);
+}
+
+void MtConnection::Accept()
+{
+    // Set suggested timeout settings for the websocket
+    ws_.set_option(
+        boost::beast::websocket::stream_base::timeout::suggested(
+            boost::beast::role_type::server));
+
+    // Set a decorator to change the Server of the handshake
+    ws_.set_option(boost::beast::websocket::stream_base::decorator(
+        [](boost::beast::websocket::response_type& res)
+        {
+        res.set(boost::beast::http::field::server,
+                std::string(BOOST_BEAST_VERSION_STRING) +
+                    " websocket-server-async");
+    }));
+    // Accept the websocket handshake
+    ws_.async_accept(
+        std::bind(
+            &MtConnection::OnAccept,
+            shared_from_this(),
+            std::placeholders::_1));
+}
+
+void MtConnection::Close()
+{
+    log_.Debug("%s: entry.", __FUNCTION__);
+
+    ws_.async_close(boost::beast::websocket::close_code::normal,
+            [](boost::beast::error_code const ec){ });
+}
+
+void MtConnection::Send(const std::string& msg)
+{
+    log_.Trace("%s: msg = %s", __FUNCTION__, msg.c_str());
+
+    send_queue_.push(msg);
+    if (send_queue_.size() == 1)
+        DoWrite();
+}
+
+void MtConnection::OnAccept(boost::beast::error_code ec)
+{
+    log_.Trace("%s: entry.", __FUNCTION__);
+
+    if (ec)
+    {
+        log_.Error("%s: %s", __FUNCTION__, ec.what().c_str());
+        OnConnectionFailed(ec.message());
+    }
+    else
+        DoRead();
+
+    OnConnected();
+}
+
+void MtConnection::DoRead()
+{
+    log_.Trace("%s: enter.", __FUNCTION__);
+
+    // Clear the buffer
+    read_text_.clear();
+
+    // Read a message into our buffer
+    ws_.async_read(
+        read_buffer_,
+        std::bind(
+            &MtConnection::OnRead,
+            shared_from_this(),
+            std::placeholders::_1,
+            std::placeholders::_2));
+}
+
+void MtConnection::OnRead(
+    boost::beast::error_code ec,
+    std::size_t bytes_transferred)
+{
+    boost::ignore_unused(bytes_transferred);
+
+    // This indicates that the session was closed
+    if (ec == boost::beast::websocket::error::closed)
+    {
+        log_.Info("%s: session was closed.", __FUNCTION__);
+        OnDisconnected();
+        return;
+    }
+
+    if (ec == boost::asio::error::operation_aborted)
+    {
+        log_.Info("%s: session was aborted.", __FUNCTION__);
+        return;
+    }
+
+    if (ec)
+    {
+        log_.Error("%s: %s", __FUNCTION__, ec.message().c_str());
+        OnConnectionFailed(ec.message());
+    }
+    else
+    {
+        auto msg = boost::beast::buffers_to_string(read_buffer_.data());
+        log_.Trace("%s: %s", __FUNCTION__, msg.c_str());
+
+        OnMessageReceived(msg);
+
+        // Clear the buffer
+        read_buffer_.consume(read_buffer_.size());
+
+        boost::asio::dispatch(ws_.get_executor(),
+                              std::bind(
+                                  &MtConnection::DoRead,
+                                  shared_from_this()));
+    }
+}
+
+void MtConnection::DoWrite()
+{
+    if (send_queue_.empty())
+        return;
+
+    log_.Trace("%s: msg = %s", __FUNCTION__, send_queue_.front().c_str());
+
+    ws_.text(ws_.got_text());
+    ws_.async_write(
+        boost::asio::buffer(send_queue_.front()),
+        std::bind(
+            &MtConnection::OnWrite,
+            shared_from_this(),
+            std::placeholders::_1,
+            std::placeholders::_2));
+}
+
+void MtConnection::OnWrite(
+    boost::beast::error_code ec,
+    std::size_t bytes_transferred)
+{
+    boost::ignore_unused(bytes_transferred);
+
+    if (ec)
+    {
+        log_.Error("%s, %s", __FUNCTION__, ec.what().c_str());
+        return;
+    }
+
+    send_queue_.pop();
+    DoWrite();
+}
+