52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
/***
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
File: WorkItem.hpp
|
|
Date: 2021-6-26
|
|
Author: Reece
|
|
***/
|
|
#pragma once
|
|
|
|
namespace Aurora::Async
|
|
{
|
|
class WorkItem : public IWorkItem, public IAsyncRunnable, public std::enable_shared_from_this<WorkItem>
|
|
{
|
|
public:
|
|
WorkItem(const DispatchTarget_t &worker_, const AuSPtr<IWorkItemHandler> &task_, bool supportsBlocking);
|
|
~WorkItem();
|
|
|
|
AuSPtr<IWorkItem> WaitFor(const AuSPtr<IWorkItem> &workItem) override;
|
|
AuSPtr<IWorkItem> WaitFor(const AuList<AuSPtr<IWorkItem>> &workItem) override;
|
|
AuSPtr<IWorkItem> SetSchedTime(AuUInt32 ms) override;
|
|
AuSPtr<IWorkItem> SetSchedTimeNs(AuUInt64 ns) override;
|
|
AuSPtr<IWorkItem> Then(const AuSPtr<IWorkItem> &next) override;
|
|
|
|
AuSPtr<IWorkItem> Dispatch() override;
|
|
|
|
bool BlockUntilComplete() override;
|
|
bool HasFinished() override;
|
|
bool HasFailed() override;
|
|
|
|
void RunAsync() override;
|
|
void CancelAsync() override;
|
|
|
|
private:
|
|
void DispatchEx(bool check);
|
|
AuSPtr<IWorkItemHandler> task_;
|
|
DispatchTarget_t worker_;
|
|
AuList<AuSPtr<IWorkItem>> waitOn_;
|
|
AuList<AuSPtr<IWorkItem>> waiters_;
|
|
Threading::Primitives::SpinLock lock;
|
|
Threading::Primitives::EventUnique_t finishedEvent_;
|
|
|
|
bool finished {};
|
|
bool failed {};
|
|
bool dispatchPending_ {};
|
|
AuUInt64 dispatchTimeNs_ {};
|
|
|
|
void Fail();
|
|
void Schedule();
|
|
void SendOff();
|
|
};
|
|
|
|
} |