线程池的设计与测试

来源：互联网发布：千元机机2017 知乎编辑：程序博客网时间：2024/06/05 06:12

编写了一个最基本的线程池类，处理用c_work表示的工作任务。C++还很不熟练，欢迎会C++的提出宝贵的修改意见。

程序有注释，所以应该很好读懂。测试程序在下面。

[cpp] view plaincopyprint?

///////////////////////////////////////////////////////
//线程池类
///////////////////////////////////////////////////////
#include <pthread.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <assert.h>
const int DEFAULT_MAX_THREAD_NUM = 10;
const int MAX_WORK_NUM = 100000;
//c_worker类
class c_work
{
public:
c_work():process(NULL), arg(NULL), next(NULL){}
c_work(void *(*prcss)(void *),void *arg):
process(prcss), arg(arg), next(NULL) {}
~c_work();
void *(*process)(void *);
void *arg;
unsigned char type;//最高位表示arg是否需要delete操作
c_work *next;
};
c_work::~c_work()
{
unsigned char ifdel = type >> 7;
if (ifdel)
{
delete arg;
arg = NULL;
}
}
class c_thread_pool
{
public:
c_thread_pool();
c_thread_pool(constint max_thread_num);
~c_thread_pool();
int add_work(c_work work);
static void *thread_routine(void *arg);
pthread_mutex_t queue_lock;
pthread_cond_t queue_cond;
// private:
c_work *queue_head;
c_work *queue_tail;
int shutdown;
pthread_t *threadid;
int max_thread_num;
int cur_queue_size;
};
c_thread_pool::c_thread_pool()
{
pthread_mutex_init(&queue_lock, NULL);
pthread_cond_init(&queue_cond, NULL);
//工作队列初始化
queue_head = NULL;
queue_tail = NULL;
max_thread_num = max_thread_num;
cur_queue_size = 0;
shutdown = 0;
max_thread_num = DEFAULT_MAX_THREAD_NUM;
threadid = new pthread_t[max_thread_num];
int i = 0;
for (i = 0; i < max_thread_num; i++)
{
pthread_create(&(threadid[i]), NULL, thread_routine, (void*)this);
}
}
c_thread_pool::c_thread_pool(int max_thread_num)
{
pthread_mutex_init(&queue_lock, NULL);
pthread_cond_init(&queue_cond, NULL);
//工作队列初始化
queue_head = NULL;
queue_tail = NULL;
max_thread_num = max_thread_num;
cur_queue_size = 0;
threadid = new pthread_t[max_thread_num];
int i = 0;
for (i = 0; i < max_thread_num; i++)
{
pthread_create(&(threadid[i]), NULL, thread_routine, (void*)this);
}
}
/*向线程池中的任务队列加入任务*/
int c_thread_pool::add_work(c_work work)
{
c_work *newwork = new c_work;
newwork->process = work.process;
newwork->arg = work.arg;
newwork->next = NULL;
pthread_mutex_lock(&queue_lock);
/*将任务加入到等待队列中*/
if (queue_head != NULL && queue_tail != NULL)
{
queue_tail->next = newwork;
queue_tail = newwork;
}
else
{
//空队列
queue_head = newwork;
queue_tail = newwork;
}
cur_queue_size++;
pthread_mutex_unlock(&queue_lock);
/*等待队列中有任务了，唤醒一个等待线程,注意如果所有线程都在忙碌，这句没有任何作用*/
pthread_cond_signal(&(queue_cond));
printf("add work returned!\n");
return 0;
}
void* c_thread_pool::thread_routine(void *arg)
{
c_thread_pool *pool = (c_thread_pool *)arg;
int i = 0;
while (1)
{
pthread_mutex_lock(&(pool->queue_lock));
//如果等待队列为0并且不销毁线程池，则处于阻塞状态; 注意
// pthread_cond_wait是一个原子操作，等待前会解锁，唤醒后会加锁
//标注：注意这一如果任务队列不为空的话，while语句将被跳过，直接执行下面的调用。
while (pool->cur_queue_size == 0 && pool->shutdown)
{
pthread_cond_wait(&(pool->queue_cond), &(pool->queue_lock));
}
//等待队列长度减去1，并取出链表中的头元素
if (pool->cur_queue_size > 0 && pool->queue_head != NULL)
{
printf("IN THREAD ROUTINE size = %d && queue head is not NULL\n", pool->cur_queue_size);
pool->cur_queue_size--;
c_work *work = pool->queue_head;
pool->queue_head = work->next;
pthread_mutex_unlock(&(pool->queue_lock));
//调用回调函数，执行测试任务
//////////////////////////////////////////
(*(work->process))(work->arg);
free(work);
work = NULL;
}
else //不可达
{
pthread_mutex_unlock(&(pool->queue_lock));
}
}
}
c_thread_pool::~c_thread_pool()
{
for (int i = 0; i < max_thread_num; ++i)
pthread_cancel(threadid[i]);
for (c_work *w_t = queue_head; w_t != NULL;)
{
c_work *temp = w_t->next;
delete w_t;
w_t = temp;
}
delete [] threadid;
}

