sourcemod/tools/daemon/smud_threads.cpp

#include "smud_threads.h"

ThreadPool::ThreadPool() 
{
}

ThreadPool::~ThreadPool()
{
}

bool ThreadPool::Start()
{
	m_pWorker = new ThreadWorker();
	
	if (!m_pWorker->Start())
	{
		delete m_pWorker;
		return false;
	}

	return true;
}

void ThreadPool::Stop()
{
	m_pWorker->CancelAndWait();
	delete m_pWorker;
}

void ThreadPool::AddConnection(int fd)
{
	m_pWorker->AddConnection(fd);
}

ThreadWorker::ThreadWorker() : m_bShouldCancel(false)
{
}

ThreadWorker::~ThreadWorker()
{
}

bool ThreadWorker::Start()
{
	m_pPool = new ConnectionPool();
	pthread_mutex_init(&m_NotifyLock, NULL);
	pthread_cond_init(&m_Notify, NULL);

	if (pthread_create(&m_Thread, NULL, ThreadCallback, this) != 0)
	{
		return false;
	}

	return true;
}

void ThreadWorker::CancelAndWait()
{
	m_bShouldCancel = true;

	pthread_join(m_Thread, NULL);

	pthread_cond_destroy(&m_Notify);
	pthread_mutex_destroy(&m_NotifyLock);

	delete m_pPool;
}

void ThreadWorker::AddConnection( int fd )
{
	m_pPool->AddConnection(fd);
}

void ThreadWorker::Process()
{
	m_pPool->Process(&m_bShouldCancel);
}

void *ThreadCallback(void *data)
{
	((ThreadWorker *)data)->Process();

	pthread_exit(NULL);
}