c++实现简单的线程池

/* * 多线程管理类 *  */ #ifndef CTHREADPOOLMANAGE_H#define CTHREADPOOLMANAGE_H#include <iostream>#include <pthread.h>#include <unistd.h> #include <list>#include <vector>#include <time.h>#include <asm/errno.h> #define USLEEP_TIME 100#define CHECK_TIME 1  using namespace std;class CDoit{public: virtual int start(void *){}; virtual int end(){};};  class CthreadPoolManage{private: int _minThreads;  //最少保留几个线程 int _maxThreads;  //最多可以有几个线程 int _waitSec;      //空闲多少秒后将线程关闭 class threadInfo{  public:  threadInfo(){   isbusy = false;    doFlag = true;  }  //  pthread_mutex_t mtx=PTHREAD_MUTEX_INITIALIZER;  pthread_cond_t cond=PTHREAD_COND_INITIALIZER;  bool isbusy;   //是否空闲  bool doFlag;  //  time_t beginTime;     //线程不工作开始时间  pthread_t cThreadPid; //线程id  pthread_attr_t cThreadAttr; //线程属性  CDoit * doit;         //任务类  void  * value;        //需要传递的值 }; //线程函数 static void* startThread(void*); //任务队列锁 pthread_mutex_t _duty_mutex; //任务队列 list<threadInfo*> _dutyList; //线程队列锁 pthread_mutex_t _thread_mutex; //线程队列 list<threadInfo*> _threadList;  ///初始化，创建最小个数线程///void initThread(); ///任务分配线程///static void* taskAllocation(void*arg);pthread_t tasktPid;///线程销毁、状态检查线程///static void* checkThread(void* arg);pthread_t checktPid;bool  checkrun; //线程异常退出清理static void threadCleanUp(void* arg); //int addThread(list<threadInfo*> *plist,threadInfo* ptinfo); public:CthreadPoolManage();/*保留的最少线程，最多线程数，空闲多久销毁，保留几个线程的冗余 */CthreadPoolManage(int min,int max,int waitSec);~CthreadPoolManage(); int start();//任务注入器int putDuty(CDoit *,void *); int getNowThreadNum(); }; #endif // CTHREADPOOLMANAGE_H