/////////////////////////////////////////////////////////线程池类 ///////////////////////////////////////////////////////#include <pthread.h>#include <stdlib.h>#include <stdio.h>#include <unistd.h>#include <assert.h>const int DEFAULT_MAX_THREAD_NUM = 10;const int MAX_WORK_NUM = 100000;//c_worker类class c_work{public:c_work():process(NULL), arg(NULL), next(NULL){}c_work(void *(*prcss)(void *), void *arg):process(prcss), arg(arg), next(NULL) {}~c_work(); void *(*process)(void *);void *arg;unsigned char type; //最高位表示arg是否需要delete操作c_work *next;};c_work::~c_work(){unsigned char ifdel = type >> 7;if (ifdel){delete arg;arg = NULL;}}class c_thread_pool{public:c_thread_pool();c_thread_pool(const int max_thread_num);~c_thread_pool();int add_work(c_work work);static void *thread_routine(void *arg);pthread_mutex_t queue_lock;pthread_cond_t queue_cond;//private:c_work *queue_head;c_work *queue_tail;int shutdown;pthread_t *threadid;int max_thread_num;int cur_queue_size;};c_thread_pool::c_thread_pool(){pthread_mutex_init(&queue_lock, NULL);pthread_cond_init(&queue_cond, NULL);//工作队列初始化queue_head = NULL;queue_tail = NULL;max_thread_num = max_thread_num;cur_queue_size = 0;shutdown = 0;max_thread_num = DEFAULT_MAX_THREAD_NUM;threadid = new pthread_t[max_thread_num];int i = 0;for (i = 0; i < max_thread_num; i++){pthread_create(&(threadid[i]), NULL, thread_routine, (void*)this);}}c_thread_pool::c_thread_pool(int max_thread_num){pthread_mutex_init(&queue_lock, NULL);pthread_cond_init(&queue_cond, NULL);//工作队列初始化queue_head = NULL;queue_tail = NULL;max_thread_num = max_thread_num;cur_queue_size = 0;threadid = new pthread_t[max_thread_num];int i = 0;for (i = 0; i < max_thread_num; i++){pthread_create(&(threadid[i]), NULL, thread_routine, (void*)this);}}/*向线程池中的任务队列加入任务*/int c_thread_pool::add_work(c_work work){c_work *newwork = new c_work;newwork->process = work.process;newwork->arg = work.arg;newwork->next = NULL;pthread_mutex_lock(&queue_lock);/*将任务加入到等待队列中*/if (queue_head != NULL && queue_tail != NULL){queue_tail->next = newwork;queue_tail = newwork;}else{//空队列queue_head = newwork;queue_tail = newwork;}cur_queue_size++;pthread_mutex_unlock(&queue_lock);/*等待队列中有任务了，唤醒一个等待线程,注意如果所有线程都在忙碌，这句没有任何作用*/pthread_cond_signal(&(queue_cond)); printf("add work returned!\n");return 0;}void* c_thread_pool::thread_routine(void *arg){c_thread_pool *pool = (c_thread_pool *)arg;int i = 0;while (1){pthread_mutex_lock(&(pool->queue_lock));//如果等待队列为0并且不销毁线程池，则处于阻塞状态; 注意 // pthread_cond_wait是一个原子操作，等待前会解锁，唤醒后会加锁//标注：注意这一如果任务队列不为空的话，while语句将被跳过，直接执行下面的调用。while (pool->cur_queue_size == 0 && pool->shutdown){pthread_cond_wait(&(pool->queue_cond), &(pool->queue_lock));}//等待队列长度减去1，并取出链表中的头元素if (pool->cur_queue_size > 0 && pool->queue_head != NULL){printf("IN THREAD ROUTINE size = %d && queue head is not NULL\n", pool->cur_queue_size);pool->cur_queue_size--;c_work *work = pool->queue_head;pool->queue_head = work->next;pthread_mutex_unlock(&(pool->queue_lock));//调用回调函数，执行测试任务//////////////////////////////////////////(*(work->process))(work->arg);free(work);work = NULL;}else //不可达{pthread_mutex_unlock(&(pool->queue_lock));}}}c_thread_pool::~c_thread_pool(){for (int i = 0; i < max_thread_num; ++i)pthread_cancel(threadid[i]);for (c_work *w_t = queue_head; w_t != NULL;){c_work *temp = w_t->next;delete w_t;w_t = temp;}delete [] threadid;}