1   
//
1   
//
2   
// Copyright (c) 2026 Steve Gerbino
2   
// Copyright (c) 2026 Steve Gerbino
3   
//
3   
//
4   
// Distributed under the Boost Software License, Version 1.0. (See accompanying
4   
// Distributed under the Boost Software License, Version 1.0. (See accompanying
5   
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
5   
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6   
//
6   
//
7   
// Official repository: https://github.com/cppalliance/corosio
7   
// Official repository: https://github.com/cppalliance/corosio
8   
//
8   
//
9  
9  
10   
#ifndef BOOST_COROSIO_IO_IO_WRITE_STREAM_HPP
10   
#ifndef BOOST_COROSIO_IO_IO_WRITE_STREAM_HPP
11   
#define BOOST_COROSIO_IO_IO_WRITE_STREAM_HPP
11   
#define BOOST_COROSIO_IO_IO_WRITE_STREAM_HPP
12  
12  
13   
#include <boost/corosio/detail/config.hpp>
13   
#include <boost/corosio/detail/config.hpp>
14   
#include <boost/corosio/io/io_object.hpp>
14   
#include <boost/corosio/io/io_object.hpp>
15   
#include <boost/corosio/io_buffer_param.hpp>
15   
#include <boost/corosio/io_buffer_param.hpp>
16   
#include <boost/capy/io_result.hpp>
16   
#include <boost/capy/io_result.hpp>
17   
#include <boost/capy/ex/executor_ref.hpp>
17   
#include <boost/capy/ex/executor_ref.hpp>
18   
#include <boost/capy/ex/io_env.hpp>
18   
#include <boost/capy/ex/io_env.hpp>
19  
19  
20   
#include <coroutine>
20   
#include <coroutine>
21   
#include <cstddef>
21   
#include <cstddef>
22   
#include <stop_token>
22   
#include <stop_token>
23   
#include <system_error>
23   
#include <system_error>
24  
24  
25   
namespace boost::corosio {
25   
namespace boost::corosio {
26  
26  
27   
/** Abstract base for streams that support async writes.
27   
/** Abstract base for streams that support async writes.
28  
28  
29   
    Provides the `write_some` operation via a pure virtual
29   
    Provides the `write_some` operation via a pure virtual
30   
    `do_write_some` dispatch point. Concrete classes override
30   
    `do_write_some` dispatch point. Concrete classes override
31   
    `do_write_some` to route through their implementation.
31   
    `do_write_some` to route through their implementation.
32  
32  
33   
    Uses virtual inheritance from @ref io_object so that
33   
    Uses virtual inheritance from @ref io_object so that
34   
    @ref io_stream can combine this with @ref io_read_stream
34   
    @ref io_stream can combine this with @ref io_read_stream
35   
    without duplicating the `io_object` base.
35   
    without duplicating the `io_object` base.
36  
36  
37   
    @par Thread Safety
37   
    @par Thread Safety
38   
    Distinct objects: Safe.
38   
    Distinct objects: Safe.
39   
    Shared objects: Unsafe.
39   
    Shared objects: Unsafe.
40  
40  
41   
    @see io_read_stream, io_stream, io_object
41   
    @see io_read_stream, io_stream, io_object
42   
*/
42   
*/
43   
class BOOST_COROSIO_DECL io_write_stream : virtual public io_object
43   
class BOOST_COROSIO_DECL io_write_stream : virtual public io_object
44   
{
44   
{
45   
protected:
45   
protected:
46   
    /// Awaitable for async write operations.
46   
    /// Awaitable for async write operations.
47   
    template<class ConstBufferSequence>
47   
    template<class ConstBufferSequence>
48   
    struct write_some_awaitable
48   
    struct write_some_awaitable
49   
    {
49   
    {
50   
        io_write_stream& ios_;
50   
        io_write_stream& ios_;
51   
        ConstBufferSequence buffers_;
51   
        ConstBufferSequence buffers_;
52   
        std::stop_token token_;
52   
        std::stop_token token_;
53   
        mutable std::error_code ec_;
53   
        mutable std::error_code ec_;
54   
        mutable std::size_t bytes_transferred_ = 0;
54   
        mutable std::size_t bytes_transferred_ = 0;
55  
55  
56   
        write_some_awaitable(
56   
        write_some_awaitable(
57   
            io_write_stream& ios, ConstBufferSequence buffers) noexcept
57   
            io_write_stream& ios, ConstBufferSequence buffers) noexcept
58   
            : ios_(ios)
58   
            : ios_(ios)
59   
            , buffers_(std::move(buffers))
59   
            , buffers_(std::move(buffers))
60   
        {
60   
        {
61   
        }
61   
        }
62  
62  
63   
        bool await_ready() const noexcept
63   
        bool await_ready() const noexcept
64   
        {
64   
        {
65   
            return token_.stop_requested();
65   
            return token_.stop_requested();
66   
        }
66   
        }
67  
67  
68   
        capy::io_result<std::size_t> await_resume() const noexcept
68   
        capy::io_result<std::size_t> await_resume() const noexcept
69   
        {
69   
        {
70   
            if (token_.stop_requested())
70   
            if (token_.stop_requested())
71   
                return {make_error_code(std::errc::operation_canceled), 0};
71   
                return {make_error_code(std::errc::operation_canceled), 0};
72   
            return {ec_, bytes_transferred_};
72   
            return {ec_, bytes_transferred_};
73   
        }
73   
        }
74  
74  
75   
        auto await_suspend(std::coroutine_handle<> h, capy::io_env const* env)
75   
        auto await_suspend(std::coroutine_handle<> h, capy::io_env const* env)
76   
            -> std::coroutine_handle<>
76   
            -> std::coroutine_handle<>
77   
        {
77   
        {
78   
            token_ = env->stop_token;
78   
            token_ = env->stop_token;
79   
            return ios_.do_write_some(
79   
            return ios_.do_write_some(
80   
                h, env->executor, buffers_, token_, &ec_, &bytes_transferred_);
80   
                h, env->executor, buffers_, token_, &ec_, &bytes_transferred_);
81   
        }
81   
        }
82   
    };
82   
    };
83  
83  
84   
    /** Dispatch a write through the concrete implementation.
84   
    /** Dispatch a write through the concrete implementation.
85  
85  
86   
        @param h Coroutine handle to resume on completion.
86   
        @param h Coroutine handle to resume on completion.
87   
        @param ex Executor for dispatching the completion.
87   
        @param ex Executor for dispatching the completion.
88   
        @param buffers Source buffer sequence.
88   
        @param buffers Source buffer sequence.
89   
        @param token Stop token for cancellation.
89   
        @param token Stop token for cancellation.
90   
        @param ec Output error code.
90   
        @param ec Output error code.
91   
        @param bytes Output bytes transferred.
91   
        @param bytes Output bytes transferred.
92  
92  
93   
        @return Coroutine handle to resume immediately.
93   
        @return Coroutine handle to resume immediately.
94   
    */
94   
    */
95   
    virtual std::coroutine_handle<> do_write_some(
95   
    virtual std::coroutine_handle<> do_write_some(
96   
        std::coroutine_handle<>,
96   
        std::coroutine_handle<>,
97   
        capy::executor_ref,
97   
        capy::executor_ref,
98   
        io_buffer_param,
98   
        io_buffer_param,
99   
        std::stop_token,
99   
        std::stop_token,
100   
        std::error_code*,
100   
        std::error_code*,
101   
        std::size_t*) = 0;
101   
        std::size_t*) = 0;
102  
102  
103   
    io_write_stream() noexcept = default;
103   
    io_write_stream() noexcept = default;
104  
104  
105   
    /// Construct from a handle.
105   
    /// Construct from a handle.
106   
    explicit io_write_stream(handle h) noexcept : io_object(std::move(h)) {}
106   
    explicit io_write_stream(handle h) noexcept : io_object(std::move(h)) {}
107  
107  
108   
public:
108   
public:
109   
    /** Asynchronously write data to the stream.
109   
    /** Asynchronously write data to the stream.
110  
110  
111   
        Suspends the calling coroutine and initiates a kernel-level
111   
        Suspends the calling coroutine and initiates a kernel-level
112   
        write. The coroutine resumes when at least one byte is written,
112   
        write. The coroutine resumes when at least one byte is written,
113   
        an error occurs, or the operation is cancelled.
113   
        an error occurs, or the operation is cancelled.
114  
114  
115   
        @param buffers The buffer sequence containing data to write.
115   
        @param buffers The buffer sequence containing data to write.
116  
116  
117   
        @return An awaitable yielding `(error_code, std::size_t)`.
117   
        @return An awaitable yielding `(error_code, std::size_t)`.
118  
118  
119   
        @see io_stream::read_some
119   
        @see io_stream::read_some
120   
    */
120   
    */
121   
    template<capy::ConstBufferSequence CB>
121   
    template<capy::ConstBufferSequence CB>
122   
    auto write_some(CB const& buffers)
122   
    auto write_some(CB const& buffers)
123   
    {
123   
    {
124   
        return write_some_awaitable<CB>(*this, buffers);
124   
        return write_some_awaitable<CB>(*this, buffers);
125   
    }
125   
    }
126   
};
126   
};
127  
127  
128   
} // namespace boost::corosio
128   
} // namespace boost::corosio
129  
129  
130   
#endif
130   
#endif