implement async.

This commit is contained in:
ykiko
2024-07-08 09:41:19 +08:00
parent d0fb4f0fc1
commit f689a25f65

View File

@@ -1 +1,117 @@
#pragma once
#include <uv.h>
#include <optional>
#include <coroutine>
namespace clice {
/// An async coroutine that runs a callback on a worker thread.
template <typename Callback>
class async {
private:
Callback callback;
std::optional<decltype(callback())> result;
uv_work_t req;
std::coroutine_handle<> handle;
public:
template <typename Fn>
requires std::is_invocable_v<Fn>
async(Fn&& fn) : callback(std::forward<Fn>(fn)) {}
bool await_ready(this const async& self) noexcept { return false; }
void await_suspend(this async& self, std::coroutine_handle<> handle) noexcept {
self.handle = handle;
self.req.data = &self;
uv_queue_work(
uv_default_loop(),
&self.req,
[](uv_work_t* req) {
auto& self = *static_cast<async*>(req->data);
self.result = self.callback();
},
[](uv_work_t* req, int status) {
if(status != 0) {
// TODO: handle error
}
auto& self = *static_cast<async*>(req->data);
self->handle.resume();
});
}
decltype(auto) await_resume(this async& self) noexcept {
assert(self.result.has_value());
return *self.result;
}
};
template <typename T>
async(T) -> async<std::decay_t<T>>;
template <typename T>
class Task {
struct promise_type {
std::optional<T> value;
Task get_return_object(this promise_type& self) {
return {std::coroutine_handle<promise_type>::from_promise(self)};
}
std::suspend_always initial_suspend() { return {}; }
std::suspend_always final_suspend() noexcept { return {}; }
void return_value(this promise_type& self, T&& value) { self.value = std::move(value); }
void return_value(this promise_type& self, const T& value) { self.value = value; }
void unhandled_exception() { std::terminate(); }
};
std::coroutine_handle<promise_type> handle;
public:
Task(std::coroutine_handle<promise_type> handle) : handle(handle) {}
~Task() {
if(!handle.done()) {
handle.destroy();
}
}
T get() { return std::move(handle.promise().value); }
};
template <>
class Task<void> {
struct promise_type {
Task get_return_object(this promise_type& self) {
return {std::coroutine_handle<promise_type>::from_promise(self)};
}
std::suspend_never initial_suspend() { return {}; }
std::suspend_never final_suspend() noexcept { return {}; }
void return_void() {}
void unhandled_exception() { std::terminate(); }
};
std::coroutine_handle<promise_type> handle;
public:
Task(std::coroutine_handle<promise_type> handle) : handle(handle) {}
~Task() {
if(!handle.done()) {
handle.destroy();
}
}
void get() {}
};
} // namespace clice