/* Tells C++ coroutines about Outcome's result
(C) 2019-2024 Niall Douglas (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
#include
#ifndef BOOST_OUTCOME_COROUTINE_HEADER_TYPE
#if __has_include()
#define BOOST_OUTCOME_COROUTINE_HEADER_TYPE 1
#elif __has_include()
#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
BOOST_OUTCOME_V2_NAMESPACE_BEGIN
namespace awaitables
{
template using coroutine_handle = std::coroutine_handle;
template using coroutine_traits = std::coroutine_traits;
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
BOOST_OUTCOME_V2_NAMESPACE_BEGIN
namespace awaitables
{
template using coroutine_handle = std::experimental::coroutine_handle;
template using coroutine_traits = std::experimental::coroutine_traits;
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
// #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 struct type_found
{
using type = T;
};
template constexpr inline type_found extract_error_type(int /*unused*/)
{
return {};
}
template constexpr inline type_found extract_error_type(...)
{
return {};
}
template constexpr inline type_found extract_exception_type(int /*unused*/)
{
return {};
}
template constexpr inline type_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))
inline bool try_set_error(T &&e, U *result)
{
new(result) U(static_cast(e));
return true;
}
template 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))
inline void set_or_rethrow(T &e, U *result)
{
new(result) U(e);
}
template inline void set_or_rethrow(T &e, ...)
{
rethrow_exception(e);
}
template 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 struct outcome_promise_type
{
using container_type = typename Awaitable::container_type;
using result_set_type = std::conditional_t, fake_atomic>;
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(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(e), {});
// Try to set error code first
if(!detail::error_is_set(ec) || !detail::try_set_error(static_cast(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 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 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 struct outcome_promise_type
{
using container_type = void;
using result_set_type = std::conditional_t, fake_atomic>;
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 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 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
constexpr inline auto move_result_from_promise_if_not_void(outcome_promise_type &p)
{
return static_cast(p.result);
}
template
constexpr inline void move_result_from_promise_if_not_void(outcome_promise_type & /*unused*/)
{
}
template struct BOOST_OUTCOME_NODISCARD awaitable
{
using container_type = Cont;
using value_type = Cont;
using executor_type = Executor;
using promise_type = outcome_promise_type::value>;
coroutine_handle _h;
awaitable(awaitable &&o) noexcept
: _h(static_cast &&>(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::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 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, fake_atomic>;
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(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(e), {});
// Try to set error code first
if(!detail::error_is_set(ec) || !detail::try_set_error(static_cast(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 self) noexcept
{
return self.promise().continuation ? self.promise().continuation : noop_coroutine();
}
#else
void await_suspend(coroutine_handle self)
{
if(self.promise().continuation)
{
return self.promise().continuation.resume();
}
}
#endif
};
return awaiter{};
}
};
coroutine_handle _h;
generator(generator &&o) noexcept
: _h(static_cast &&>(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::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(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(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 using eager = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable;
/*! AWAITING HUGO JSON CONVERSION TOOL
SIGNATURE NOT RECOGNISED
*/
template using atomic_eager = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable;
/*! AWAITING HUGO JSON CONVERSION TOOL
SIGNATURE NOT RECOGNISED
*/
template using lazy = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable;
/*! AWAITING HUGO JSON CONVERSION TOOL
SIGNATURE NOT RECOGNISED
*/
template using atomic_lazy = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::awaitable;
/*! AWAITING HUGO JSON CONVERSION TOOL
SIGNATURE NOT RECOGNISED
*/
template using generator = BOOST_OUTCOME_V2_NAMESPACE::awaitables::detail::generator;
BOOST_OUTCOME_COROUTINE_SUPPORT_NAMESPACE_END
#endif