FIBER_START_FUNCTION_CLASS_IMPL(TaskScheduler, FiberStart) {
GlobalArgs *globalArgs = reinterpret_cast<GlobalArgs *>(arg);
TaskScheduler *taskScheduler = &globalArgs->g_taskScheduler;
while (!taskScheduler->m_quit.load()) {
// Check if any of the waiting tasks are ready
WaitingTask waitingTask;
bool waitingTaskReady = false;
taskScheduler->m_waitingTaskLock.lock();
auto iter = taskScheduler->m_waitingTasks.begin();
for (; iter != taskScheduler->m_waitingTasks.end(); ++iter) {
if (iter->Counter->load() == iter->Value) {
waitingTaskReady = true;
break;
}
}
if (waitingTaskReady) {
waitingTask = *iter;
// Optimization for removing an item from a vector as suggested by ryeguy on reddit
// Explained here: http://stackoverflow.com/questions/4442477/remove-ith-item-from-c-stdvector/4442529#4442529
// Essentially, rather than forcing a memcpy to shift all the remaining elements down after the erase,
// we move the last element into the place where the erased element was. Then we pop off the last element
// Check that we're not already the last item
// Move assignment to self is not defined
if (iter != (--taskScheduler->m_waitingTasks.end())) {
*iter = std::move(taskScheduler->m_waitingTasks.back());
}
taskScheduler->m_waitingTasks.pop_back();
}
taskScheduler->m_waitingTaskLock.unlock();
if (waitingTaskReady) {
taskScheduler->SwitchFibers(waitingTask.Fiber);
}
TaskBundle nextTask;
if (!taskScheduler->GetNextTask(&nextTask)) {
std::this_thread::yield();
} else {
nextTask.TaskToExecute.Function(&globalArgs->g_taskScheduler, &globalArgs->g_heap, &globalArgs->g_allocator, nextTask.TaskToExecute.ArgData);
nextTask.Counter->fetch_sub(1);
}
}
FTLConvertFiberToThread(FTLGetCurrentFiber());
globalArgs->g_taskScheduler.m_numActiveWorkerThreads.fetch_sub(1);
FTLEndCurrentThread();
}
void TaskScheduler::FiberStart(void *arg) {
GlobalArgs *globalArgs = (GlobalArgs *)arg;
TaskScheduler *taskScheduler = &globalArgs->TaskScheduler;
while (!taskScheduler->m_quit.load()) {
// Check if any of the waiting tasks are ready
WaitingTask waitingTask;
bool waitingTaskReady = false;
EnterCriticalSection(&taskScheduler->m_waitingTaskLock);
auto iter = taskScheduler->m_waitingTasks.begin();
for ( ; iter != taskScheduler->m_waitingTasks.end(); ++iter) {
if (iter->Counter->load() == iter->Value) {
waitingTaskReady = true;
break;
}
}
if (waitingTaskReady) {
waitingTask = *iter;
taskScheduler->m_waitingTasks.erase(iter);
}
LeaveCriticalSection(&taskScheduler->m_waitingTaskLock);
if (waitingTaskReady) {
taskScheduler->SwitchFibers(waitingTask.Fiber);
}
TaskBundle nextTask;
if (!taskScheduler->GetNextTask(&nextTask)) {
SwitchToThread();
} else {
nextTask.Task.Function(&globalArgs->TaskScheduler, &globalArgs->Heap, &globalArgs->Allocator, nextTask.Task.ArgData);
nextTask.Counter->fetch_sub(1);
}
}
}
