2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
59 lines
1.3 KiB
C++
59 lines
1.3 KiB
C++
#include <mln/util/async_task.hpp>
|
|
|
|
#include <CoreFoundation/CoreFoundation.h>
|
|
|
|
#include <atomic>
|
|
|
|
namespace mln {
|
|
namespace util {
|
|
|
|
class AsyncTask::Impl {
|
|
public:
|
|
Impl(std::function<void()>&& fn)
|
|
: task(std::move(fn)),
|
|
loop(CFRunLoopGetCurrent()) {
|
|
CFRunLoopSourceContext context = {
|
|
0, this, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, perform};
|
|
source = CFRunLoopSourceCreate(kCFAllocatorDefault, 0, &context);
|
|
CFRunLoopAddSource(loop, source, kCFRunLoopCommonModes);
|
|
}
|
|
|
|
~Impl() {
|
|
CFRunLoopSourceInvalidate(source);
|
|
CFRelease(source);
|
|
}
|
|
|
|
void maySend() {
|
|
if (!queued.test_and_set()) {
|
|
CFRunLoopSourceSignal(source);
|
|
CFRunLoopWakeUp(loop);
|
|
}
|
|
}
|
|
|
|
void runTask() {
|
|
queued.clear();
|
|
task();
|
|
}
|
|
|
|
private:
|
|
static void perform(void* info) { reinterpret_cast<Impl*>(info)->runTask(); }
|
|
|
|
std::function<void()> task;
|
|
std::atomic_flag queued = ATOMIC_FLAG_INIT;
|
|
|
|
CFRunLoopRef loop;
|
|
CFRunLoopSourceRef source;
|
|
};
|
|
|
|
AsyncTask::AsyncTask(std::function<void()>&& fn)
|
|
: impl(std::make_unique<Impl>(std::move(fn))) {}
|
|
|
|
AsyncTask::~AsyncTask() = default;
|
|
|
|
void AsyncTask::send() {
|
|
impl->maySend();
|
|
}
|
|
|
|
} // namespace util
|
|
} // namespace mln
|