// 线程池管理类实现

#include "StdAfx.h"
#include "ThreadPoolMang.h"
#include "WorkThread.h"
#include "Task.h"

CThreadPoolMang::CThreadPoolMang(int nThreadCount)
	: m_nThreadCount(nThreadCount)	
{	
	Initial();
}


CThreadPoolMang::~CThreadPoolMang(void)
{
	Release();
}

void CThreadPoolMang::Initial()
{
	// 创建线程池
	int i = 0;
	while ( i < m_nThreadCount )
	{
		m_sBlockThread.push(new CWorkThread(this, i+1));
		//m_lGlobalCounter.Lock();不是直接传入线程中，所以无需加锁
		i++;
		//m_lGlobalCounter.UnLock();
	}
}

void CThreadPoolMang::Release()
{
	// 释放资源
	while (!m_sBlockThread.empty())
	{
		CWorkThread* pThread = m_sBlockThread.top();
		//pThread->ExitThread();
		delete pThread;
		m_sBlockThread.pop();
	}
}

void CThreadPoolMang::AddTask( CTask* pTask )
{
	// 1.是否有空闲线程，有则开始执行任务
	if (!m_sBlockThread.empty())
	{
		// 操作stl，阻塞线程加锁
		m_lBlockThread.Lock();		
		CWorkThread* pThread = m_sBlockThread.top();
		assert(pThread != NULL);
		m_sBlockThread.pop();
		m_lBlockThread.UnLock();

		// 通知线程未激活状态
		pThread->BeforeWork(pTask);

		// 将线程加入到激活线程中
		m_lActiveThread.Lock();
		m_lstActiveThread.push_back(pThread);
		m_lActiveThread.UnLock();
		return;
	}

	// 2.没有则，加入任务列表等待
	m_lTaskDeque.Lock();
	if (HIGH == pTask->PriorityLevel())
	{
		m_deqTask.push_front(pTask);
	}
	else
	{
		m_deqTask.push_back(pTask);
	}
	m_lTaskDeque.UnLock();
}

void CThreadPoolMang::BlockThread( CWorkThread* pThread )
{
	assert(pThread != NULL);
	// 先判断是否有任务还未执行
	if (!m_deqTask.empty())
	{
		// 取出任务
		m_lTaskDeque.Lock();
		CTask* pTask = m_deqTask.front();
		assert(pTask != NULL);
		m_deqTask.pop_front();
		m_lTaskDeque.UnLock();

		// 为线程分配任务,
		pThread->BeforeWork(pTask);
		return;
	}

	// 2.没有任务，则将线程加入到阻塞队列中来
	m_lBlockThread.Lock();
	m_sBlockThread.push(pThread);
	m_lBlockThread.UnLock();

	// 3.并且冲激活线程列表中移除
	m_lActiveThread.Lock();
	m_lstActiveThread.remove(pThread);
	m_lActiveThread.UnLock();
}

HANDLE CThreadPoolMang::GetThreadHandle()
{
	return new HANDLE[10];
}