90 lines
3.2 KiB
C++
90 lines
3.2 KiB
C++
/***
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
File: AsyncApp.hpp
|
|
Date: 2021-6-26
|
|
Author: Reece
|
|
***/
|
|
#pragma once
|
|
|
|
namespace Aurora::Async
|
|
{
|
|
struct GroupState;
|
|
struct ThreadState;
|
|
//class WorkItem;
|
|
|
|
|
|
void DecRunningTasks();
|
|
void IncRunningTasks();
|
|
|
|
class AsyncApp : public IAsyncApp
|
|
{
|
|
public:
|
|
AsyncApp();
|
|
|
|
// Main thread logic
|
|
void Start() override;
|
|
void Main() override;
|
|
void Shutdown() override;
|
|
bool Exiting() override;
|
|
void SetConsoleCommandDispatcher(WorkerId_t id) override;
|
|
|
|
// Spawning
|
|
bool Spawn(WorkerId_t) override;
|
|
Threading::Threads::ThreadShared_t ResolveHandle(WorkerId_t) override;
|
|
AuBST<ThreadGroup_t, AuList<ThreadId_t>> GetThreads() override;
|
|
WorkerId_t GetCurrentThread() override;
|
|
|
|
// Synchronization
|
|
bool Sync(WorkerId_t group, AuUInt32 timeoutMs, bool requireSignal) override;
|
|
void Signal(WorkerId_t group) override;
|
|
|
|
bool WaitFor(WorkerId_t unlocker, const AuSPtr<Threading::IWaitable> &primitive, AuUInt32 ms); // when unlocker = this, pump event loop
|
|
//bool WaitFor(DispatchTarget_t unlocker, Threading::IWaitable *primitive, AuUInt32 ms) override; // when unlocker = this, pump event loop
|
|
|
|
void SyncAllSafe() override;
|
|
|
|
// Features
|
|
void AddFeature(WorkerId_t id, AuSPtr<Threading::Threads::IThreadFeature> feature, bool async = false) override;
|
|
|
|
// Debug
|
|
void AssertInThreadGroup(ThreadGroup_t group) override;
|
|
void AssertWorker(WorkerId_t id) override;
|
|
|
|
void Run(WorkerId_t target, AuSPtr<IAsyncRunnable> runnable);
|
|
|
|
bool Poll(bool block) override;
|
|
bool PollInternal(bool block);
|
|
bool PollLoopSource(bool block);
|
|
|
|
size_t GetThreadWorkersCount(ThreadGroup_t group);
|
|
|
|
bool CtxYield();
|
|
int CfxPollPush();
|
|
void CtxPollReturn(const AuSPtr<ThreadState> &state, int status, bool hitTask);
|
|
bool ScheduleLoopSource(const AuSPtr<Loop::ILoopSource> &loopSource, WorkerId_t workerId, AuUInt32 timeout, const AuConsumer<AuSPtr<Loop::ILoopSource>, bool> &callback) override;
|
|
|
|
private:
|
|
AuSPtr<ThreadState> GetThreadHandle(WorkerId_t id);
|
|
|
|
void ThisExiting();
|
|
void ShutdownZero();
|
|
|
|
// TODO: BarrierMultiple
|
|
bool Barrier(WorkerId_t, AuUInt32 ms, bool requireSignal, bool drop);
|
|
|
|
Threading::Primitives::RWLockUnique_t rwlock_;
|
|
|
|
AuSPtr<GroupState> GetGroup(ThreadGroup_t type);
|
|
|
|
AuSPtr<ThreadState> GetThreadState();
|
|
|
|
void Entrypoint(WorkerId_t id);
|
|
|
|
using ThreadDb_t = AuBST<ThreadGroup_t, AuSPtr<GroupState>>;
|
|
|
|
ThreadDb_t threads_;
|
|
bool shuttingdown_ {};
|
|
AuOptional<WorkerId_t> commandDispatcher_;
|
|
};
|
|
} |