mirror of
https://github.com/SerenityOS/serenity
synced 2024-10-01 13:44:21 +00:00
49 lines
859 B
C
49 lines
859 B
C
|
/*
|
||
|
* Copyright (c) 2024, Ali Mohammad Pur <mpfard@serenityos.org>
|
||
|
*
|
||
|
* SPDX-License-Identifier: BSD-2-Clause
|
||
|
*/
|
||
|
|
||
|
#pragma once
|
||
|
|
||
|
#include <AK/Coroutine.h>
|
||
|
#include <AK/Error.h>
|
||
|
#include <AK/Function.h>
|
||
|
|
||
|
namespace AK {
|
||
|
class GenericAwaiter {
|
||
|
AK_MAKE_NONMOVABLE(GenericAwaiter);
|
||
|
AK_MAKE_NONCOPYABLE(GenericAwaiter);
|
||
|
|
||
|
public:
|
||
|
GenericAwaiter(Function<void(Function<void()>)>&& fn)
|
||
|
{
|
||
|
fn([this] { ready(); });
|
||
|
}
|
||
|
|
||
|
bool await_ready() const { return false; }
|
||
|
void await_suspend(std::coroutine_handle<> handle)
|
||
|
{
|
||
|
m_handle = handle;
|
||
|
}
|
||
|
ErrorOr<void> await_resume()
|
||
|
{
|
||
|
m_handle = {};
|
||
|
return {};
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
void ready()
|
||
|
{
|
||
|
if (m_handle)
|
||
|
m_handle.resume();
|
||
|
}
|
||
|
|
||
|
std::coroutine_handle<> m_handle;
|
||
|
};
|
||
|
}
|
||
|
|
||
|
#ifdef USING_AK_GLOBALLY
|
||
|
using AK::GenericAwaiter;
|
||
|
#endif
|