AuroraSupport/AuroraRuntime
2
1
Fork 0
Code Issues 27 Pull Requests Releases Wiki Activity

[*] addd4080 cont

Browse Source
This commit is contained in:
Reece Wilson 2023-09-15 16:53:18 +01:00
parent f1e9aa1d22
commit d49d4ad4fc
1 changed files with 34 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
Source/Async/WorkItem.cpp
View File

@ -478,7 +478,11 @@ namespace Aurora::Async
}
auto pWorker = GetCurrentWorkerPId().pool;
if (!pWorker) AuUnsafeRaiiToShared(gAsyncApp);
if (!pWorker)
{
pWorker = AuUnsafeRaiiToShared(gAsyncApp);
}
return AuMakeShared<WorkItem>(ToInternal(pWorker).get(), WorkerPId_t { pWorker , worker }, task);
}
@ -492,7 +496,18 @@ namespace Aurora::Async
return {};
}
return AuMakeSharedThrow<FuncWorker>(ToInternal(worker.pool).get(), worker, AuMove(func));
auto pWorker = worker.pool;
if (!pWorker)
{
pWorker = GetCurrentWorkerPId().pool;
}
if (!pWorker)
{
pWorker = AuUnsafeRaiiToShared(gAsyncApp);
}
return AuMakeSharedThrow<FuncWorker>(ToInternal(pWorker).get(), worker, AuMove(func));
}
AUKN_SYM AuSPtr<IWorkItem> NewWorkItem(const WorkerPId_t &worker, const AuSPtr<IWorkItemHandler> &task)
@ -504,14 +519,28 @@ namespace Aurora::Async
SysPushErrorArg("WorkItem has null task. Running out of memory?");
return {};
}
return AuMakeSharedThrow<WorkItem>(ToInternal(worker.pool).get(), worker, task);
auto pWorker = worker.pool;
if (!pWorker)
{
pWorker = GetCurrentWorkerPId().pool;
}
if (!pWorker)
{
pWorker = AuUnsafeRaiiToShared(gAsyncApp);
}
return AuMakeSharedThrow<WorkItem>(ToInternal(pWorker).get(), worker, task);
}
AUKN_SYM AuSPtr<IWorkItem> NewFence()
{
auto pWorker = GetCurrentWorkerPId().pool;
if (!pWorker) AuUnsafeRaiiToShared(gAsyncApp);
if (!pWorker)
{
pWorker = AuUnsafeRaiiToShared(gAsyncApp);
}
return AuMakeShared<WorkItem>((IThreadPoolInternal *)ToInternal(pWorker).get(), WorkerPId_t {}, AuSPtr<IWorkItemHandler>{});
}