91 lines
2.2 KiB
C++
91 lines
2.2 KiB
C++
/***
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
File: Schedular.cpp
|
|
Date: 2021-6-26
|
|
Author: Reece
|
|
***/
|
|
#include <RuntimeInternal.hpp>
|
|
#include "Async.hpp"
|
|
#include "Schedular.hpp"
|
|
#include "AsyncApp.hpp"
|
|
|
|
namespace Aurora::Async
|
|
{
|
|
struct SchedEntry
|
|
{
|
|
AuUInt64 ns;
|
|
DispatchTarget_t target;
|
|
AuSPtr<IAsyncRunnable> runnable;
|
|
};
|
|
|
|
static Threading::Threads::ThreadUnique_t gThread;
|
|
static Threading::Primitives::RWLockUnique_t gSchedLock;
|
|
static AuList<SchedEntry> gEntries;
|
|
|
|
static void GetDispatchableTasks(AuList<SchedEntry> &pending)
|
|
{
|
|
Threading::LockGuardPtr lock(gSchedLock->AsReadable());
|
|
|
|
auto time = Time::CurrentClockNS();
|
|
|
|
for (auto itr = gEntries.begin(); itr != gEntries.end(); )
|
|
{
|
|
if (itr->ns <= time)
|
|
{
|
|
pending.push_back(std::move(*itr));
|
|
itr = gEntries.erase(itr);
|
|
}
|
|
else
|
|
{
|
|
itr ++;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void SchedThread()
|
|
{
|
|
auto thread = Threading::Threads::GetThread();
|
|
AuList<SchedEntry> pending;
|
|
|
|
while (!thread->Exiting())
|
|
{
|
|
Threading::SleepNs(1000000 / 2);
|
|
|
|
AuList<SchedEntry> pending;
|
|
GetDispatchableTasks(pending);
|
|
|
|
for (auto &entry : pending)
|
|
{
|
|
static_cast<AsyncApp *>(GetAsyncApp())->Run(entry.target, entry.runnable);
|
|
DecRunningTasks();
|
|
}
|
|
}
|
|
}
|
|
|
|
void InitSched()
|
|
{
|
|
gSchedLock = Threading::Primitives::RWLockUnique();
|
|
|
|
Threading::Threads::AbstractThreadVectors handler;
|
|
handler.DoRun = [=](const Threading::Threads::IAuroraThread *thread)
|
|
{
|
|
SchedThread();
|
|
};
|
|
gThread = Threading::Threads::ThreadUnique(handler);
|
|
gThread->Run();
|
|
}
|
|
|
|
void ShutdownSched()
|
|
{
|
|
gThread.reset();
|
|
gSchedLock.reset();
|
|
}
|
|
|
|
void Schedule(AuUInt64 ns, DispatchTarget_t target, AuSPtr<IAsyncRunnable> runnable)
|
|
{
|
|
Threading::LockGuardPtr lock(gSchedLock->AsWritable());
|
|
IncRunningTasks();
|
|
gEntries.push_back({ns, target, runnable});
|
|
}
|
|
} |