Core/TaskScheduler: Increase shutdown performance
Tell all threads to exit and join them all at once instead of one by one
This commit is contained in:
@@ -96,10 +96,6 @@ namespace Nz
|
|||||||
|
|
||||||
~Worker()
|
~Worker()
|
||||||
{
|
{
|
||||||
m_running = false;
|
|
||||||
if (!m_notifier.test_and_set())
|
|
||||||
m_notifier.notify_one();
|
|
||||||
|
|
||||||
m_thread.join();
|
m_thread.join();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -170,6 +166,13 @@ namespace Nz
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Shutdown()
|
||||||
|
{
|
||||||
|
m_running = false;
|
||||||
|
if (!m_notifier.test_and_set())
|
||||||
|
m_notifier.notify_one();
|
||||||
|
}
|
||||||
|
|
||||||
TaskScheduler::Task* StealTask()
|
TaskScheduler::Task* StealTask()
|
||||||
{
|
{
|
||||||
return m_tasks.steal();
|
return m_tasks.steal();
|
||||||
@@ -214,12 +217,17 @@ namespace Nz
|
|||||||
for (unsigned int i = 0; i < m_workerCount; ++i)
|
for (unsigned int i = 0; i < m_workerCount; ++i)
|
||||||
m_workers.emplace_back(*this, i);
|
m_workers.emplace_back(*this, i);
|
||||||
|
|
||||||
for (unsigned int i = 0; i < m_workerCount; ++i)
|
for (Worker& worker : m_workers)
|
||||||
m_workers[i].WakeUp();
|
worker.WakeUp();
|
||||||
}
|
}
|
||||||
|
|
||||||
TaskScheduler::~TaskScheduler()
|
TaskScheduler::~TaskScheduler()
|
||||||
{
|
{
|
||||||
|
// Wake up workers and tell them to exit
|
||||||
|
for (Worker& worker : m_workers)
|
||||||
|
worker.Shutdown();
|
||||||
|
|
||||||
|
// wait for them to have exited
|
||||||
m_workers.clear();
|
m_workers.clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user