ET9_WebGL_Base/Src/Egametang/Thread/ThreadPool.cc

#include <glog/logging.h>
#include "Thread/ThreadPool.h"

namespace Egametang {

ThreadPool::ThreadPool(int num) :
	thread_num(num), running(true), work_num(0)
{
	if (num == 0)
	{
		thread_num = boost::thread::hardware_concurrency();
	}
	for (int i = 0; i < thread_num; ++i)
	{
		ThreadPtr t(new boost::thread(
				boost::bind(&ThreadPool::Runner, this)));
		threads.push_back(t);
		t->detach();
		++work_num;
	}
}

ThreadPool::~ThreadPool()
{
}

void ThreadPool::Wait()
{
	boost::mutex::scoped_lock lock(mutex);
	running = false;
	cond.notify_all();
	while (work_num > 0)
	{
		done.wait(lock);
	}
	running = true;
}

void ThreadPool::Runner()
{
	bool continued = true;
	while (continued)
	{
		boost::function<void (void)> task;
		{
			boost::mutex::scoped_lock lock(mutex);
			while (running && tasks.empty())
			{
				cond.wait(lock);
			}
			if (!tasks.empty())
			{
				task = tasks.front();
				tasks.pop_front();
			}
			continued = running || !tasks.empty();
		}

		if (task)
		{
			task();
		}
	}
	if (--work_num == 0)
	{
		done.notify_one();
	}
}

bool ThreadPool::Schedule(boost::function<void (void)> task)
{
	{
		boost::mutex::scoped_lock lock(mutex);
		if (!running)
		{
			return false;
		}
		tasks.push_back(task);
	}
	cond.notify_one();
	return true;
}

} // namespace Egametang