linux 线程池封装类

#include "threadpool.h" CTask::~CTask() { } CTask::CTask() { m_nDoFlag=1; } CThreadPool::CThreadPool( int nThreadCount ) { m_nThreadCount=nThreadCount; } CThreadPool::~CThreadPool() { Fini(); } void CThreadPool::AddThread(int nCount) { } int CThreadPool::Init() { pthread_mutex_init(&m_taskMutex,NULL); pthread_mutex_init(&m_taskListMutex,NULL); pthread_mutex_init(&m_threadListMutex,NULL); pthread_cond_init(&m_taskCond,NULL); // init thread for (int i=0;i<m_nThreadCount;i++) { CThreadItem* pItem=new CThreadItem(i,this); m_threadList.push_back(pItem); } return 0; } void CThreadPool::Fini() { list<CThreadItem*>::iterator itor; for (itor=m_threadList.begin();itor!=m_threadList.end();itor++) { CThreadItem* pItem=*itor; if (pItem) { pItem->SetQuit(); } } pthread_cond_broadcast(&m_taskCond); for (itor=m_threadList.begin();itor!=m_threadList.end();itor++) { CThreadItem* pItem=*itor; if (pItem) { delete pItem; } } pthread_mutex_destroy(&m_taskMutex); pthread_mutex_destroy(&m_taskListMutex); pthread_mutex_destroy(&m_threadListMutex); } void CThreadPool::AddTask( CTask* pTask ) { if (!pTask) { return; } pthread_mutex_lock(&m_taskListMutex); m_taskList.push_back(pTask); pthread_mutex_unlock(&m_taskListMutex); pthread_cond_signal(&m_taskCond); } CThreadItem::CThreadItem(int nID,CThreadPool* pThreadPool ) { m_nID=nID; m_hThreadHandle=0; m_nThreadFlag=1; m_pThreadPool=pThreadPool; // create thread int nError=pthread_create(&m_hThreadHandle,NULL,TreatTaskThread,this); if (0!=nError) { printf("create thread failed nid:%d \n",m_nID); } else { printf("create thread nid:%d threadhandle:%lu \n",m_nID,m_hThreadHandle); } } CThreadItem::~CThreadItem() { if (m_hThreadHandle) { m_nThreadFlag=0; pthread_join(m_hThreadHandle,NULL); printf("leave the thread nID: %d threadhandle %lu \n",m_nID,m_hThreadHandle); m_hThreadHandle=0; } } void CThreadItem::OnTreatTaskThread() { while (m_nThreadFlag) { pthread_mutex_lock(&m_pThreadPool->m_taskMutex); pthread_cond_wait(&m_pThreadPool->m_taskCond, &m_pThreadPool->m_taskMutex); pthread_mutex_unlock(&m_pThreadPool->m_taskMutex); // get task and do it pthread_mutex_lock(&m_pThreadPool->m_taskListMutex); if (m_pThreadPool->m_taskList.size()>0) { CTask* pTask=m_pThreadPool->m_taskList.front(); m_pThreadPool->m_taskList.pop_front(); pthread_mutex_unlock(&m_pThreadPool->m_taskListMutex); if (pTask) { int nRet=pTask->Do(); printf("task being treated by thread handle :%lu \n",m_hThreadHandle); if (0==nRet) { delete pTask; } } } else { pthread_mutex_unlock(&m_pThreadPool->m_taskListMutex); } } }