2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
68 lines
1.4 KiB
C++
68 lines
1.4 KiB
C++
#include <mln/actor/scheduler.hpp>
|
|
#include <mln/util/async_task.hpp>
|
|
#include <mln/util/monotonic_timer.hpp>
|
|
#include <mln/util/run_loop.hpp>
|
|
#include <mln/util/string.hpp>
|
|
|
|
#include <CoreFoundation/CoreFoundation.h>
|
|
|
|
namespace mln {
|
|
namespace util {
|
|
|
|
class RunLoop::Impl {
|
|
public:
|
|
std::unique_ptr<AsyncTask> async;
|
|
};
|
|
|
|
RunLoop* RunLoop::Get() {
|
|
assert(static_cast<RunLoop*>(Scheduler::GetCurrent()));
|
|
return static_cast<RunLoop*>(Scheduler::GetCurrent());
|
|
}
|
|
|
|
RunLoop::RunLoop(Type)
|
|
: impl(std::make_unique<Impl>()) {
|
|
assert(!Scheduler::GetCurrent(false));
|
|
Scheduler::SetCurrent(this);
|
|
impl->async = std::make_unique<AsyncTask>(std::bind(&RunLoop::process, this));
|
|
}
|
|
|
|
RunLoop::~RunLoop() {
|
|
Scheduler::SetCurrent(nullptr);
|
|
}
|
|
|
|
void RunLoop::wake() {
|
|
impl->async->send();
|
|
}
|
|
|
|
void RunLoop::run() {
|
|
CFRunLoopRun();
|
|
}
|
|
|
|
void RunLoop::runOnce() {
|
|
wake();
|
|
CFRunLoopRunInMode(kCFRunLoopDefaultMode, 0, false);
|
|
}
|
|
|
|
void RunLoop::stop() {
|
|
invoke([&] { CFRunLoopStop(CFRunLoopGetCurrent()); });
|
|
}
|
|
|
|
void RunLoop::waitForEmpty([[maybe_unused]] const SimpleIdentity tag) {
|
|
while (true) {
|
|
std::size_t remaining;
|
|
{
|
|
std::scoped_lock lock(mutex);
|
|
remaining = defaultQueue.size() + highPriorityQueue.size();
|
|
}
|
|
|
|
if (remaining == 0) {
|
|
return;
|
|
}
|
|
|
|
runOnce();
|
|
}
|
|
}
|
|
|
|
} // namespace util
|
|
} // namespace mln
|