Runtime/Threading/WorkThread.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

#include "Thread.h"
#include "Runtime/Debug/Log.h"

void WorkThread::Run()
{
	while (true) 
	{
		Lock(m_PendingMutex)
		{
			for (auto iter = m_PendingJobs.begin(); iter != m_PendingJobs.end();)
			{
				Job* job = *iter;
				job->Process();
				if (job->IsFinished())
				{
					Lock(m_FinishedMutex) {
						m_FinishedJobs.push_back(job);
					}
					iter = m_PendingJobs.erase(iter);
					continue;
				}
				++iter;
			}
			if (m_PendingJobs.size() == 0)
				::Sleep(100);
		}
	}
}

void WorkThread::Dispatch(void* param)
{
	Lock(m_FinishedMutex)
	{
		for (int i = 0; i < m_FinishedJobs.size(); ++i)
		{
			m_FinishedJobs[i]->Dispacth(param);
			delete m_FinishedJobs[i];
		}
		m_FinishedJobs.clear();
	}
}

void WorkThread::AddJobAtEnd(Job* job)
{
	Lock(m_PendingMutex)
	{
		m_PendingJobs.push_back(job);
	}
}