linux的创建线程池,Linux下通用线程池的创建与使用(上) （3）

线程的状态可以分为四种，空闲、忙碌、挂起、终止(包括正常退出和非正常退出)。由于目前Linux线程库不支持挂起操作，因此，我们的此处的挂起操作类似于暂停。如果线程创建后不想立即执行任务，那么我们可以将其“暂停”，如果需要运行，则唤醒。有一点必须注意的是，一旦线程开始执行任务，将不能被挂起，其将一直执行任务至完毕。

线程类的相关操作均十分简单。线程的执行入口是从Start()函数开始，其将调用函数ThreadFunction，ThreadFunction再调用实际的Run函数，执行实际的任务。

CThreadPool

CThreadPool是线程的承载容器，一般可以将其实现为堆栈、单向队列或者双向队列。在我们的系统中我们使用STL Vector对线程进行保存。CThreadPool的实现代码如下：

class CThreadPool

{

friend class CWorkerThread;

private:

unsigned int m_MaxNum; //the max thread num that can create at the same time

unsigned int m_AvailLow; //The min num of idle thread that shoule kept

unsigned int m_AvailHigh; //The max num of idle thread that kept at the same time

unsigned int m_AvailNum; //the normal thread num of idle num;

unsigned int m_InitNum; //Normal thread num;

protected:

CWorkerThread* GetIdleThread(void);

void AppendToIdleList(CWorkerThread* jobthread);

void MoveToBusyList(CWorkerThread* idlethread);

void MoveToIdleList(CWorkerThread* busythread);

void DeleteIdleThread(int num);

void CreateIdleThread(int num);

public:

CThreadMutex m_BusyMutex; //when visit busy list,use m_BusyMutex to lock and unlock

CThreadMutex m_IdleMutex; //when visit idle list,use m_IdleMutex to lock and unlock

CThreadMutex m_JobMutex; //when visit job list,use m_JobMutex to lock and unlock

CThreadMutex m_VarMutex;

CCondition m_BusyCond; //m_BusyCond is used to sync busy thread list

CCondition m_IdleCond; //m_IdleCond is used to sync idle thread list

CCondition m_IdleJobCond; //m_JobCond is used to sync job list

CCondition m_MaxNumCond;

vector m_ThreadList;

vector m_BusyList; //Thread List

vector m_IdleList; //Idle List

CThreadPool();

CThreadPool(int initnum);

virtual ~CThreadPool();

void SetMaxNum(int maxnum){m_MaxNum = maxnum;}

int GetMaxNum(void){return m_MaxNum;}

void SetAvailLowNum(int minnum){m_AvailLow = minnum;}

int GetAvailLowNum(void){return m_AvailLow;}

void SetAvailHighNum(int highnum){m_AvailHigh = highnum;}

int GetAvailHighNum(void){return m_AvailHigh;}

int GetActualAvailNum(void){return m_AvailNum;}

int GetAllNum(void){return m_ThreadList.size();}

int GetBusyNum(void){return m_BusyList.size();}

void SetInitNum(int initnum){m_InitNum = initnum;}

int GetInitNum(void){return m_InitNum;}

void TerminateAll(void);

void Run(CJob* job,void* jobdata);

};

CThreadPool::CThreadPool()

{

m_MaxNum = 50;

m_AvailLow = 5;

m_InitNum=m_AvailNum = 10 ;

m_AvailHigh = 20;

m_BusyList.clear();

m_IdleList.clear();

for(int i=0;i

CWorkerThread* thr = new CWorkerThread();

thr->SetThreadPool(this);

AppendToIdleList(thr);

thr->Start();

}

}

CThreadPool::CThreadPool(int initnum)

{

assert(initnum>0 && initnum<=30);

m_MaxNum = 30;

m_AvailLow = initnum-10>0?initnum-10:3;

m_InitNum=m_AvailNum = initnum ;

m_AvailHigh = initnum+10;

m_BusyList.clear();

m_IdleList.clear();

for(int i=0;i

CWorkerThread* thr = new CWorkerThread();

AppendToIdleList(thr);

thr->SetThreadPool(this);

thr->Start(); //begin the thread,the thread wait for job

}

}