/* * 线程池，线程管理类 *  */ #include "cthreadpoolmanage.h" CthreadPoolManage::CthreadPoolManage(){ _minThreads = 5;  //最少保留几个线程 _maxThreads = 5;  //最多可以有几个线程 _waitSec = 10;      //空闲多少秒后将线程关闭 pthread_mutex_init(&_duty_mutex, NULL); pthread_mutex_init(&_thread_mutex, NULL); checkrun = true;}  CthreadPoolManage::CthreadPoolManage(int min, int max, int waitSec){  CthreadPoolManage();  _minThreads = min;  //最少保留几个线程  _maxThreads = max;  //最多可以有几个线程  _waitSec = waitSec;      //空闲多少秒后将线程关闭} CthreadPoolManage::~CthreadPoolManage(){ }void CthreadPoolManage::threadCleanUp(void* arg){ threadInfo* tinfo = (threadInfo*)arg; tinfo->isbusy = false; pthread_mutex_unlock(&tinfo->mtx); pthread_attr_destroy (&tinfo->cThreadAttr); delete tinfo;} void* CthreadPoolManage::startThread(void* arg){ cout<<"线程开始工作"<<endl; threadInfo* tinfo = (threadInfo*)arg; pthread_cleanup_push(threadCleanUp,arg); while(tinfo->doFlag){  pthread_mutex_lock(&tinfo->mtx);  if(tinfo->doit == NULL)  {   cout<<"开始等待任务"<<endl;   pthread_cond_wait(&tinfo->cond,&tinfo->mtx);   cout<<"有任务了"<<endl;  }  tinfo->isbusy = true;  tinfo->doit->start(tinfo->value);  tinfo->doit->end();  tinfo->doit=NULL;  tinfo->isbusy = false;  time( &tinfo->beginTime);  pthread_mutex_unlock(&tinfo->mtx); } //0正常执行到这儿不执行清理函数，异常会执行 pthread_cleanup_pop(0); pthread_attr_destroy (&tinfo->cThreadAttr); delete tinfo; cout<<"线程结束"<<endl;} void CthreadPoolManage::initThread(){ int i = 0; for(i = 0;i<this->_minThreads;i++) {   threadInfo *tinfo = new threadInfo;   tinfo->doit = NULL;   tinfo->value = NULL;   tinfo->isbusy = false;   tinfo->doFlag = true;  // PTHREAD_CREATE_DETACHED (分离线程) 和 PTHREAD _CREATE_JOINABLE (非分离线程)   pthread_attr_init(&tinfo->cThreadAttr);   pthread_attr_setdetachstate(&tinfo->cThreadAttr,PTHREAD_CREATE_DETACHED );   cout<<"初始化了一个线程"<<endl;   if(pthread_create(&tinfo->cThreadPid,&tinfo->cThreadAttr,startThread,(void *)tinfo) != 0)  {  cout<<"创建线程失败"<<endl;  break;  }  this->_threadList.push_back(tinfo); }} int CthreadPoolManage::addThread(std::list< CthreadPoolManage::threadInfo* >* plist, CthreadPoolManage::threadInfo* ptinfo){   threadInfo *tinfo = new threadInfo;   tinfo->doit = ptinfo->doit;   tinfo->value = ptinfo->value;   tinfo->isbusy = true;   if(pthread_create(&tinfo->cThreadPid,NULL,startThread,(void *)tinfo) != 0)  {  cout<<"创建线程失败"<<endl;  return -1;  }  plist->push_back(tinfo);  return 0;}  int CthreadPoolManage::putDuty(CDoit* doit, void* value){ threadInfo *tinfo = new threadInfo; time( &tinfo->beginTime); tinfo->doit= doit; tinfo->value = value; pthread_mutex_lock(&_duty_mutex);  this->_dutyList.push_back(tinfo); pthread_mutex_unlock(&_duty_mutex); return 0;} void* CthreadPoolManage::taskAllocation(void*arg){ CthreadPoolManage * ptmanage = (CthreadPoolManage*)arg; int size_1 = 0; int size_2 = 0; int i_1 = 0; int i_2 = 0; bool a_1 = true; bool a_2 = true; threadInfo* ptinfo; threadInfo* ptinfoTmp; while(true){   size_1 = 0;   size_2 = 0;   pthread_mutex_lock(&ptmanage->_duty_mutex);   pthread_mutex_lock(&ptmanage->_thread_mutex);   size_1 = ptmanage->_dutyList.size();   size_2 =ptmanage->_threadList.size();   for(list<threadInfo*>::iterator itorti1 = ptmanage->_dutyList.begin();itorti1 !=ptmanage->_dutyList.end();)   {  ptinfo = *itorti1;  a_1 = true;  for(list<threadInfo*>::iterator itorti2 = ptmanage->_threadList.begin();itorti2!=ptmanage->_threadList.end();itorti2++){   ptinfoTmp = *itorti2;   if(EBUSY == pthread_mutex_trylock(&ptinfoTmp->mtx))   {    continue;   }   if(!ptinfoTmp->isbusy)   {    ptinfoTmp->doit = ptinfo->doit;    ptinfoTmp->value = ptinfo->value;    ptinfoTmp->isbusy = true;    pthread_cond_signal(&ptinfoTmp->cond);    pthread_mutex_unlock(&ptinfoTmp->mtx);    a_1 = false;    delete ptinfo;    break;   }   pthread_mutex_unlock(&ptinfoTmp->mtx);    }    if(a_1){   if(ptmanage->_threadList.size()>ptmanage->_maxThreads||ptmanage->addThread(&ptmanage->_threadList,ptinfo)!=0)   {    itorti1++;    continue;   }else{    itorti1 = ptmanage->_dutyList.erase(itorti1);   }   delete ptinfo;    }else{   itorti1 = ptmanage->_dutyList.erase(itorti1);    }   }   pthread_mutex_unlock(&ptmanage->_duty_mutex);   pthread_mutex_unlock(&ptmanage->_thread_mutex);   usleep(USLEEP_TIME); } return 0;} void* CthreadPoolManage::checkThread(void* arg){ CthreadPoolManage * ptmanage = (CthreadPoolManage*)arg; threadInfo* ptinfo; time_t nowtime; while(ptmanage->checkrun){  sleep(CHECK_TIME);  pthread_mutex_lock(&ptmanage->_thread_mutex);  if(ptmanage->_threadList.size()<=ptmanage->_minThreads)  {   continue;  }  for(list<threadInfo*>::iterator itorti2 = ptmanage->_threadList.begin();itorti2!=ptmanage->_threadList.end();){   ptinfo = *itorti2;   if(EBUSY == pthread_mutex_trylock(&ptinfo->mtx))  {   itorti2++;   continue;  }  time(&nowtime);  if(ptinfo->isbusy == false && nowtime-ptinfo->beginTime>ptmanage->_waitSec)  {   ptinfo->doFlag = false;   itorti2 = ptmanage->_threadList.erase(itorti2);  }else{   itorti2++;  }  pthread_mutex_unlock(&ptinfo->mtx);  }  pthread_mutex_unlock(&ptmanage->_thread_mutex); }} int CthreadPoolManage::start(){ //初始化 this->initThread(); //启动任务分配线程  if(pthread_create(&tasktPid,NULL,taskAllocation,(void *)this) != 0)  {  cout<<"创建任务分配线程失败"<<endl;  return -1;  }  //创建现程状态分配管理线程  if(pthread_create(&checktPid,NULL,checkThread,(void *)this) != 0)  {  cout<<"创建线程状态分配管理线程失败"<<endl;  return -1;  } return 0;} ///////////////////////////////int CthreadPoolManage::getNowThreadNum(){ int num = 0; pthread_mutex_lock(&this->_thread_mutex);  num = this->_threadList.size(); pthread_mutex_unlock(&this->_thread_mutex); return num ;}