co_spawn.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_ASYNC_CO_SPAWN_HPP__
#define NEFORCE_CORE_ASYNC_CO_SPAWN_HPP__

 
#include "NeForce/core/async/coroutine.hpp"
#ifdef NEFORCE_STANDARD_20
#    include "NeForce/core/exception/exception_ptr.hpp"
NEFORCE_BEGIN_NAMESPACE__


struct co_spawn_task {
    struct promise_type {
        exception_ptr exception_;
 
        co_spawn_task get_return_object() { return co_spawn_task{coroutine_handle<promise_type>::from_promise(*this)}; }
        suspend_always initial_suspend() noexcept { return {}; }
        auto final_suspend() noexcept {
            struct final_awaiter {
                bool await_ready() noexcept { return false; }
                void await_suspend(coroutine_handle<promise_type> h) noexcept { h.destroy(); }
                void await_resume() noexcept {}
            };
            return final_awaiter{};
        }
        void return_void() noexcept {}
        void unhandled_exception() { exception_ = current_exception(); }
 
        template <typename U>
        auto await_transform(U&& u) {
            return forward<U>(u);
        }
    };
 
    coroutine_handle<promise_type> handle_;
 
    explicit co_spawn_task(coroutine_handle<promise_type> h) :
    handle_(h) {}
};

template <typename Executor, typename Func>
void co_spawn(Executor&& exec, Func func) {
    auto inner = [](Func f) -> co_spawn_task { co_await f(); };
    auto task = inner(func);
    exec.execute([h = task.handle_]() mutable { h.resume(); });
}
 // CoroutineSpawn
 
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_STANDARD_20
#endif // NEFORCE_CORE_ASYNC_CO_SPAWN_HPP__