CThreadPool::~CThreadPool()

{

TerminateAll();

}

void CThreadPool::TerminateAll()

{

for(int i=0;i < m_ThreadList.size();i++) {

CWorkerThread* thr = m_ThreadList[i];

thr->Join();

}

return;

}

CWorkerThread* CThreadPool::GetIdleThread(void)

{

while(m_IdleList.size() ==0 )

m_IdleCond.Wait();

m_IdleMutex.Lock();

if(m_IdleList.size() > 0 )

{

CWorkerThread* thr = (CWorkerThread*)m_IdleList.front();

printf("Get Idle thread %dn",thr->GetThreadID());

m_IdleMutex.Unlock();

return thr;

}

m_IdleMutex.Unlock();

return NULL;

}

//add an idle thread to idle list

void CThreadPool::AppendToIdleList(CWorkerThread* jobthread)

{

m_IdleMutex.Lock();

m_IdleList.push_back(jobthread);

m_ThreadList.push_back(jobthread);

m_IdleMutex.Unlock();

}

//move and idle thread to busy thread

void CThreadPool::MoveToBusyList(CWorkerThread* idlethread)

{

m_BusyMutex.Lock();

m_BusyList.push_back(idlethread);

m_AvailNum--;

m_BusyMutex.Unlock();

m_IdleMutex.Lock();

vector::iterator pos;

pos = find(m_IdleList.begin(),m_IdleList.end(),idlethread);

if(pos !=m_IdleList.end())

m_IdleList.erase(pos);

m_IdleMutex.Unlock();

}

void CThreadPool::MoveToIdleList(CWorkerThread* busythread)

{

m_IdleMutex.Lock();

m_IdleList.push_back(busythread);

m_AvailNum++;

m_IdleMutex.Unlock();

m_BusyMutex.Lock();

vector::iterator pos;

pos = find(m_BusyList.begin(),m_BusyList.end(),busythread);

if(pos!=m_BusyList.end())

m_BusyList.erase(pos);

m_BusyMutex.Unlock();

m_IdleCond.Signal();

m_MaxNumCond.Signal();

}

//create num idle thread and put them to idlelist

void CThreadPool::CreateIdleThread(int num)

{

for(int i=0;i

CWorkerThread* thr = new CWorkerThread();

thr->SetThreadPool(this);

AppendToIdleList(thr);

m_VarMutex.Lock();

m_AvailNum++;

m_VarMutex.Unlock();

thr->Start(); //begin the thread,the thread wait for job

}

}

void CThreadPool::DeleteIdleThread(int num)

{

printf("Enter into CThreadPool::DeleteIdleThreadn");

m_IdleMutex.Lock();

printf("Delete Num is %dn",num);

for(int i=0;i

CWorkerThread* thr;

if(m_IdleList.size() > 0 ){

thr = (CWorkerThread*)m_IdleList.front();

printf("Get Idle thread %dn",thr->GetThreadID());

}

vector::iterator pos;

pos = find(m_IdleList.begin(),m_IdleList.end(),thr);

if(pos!=m_IdleList.end())

m_IdleList.erase(pos);

m_AvailNum--;

printf("The idle thread available num:%d n",m_AvailNum);

printf("The idlelist num:%d n",m_IdleList.size());

}

m_IdleMutex.Unlock();

}

void CThreadPool::Run(CJob* job,void* jobdata)

{

assert(job!=NULL);

//if the busy thread num adds to m_MaxNum,so we should wait

if(GetBusyNum() == m_MaxNum)

m_MaxNumCond.Wait();

if(m_IdleList.size()

{

if(GetAllNum()+m_InitNum-m_IdleList.size() < m_MaxNum )

CreateIdleThread(m_InitNum-m_IdleList.size());

else

CreateIdleThread(m_MaxNum-GetAllNum());

}

CWorkerThread* idlethr = GetIdleThread();

if(idlethr !=NULL)

{

idlethr->m_WorkMutex.Lock();

MoveToBusyList(idlethr);

idlethr->SetThreadPool(this);

job->SetWorkThread(idlethr);

printf("Job is set to thread %d n",idlethr->GetThreadID());

idlethr->SetJob(job,jobdata);

}

}