update boost

linx64/include/boost/mysql/impl/internal/channel/write_message.hpp

@@ -0,0 +1,97 @@
+//
+// Copyright (c) 2019-2023 Ruben Perez Hidalgo (rubenperez038 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)
+//
+
+#ifndef BOOST_MYSQL_IMPL_INTERNAL_CHANNEL_WRITE_MESSAGE_HPP
+#define BOOST_MYSQL_IMPL_INTERNAL_CHANNEL_WRITE_MESSAGE_HPP
+
+#include <boost/mysql/error_code.hpp>
+
+#include <boost/mysql/detail/any_stream.hpp>
+
+#include <boost/mysql/impl/internal/channel/message_writer.hpp>
+#include <boost/mysql/impl/internal/protocol/constants.hpp>
+
+#include <boost/asio/async_result.hpp>
+#include <boost/asio/buffer.hpp>
+#include <boost/asio/compose.hpp>
+#include <boost/asio/coroutine.hpp>
+
+#include <cstddef>
+#include <cstdint>
+
+namespace boost {
+namespace mysql {
+namespace detail {
+
+// Writes an entire message to stream; partitions the message into
+// chunks and adds the required headers
+inline void write_message(any_stream& stream, message_writer& processor, error_code& ec)
+{
+    while (!processor.done())
+    {
+        std::size_t bytes_written = stream.write_some(asio::buffer(processor.next_chunk()), ec);
+        if (ec)
+            break;
+        processor.on_bytes_written(bytes_written);
+    }
+}
+
+struct write_message_op : boost::asio::coroutine
+{
+    any_stream& stream_;
+    message_writer& processor_;
+
+    write_message_op(any_stream& stream, message_writer& processor) noexcept
+        : stream_(stream), processor_(processor)
+    {
+    }
+
+    template <class Self>
+    void operator()(Self& self, error_code ec = {}, std::size_t bytes_written = 0)
+    {
+        // Error handling
+        if (ec)
+        {
+            self.complete(ec);
+            return;
+        }
+
+        // Non-error path
+        BOOST_ASIO_CORO_REENTER(*this)
+        {
+            // done() never returns false after a call to prepare_buffer(), so no post() needed
+            BOOST_ASSERT(!processor_.done());
+            while (!processor_.done())
+            {
+                BOOST_ASIO_CORO_YIELD stream_.async_write_some(
+                    asio::buffer(processor_.next_chunk()),
+                    std::move(self)
+                );
+                processor_.on_bytes_written(bytes_written);
+            };
+
+            self.complete(error_code());
+        }
+    }
+};
+
+template <BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code)) CompletionToken>
+BOOST_ASIO_INITFN_AUTO_RESULT_TYPE(CompletionToken, void(boost::mysql::error_code))
+async_write_message(any_stream& stream, message_writer& processor, CompletionToken&& token)
+{
+    return boost::asio::async_compose<CompletionToken, void(error_code)>(
+        write_message_op(stream, processor),
+        token,
+        stream
+    );
+}
+
+}  // namespace detail
+}  // namespace mysql
+}  // namespace boost
+
+#endif