123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700 |
- /* Tells C++ coroutines about Outcome's result
- (C) 2019-2024 Niall Douglas <http://www.nedproductions.biz/> (12 commits)
- File Created: Oct 2019
- Boost Software License - Version 1.0 - August 17th, 2003
- Permission is hereby granted, free of charge, to any person or organization
- obtaining a copy of the software and accompanying documentation covered by
- this license (the "Software") to use, reproduce, display, distribute,
- execute, and transmit the Software, and to prepare derivative works of the
- Software, and to permit third-parties to whom the Software is furnished to
- do so, all subject to the following:
- The copyright notices in the Software and this entire statement, including
- the above license grant, this restriction and the following disclaimer,
- must be included in all copies of the Software, in whole or in part, and
- all derivative works of the Software, unless such copies or derivative
- works are solely in the form of machine-executable object code generated by
- a source language processor.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
- SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
- FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
- ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
- DEALINGS IN THE SOFTWARE.
- */
- #ifndef BOOST_OUTCOME_COROUTINE_SUPPORT_NAMESPACE_BEGIN
- #error This header must only be included by outcome/coroutine_support.hpp or outcome/experimental/coroutine_support.hpp
- #endif
- #ifndef BOOST_OUTCOME_DETAIL_COROUTINE_SUPPORT_HPP
- #define BOOST_OUTCOME_DETAIL_COROUTINE_SUPPORT_HPP
- #include <atomic>
- #include <exception>
- #ifndef BOOST_OUTCOME_COROUTINE_HEADER_TYPE
- #if __has_include(<coroutine>)
- #define BOOST_OUTCOME_COROUTINE_HEADER_TYPE 1
- #elif __has_include(<experimental/coroutine>)
- #define BOOST_OUTCOME_COROUTINE_HEADER_TYPE 2
- #else
- #define BOOST_OUTCOME_COROUTINE_HEADER_TYPE 0
- #endif
- #endif
- #if BOOST_OUTCOME_COROUTINE_HEADER_TYPE && (__cpp_impl_coroutine || (defined(_MSC_VER) && __cpp_coroutines))
- #ifndef BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- #if defined(__has_builtin)
- #if __has_builtin(__builtin_coro_noop) || (!defined(__clang__) && __GNUC__ >= 10)
- #define BOOST_OUTCOME_HAVE_NOOP_COROUTINE 1
- #endif
- #endif
- #endif
- #ifndef BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- #if _MSC_VER >= 1928 || (!defined(__clang__) && __GNUC__ >= 10)
- #define BOOST_OUTCOME_HAVE_NOOP_COROUTINE 1
- #else
- #define BOOST_OUTCOME_HAVE_NOOP_COROUTINE 0
- #endif
- #endif
- #if BOOST_OUTCOME_COROUTINE_HEADER_TYPE == 1
- #include <coroutine>
- BOOST_OUTCOME_V2_NAMESPACE_BEGIN
- namespace awaitables
- {
- template <class Promise = void> using coroutine_handle = std::coroutine_handle<Promise>;
- template <class... Args> using coroutine_traits = std::coroutine_traits<Args...>;
- using std::suspend_always;
- using std::suspend_never;
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- using std::noop_coroutine;
- #endif
- } // namespace awaitables
- BOOST_OUTCOME_V2_NAMESPACE_END
- #define BOOST_OUTCOME_FOUND_COROUTINE_HEADER 1
- #elif BOOST_OUTCOME_COROUTINE_HEADER_TYPE == 2
- #include <experimental/coroutine>
- BOOST_OUTCOME_V2_NAMESPACE_BEGIN
- namespace awaitables
- {
- template <class Promise = void> using coroutine_handle = std::experimental::coroutine_handle<Promise>;
- template <class... Args> using coroutine_traits = std::experimental::coroutine_traits<Args...>;
- using std::experimental::suspend_always;
- using std::experimental::suspend_never;
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- using std::experimental::noop_coroutine;
- #endif
- } // namespace awaitables
- BOOST_OUTCOME_V2_NAMESPACE_END
- #define BOOST_OUTCOME_FOUND_COROUTINE_HEADER 1
- #endif
- #endif
- #ifndef BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER
- // #include <iostream>
- // #define BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(...) std::cout << __VA_ARGS__ << std::endl;
- #define BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(...)
- #endif
- BOOST_OUTCOME_V2_NAMESPACE_EXPORT_BEGIN
- namespace awaitables
- {
- namespace detail
- {
- struct error_type_not_found
- {
- };
- struct exception_type_not_found
- {
- };
- template <class T> struct type_found
- {
- using type = T;
- };
- template <class T, class U = typename T::error_type> constexpr inline type_found<U> extract_error_type(int /*unused*/)
- {
- return {};
- }
- template <class T> constexpr inline type_found<error_type_not_found> extract_error_type(...)
- {
- return {};
- }
- template <class T, class U = typename T::exception_type> constexpr inline type_found<U> extract_exception_type(int /*unused*/)
- {
- return {};
- }
- template <class T> constexpr inline type_found<exception_type_not_found> extract_exception_type(...)
- {
- return {};
- }
- BOOST_OUTCOME_TEMPLATE(class T, class U)
- BOOST_OUTCOME_TREQUIRES(BOOST_OUTCOME_TPRED(BOOST_OUTCOME_V2_NAMESPACE::detail::is_constructible<U, T>))
- inline bool try_set_error(T &&e, U *result)
- {
- new(result) U(static_cast<T &&>(e));
- return true;
- }
- template <class T> inline bool try_set_error(T && /*unused*/, ...)
- {
- return false;
- }
- BOOST_OUTCOME_TEMPLATE(class T, class U)
- BOOST_OUTCOME_TREQUIRES(BOOST_OUTCOME_TPRED(BOOST_OUTCOME_V2_NAMESPACE::detail::is_constructible<U, T>))
- inline void set_or_rethrow(T &e, U *result)
- {
- new(result) U(e);
- }
- template <class T> inline void set_or_rethrow(T &e, ...)
- {
- rethrow_exception(e);
- }
- template <class T> class fake_atomic
- {
- T _v;
- public:
- constexpr fake_atomic(T v)
- : _v(v)
- {
- }
- T load(std::memory_order /*unused*/) { return _v; }
- void store(T v, std::memory_order /*unused*/) { _v = v; }
- bool compare_exchange_strong(T &expected, T v, std::memory_order /*unused*/, std::memory_order /*unused*/)
- {
- if(_v == expected)
- {
- _v = v;
- return true;
- }
- return false;
- }
- };
- #ifdef BOOST_OUTCOME_FOUND_COROUTINE_HEADER
- template <class Awaitable, bool suspend_initial, bool use_atomic, bool is_void> struct outcome_promise_type
- {
- using container_type = typename Awaitable::container_type;
- using result_set_type = std::conditional_t<use_atomic, std::atomic<bool>, fake_atomic<bool>>;
- union
- {
- BOOST_OUTCOME_V2_NAMESPACE::detail::empty_type _default{};
- container_type result;
- };
- result_set_type result_set{false}, pending_first_resumption{is_initially_suspended};
- coroutine_handle<> continuation;
- static constexpr bool is_initially_suspended = suspend_initial;
- static constexpr bool is_using_atomics = use_atomic;
- outcome_promise_type() noexcept { BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise constructed"); }
- outcome_promise_type(const outcome_promise_type &) = delete;
- outcome_promise_type(outcome_promise_type &&) = delete;
- outcome_promise_type &operator=(const outcome_promise_type &) = delete;
- outcome_promise_type &operator=(outcome_promise_type &&) = delete;
- ~outcome_promise_type()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise destructs");
- if(result_set.load(std::memory_order_acquire))
- {
- result.~container_type(); // could throw
- }
- }
- auto get_return_object()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise returns awaitable");
- return Awaitable{*this}; // could throw bad_alloc
- }
- void return_value(container_type &&value)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise returns value");
- BOOST_OUTCOME_ASSERT(!result_set.load(std::memory_order_acquire));
- if(result_set.load(std::memory_order_acquire))
- {
- result.~container_type(); // could throw
- }
- new(&result) container_type(static_cast<container_type &&>(value)); // could throw
- result_set.store(true, std::memory_order_release);
- }
- void return_value(const container_type &value)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise returns value");
- BOOST_OUTCOME_ASSERT(!result_set.load(std::memory_order_acquire));
- if(result_set.load(std::memory_order_acquire))
- {
- result.~container_type(); // could throw
- }
- new(&result) container_type(value); // could throw
- result_set.store(true, std::memory_order_release);
- }
- void unhandled_exception()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise unhandled exception");
- BOOST_OUTCOME_ASSERT(!result_set.load(std::memory_order_acquire));
- if(result_set.load(std::memory_order_acquire))
- {
- result.~container_type();
- }
- #ifndef BOOST_NO_EXCEPTIONS
- auto e = std::current_exception();
- auto ec = detail::error_from_exception(static_cast<decltype(e) &&>(e), {});
- // Try to set error code first
- if(!detail::error_is_set(ec) || !detail::try_set_error(static_cast<decltype(ec) &&>(ec), &result))
- {
- detail::set_or_rethrow(e, &result); // could throw
- }
- #else
- std::terminate();
- #endif
- result_set.store(true, std::memory_order_release);
- }
- auto initial_suspend() noexcept
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise initial suspend = " << suspend_initial);
- struct awaiter
- {
- constexpr bool await_ready() noexcept { return !suspend_initial; }
- void await_resume() noexcept {}
- void await_suspend(coroutine_handle<> /*unused*/) noexcept {}
- };
- return awaiter{};
- }
- auto final_suspend() noexcept
- {
- struct awaiter
- {
- // If we don't force a final suspend, promise will get deleted before awaitable
- // TODO: Implement detachable awaitables
- constexpr bool await_ready() noexcept { return false; }
- void await_resume() noexcept {}
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- coroutine_handle<> await_suspend(coroutine_handle<outcome_promise_type> self) noexcept
- {
- if(self.promise().continuation)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will resume coroutine " << self.promise().continuation.address());
- }
- else
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will exit");
- }
- return self.promise().continuation ? self.promise().continuation : noop_coroutine();
- }
- #else
- void await_suspend(coroutine_handle<outcome_promise_type> self)
- {
- if(self.promise().continuation)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will resume coroutine " << self.promise().continuation.address());
- return self.promise().continuation.resume();
- }
- else
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will exit");
- }
- }
- #endif
- };
- return awaiter{};
- }
- };
- template <class Awaitable, bool suspend_initial, bool use_atomic> struct outcome_promise_type<Awaitable, suspend_initial, use_atomic, true>
- {
- using container_type = void;
- using result_set_type = std::conditional_t<use_atomic, std::atomic<bool>, fake_atomic<bool>>;
- result_set_type result_set{false}, pending_first_resumption{is_initially_suspended};
- coroutine_handle<> continuation;
- static constexpr bool is_initially_suspended = suspend_initial;
- static constexpr bool is_using_atomics = use_atomic;
- outcome_promise_type() { BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise constructed"); }
- outcome_promise_type(const outcome_promise_type &) = delete;
- outcome_promise_type(outcome_promise_type &&) = delete;
- outcome_promise_type &operator=(const outcome_promise_type &) = delete;
- outcome_promise_type &operator=(outcome_promise_type &&) = delete;
- ~outcome_promise_type() = default;
- auto get_return_object()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise returns awaitable");
- return Awaitable{*this}; // could throw bad_alloc
- }
- void return_void() noexcept
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise returns void");
- BOOST_OUTCOME_ASSERT(!result_set.load(std::memory_order_acquire));
- result_set.store(true, std::memory_order_release);
- }
- void unhandled_exception()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise unhandled exception");
- BOOST_OUTCOME_ASSERT(!result_set.load(std::memory_order_acquire));
- std::rethrow_exception(std::current_exception()); // throws
- }
- auto initial_suspend() noexcept
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(this << " promise initial suspend = " << suspend_initial);
- struct awaiter
- {
- constexpr bool await_ready() noexcept { return !suspend_initial; }
- void await_resume() noexcept {}
- void await_suspend(coroutine_handle<> /*unused*/) noexcept {}
- };
- return awaiter{};
- }
- auto final_suspend() noexcept
- {
- struct awaiter
- {
- // If we don't force a final suspend, promise will get deleted before awaitable
- // TODO: Implement detachable awaitables
- constexpr bool await_ready() noexcept { return false; }
- void await_resume() noexcept {}
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- coroutine_handle<> await_suspend(coroutine_handle<outcome_promise_type> self) noexcept
- {
- if(self.promise().continuation)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will resume coroutine " << self.promise().continuation.address());
- }
- else
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will exit");
- }
- return self.promise().continuation ? self.promise().continuation : noop_coroutine();
- }
- #else
- void await_suspend(coroutine_handle<outcome_promise_type> self)
- {
- if(self.promise().continuation)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will resume coroutine " << self.promise().continuation.address());
- return self.promise().continuation.resume();
- }
- else
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&self.promise() << " promise final suspend will exit");
- }
- }
- #endif
- };
- return awaiter{};
- }
- };
- template <class Awaitable, bool suspend_initial, bool use_atomic>
- constexpr inline auto move_result_from_promise_if_not_void(outcome_promise_type<Awaitable, suspend_initial, use_atomic, false> &p)
- {
- return static_cast<typename Awaitable::container_type &&>(p.result);
- }
- template <class Awaitable, bool suspend_initial, bool use_atomic>
- constexpr inline void move_result_from_promise_if_not_void(outcome_promise_type<Awaitable, suspend_initial, use_atomic, true> & /*unused*/)
- {
- }
- template <class Cont, class Executor, bool suspend_initial, bool use_atomic> struct BOOST_OUTCOME_NODISCARD awaitable
- {
- using container_type = Cont;
- using value_type = Cont;
- using executor_type = Executor;
- using promise_type = outcome_promise_type<awaitable, suspend_initial, use_atomic, std::is_void<container_type>::value>;
- coroutine_handle<promise_type> _h;
- awaitable(awaitable &&o) noexcept
- : _h(static_cast<coroutine_handle<promise_type> &&>(o._h))
- {
- o._h = nullptr;
- }
- awaitable(const awaitable &o) = delete;
- awaitable &operator=(awaitable &&) = delete; // as per P1056
- awaitable &operator=(const awaitable &) = delete;
- ~awaitable()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " awaitable destructs");
- if(_h)
- {
- _h.destroy();
- }
- }
- explicit awaitable(promise_type &p) // could throw
- : _h(coroutine_handle<promise_type>::from_promise(p))
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " awaitable constructs for coroutine " << _h.address()
- << " shall resume on first suspend = " << promise_type::is_initially_suspended);
- }
- bool valid() const noexcept { return _h != nullptr; }
- bool await_ready() noexcept
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " await_ready = " << _h.promise().result_set.load(std::memory_order_acquire));
- return _h.promise().result_set.load(std::memory_order_acquire);
- }
- container_type await_resume()
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " await_resume");
- BOOST_OUTCOME_ASSERT(_h.promise().result_set.load(std::memory_order_acquire));
- if(!_h.promise().result_set.load(std::memory_order_acquire))
- {
- std::terminate();
- }
- return detail::move_result_from_promise_if_not_void(_h.promise());
- }
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- coroutine_handle<> await_suspend(coroutine_handle<> cont) noexcept
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " await_suspend suspends coroutine " << cont.address());
- auto &p = _h.promise();
- p.continuation = cont;
- bool expected = true;
- if(p.pending_first_resumption.compare_exchange_strong(expected, false, std::memory_order_acq_rel, std::memory_order_relaxed))
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise()
- << " await_suspend does one time first resumption of initially suspended coroutine " << _h.address());
- return _h;
- }
- return noop_coroutine();
- }
- #else
- void await_suspend(coroutine_handle<> cont)
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise() << " await_suspend suspends coroutine " << cont.address());
- auto &p = _h.promise();
- p.continuation = cont;
- bool expected = true;
- if(p.pending_first_resumption.compare_exchange_strong(expected, false, std::memory_order_acq_rel, std::memory_order_relaxed))
- {
- BOOST_OUTCOME_V2_AWAITABLES_DEBUG_PRINTER(&_h.promise()
- << " await_suspend does one time first resumption of initially suspended coroutine " << _h.address());
- _h.resume();
- }
- }
- #endif
- };
- template <class ContType, class Executor, bool suspend_initial, bool use_atomic> struct generator
- {
- using container_type = ContType;
- using value_type = ContType;
- using executor_type = Executor;
- class promise_type
- {
- friend struct generator;
- using result_set_type = std::conditional_t<use_atomic, std::atomic<int8_t>, fake_atomic<int8_t>>;
- union
- {
- BOOST_OUTCOME_V2_NAMESPACE::detail::empty_type _default{};
- container_type result;
- };
- result_set_type result_set{0};
- coroutine_handle<> continuation;
- public:
- promise_type() {}
- promise_type(const promise_type &) = delete;
- promise_type(promise_type &&) = delete;
- promise_type &operator=(const promise_type &) = delete;
- promise_type &operator=(promise_type &&) = delete;
- ~promise_type()
- {
- if(result_set.load(std::memory_order_acquire) == 1)
- {
- result.~container_type(); // could throw
- }
- }
- auto get_return_object()
- {
- return generator{*this}; // could throw bad_alloc
- }
- void return_void() noexcept
- {
- BOOST_OUTCOME_ASSERT(result_set.load(std::memory_order_acquire) >= 0);
- if(result_set.load(std::memory_order_acquire) == 1)
- {
- result.~container_type(); // could throw
- }
- result_set.store(-1, std::memory_order_release);
- }
- suspend_always yield_value(container_type &&value)
- {
- BOOST_OUTCOME_ASSERT(result_set.load(std::memory_order_acquire) >= 0);
- if(result_set.load(std::memory_order_acquire) == 1)
- {
- result.~container_type(); // could throw
- }
- new(&result) container_type(static_cast<container_type &&>(value)); // could throw
- result_set.store(1, std::memory_order_release);
- return {};
- }
- suspend_always yield_value(const container_type &value)
- {
- BOOST_OUTCOME_ASSERT(result_set.load(std::memory_order_acquire) >= 0);
- if(result_set.load(std::memory_order_acquire) == 1)
- {
- result.~container_type(); // could throw
- }
- new(&result) container_type(value); // could throw
- result_set.store(1, std::memory_order_release);
- return {};
- }
- void unhandled_exception()
- {
- BOOST_OUTCOME_ASSERT(result_set.load(std::memory_order_acquire) >= 0);
- if(result_set.load(std::memory_order_acquire) == 1)
- {
- result.~container_type();
- }
- #ifndef BOOST_NO_EXCEPTIONS
- auto e = std::current_exception();
- auto ec = detail::error_from_exception(static_cast<decltype(e) &&>(e), {});
- // Try to set error code first
- if(!detail::error_is_set(ec) || !detail::try_set_error(static_cast<decltype(ec) &&>(ec), &result))
- {
- detail::set_or_rethrow(e, &result); // could throw
- }
- #else
- std::terminate();
- #endif
- result_set.store(1, std::memory_order_release);
- }
- auto initial_suspend() noexcept
- {
- struct awaiter
- {
- bool await_ready() noexcept { return !suspend_initial; }
- void await_resume() noexcept {}
- void await_suspend(coroutine_handle<> /*unused*/) noexcept {}
- };
- return awaiter{};
- }
- auto final_suspend() noexcept
- {
- struct awaiter
- {
- bool await_ready() noexcept { return false; }
- void await_resume() noexcept {}
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- coroutine_handle<> await_suspend(coroutine_handle<promise_type> self) noexcept
- {
- return self.promise().continuation ? self.promise().continuation : noop_coroutine();
- }
- #else
- void await_suspend(coroutine_handle<promise_type> self)
- {
- if(self.promise().continuation)
- {
- return self.promise().continuation.resume();
- }
- }
- #endif
- };
- return awaiter{};
- }
- };
- coroutine_handle<promise_type> _h;
- generator(generator &&o) noexcept
- : _h(static_cast<coroutine_handle<promise_type> &&>(o._h))
- {
- o._h = nullptr;
- }
- generator(const generator &o) = delete;
- generator &operator=(generator &&) = delete; // as per P1056
- generator &operator=(const generator &) = delete;
- ~generator()
- {
- if(_h)
- {
- _h.destroy();
- }
- }
- explicit generator(promise_type &p) // could throw
- : _h(coroutine_handle<promise_type>::from_promise(p))
- {
- }
- explicit operator bool() const // could throw
- {
- return valid();
- }
- bool valid() const // could throw
- {
- auto &p = _h.promise();
- if(p.result_set.load(std::memory_order_acquire) == 0)
- {
- const_cast<generator *>(this)->_h();
- }
- return p.result_set.load(std::memory_order_acquire) >= 0;
- }
- container_type operator()() // could throw
- {
- auto &p = _h.promise();
- if(p.result_set.load(std::memory_order_acquire) == 0)
- {
- _h();
- }
- BOOST_OUTCOME_ASSERT(p.result_set.load(std::memory_order_acquire) >= 0);
- if(p.result_set.load(std::memory_order_acquire) < 0)
- {
- std::terminate();
- }
- container_type ret(static_cast<container_type &&>(p.result));
- p.result.~container_type(); // could throw
- p.result_set.store(0, std::memory_order_release);
- return ret;
- }
- #if BOOST_OUTCOME_HAVE_NOOP_COROUTINE
- coroutine_handle<> await_suspend(coroutine_handle<> cont) noexcept
- {
- _h.promise().continuation = cont;
- return _h;
- }
- #else
- void await_suspend(coroutine_handle<> cont)
- {
- _h.promise().continuation = cont;
- _h.resume();
- }
- #endif
- };
- #endif
- } // namespace detail
- } // namespace awaitables
- BOOST_OUTCOME_V2_NAMESPACE_END
- #endif
- #ifdef BOOST_OUTCOME_FOUND_COROUTINE_HEADER
- BOOST_OUTCOME_COROUTINE_SUPPORT_NAMESPACE_EXPORT_BEGIN
- /*! AWAITING HUGO JSON CONVERSION TOOL
- SIGNATURE NOT RECOGNISED
- */
- template <class T, class Executor = void> using eager = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable<T, Executor, false, false>;
- /*! AWAITING HUGO JSON CONVERSION TOOL
- SIGNATURE NOT RECOGNISED
- */
- template <class T, class Executor = void> using atomic_eager = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable<T, Executor, false, true>;
- /*! AWAITING HUGO JSON CONVERSION TOOL
- SIGNATURE NOT RECOGNISED
- */
- template <class T, class Executor = void> using lazy = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable<T, Executor, true, false>;
- /*! AWAITING HUGO JSON CONVERSION TOOL
- SIGNATURE NOT RECOGNISED
- */
- template <class T, class Executor = void> using atomic_lazy = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable<T, Executor, true, true>;
- /*! AWAITING HUGO JSON CONVERSION TOOL
- SIGNATURE NOT RECOGNISED
- */
- template <class T, class Executor = void> using generator = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::generator<T, Executor, true, false>;
- BOOST_OUTCOME_COROUTINE_SUPPORT_NAMESPACE_END
- #endif
|