VB线程池.DOCVIP

#include #include #include #include #include #include #include #include #define SIZE 50sem_t sem1,sem2,sem3; //sem1:线程个数,sem2:要执行任务个数,sem3 排队任务总数 struct process //任务结构 {int (*process_fun)(const char *); //任务函数 char process_arg[SIZE]; //任务参数 };struct pthreadnode //线程结构 {struct process *pth_pt; //线程任务 pthread_t pth_id; //线程ID int pth_flag; //线程使用标志 pthread_mutex_t pth_mutex; //线程锁 };struct lake //线程池 {struct process *lake_prohead; //任务排队列表,这里申请为数组头结点 int w; //w 后续任务插入位置,r为将要执行任务位置 int r; int prono; //排队任务的个数,即lake_prohead数组大小 struct pthreadnode *lake_pthhead; //线程列表, 这里申请为数组头结点int lake_pthno; //线程个数 pthread_t lake_id; //线程池 id };/*------------任务添加函数---------*/ /* T 为线程池指针process_fun 为任务函数 process_arg 任务参数 */ void fun_add(struct lake *T,int (*process_fun)(const char *),const char *process_arg){(T-lake_prohead+T-w)-process_fun=process_fun; memcpy((T-lake_prohead+T-w)-process_arg,process_arg,SIZE);++(T-w); if(T-w == T-prono){T-w=0; } }/*------------线程清理函数---------*/ void clean(void *p){pthread_mutex_unlock((pthread_mutex_t *)p);}/*------------线程执行函数---------*/ void *pthread_fun(void *arg){struct pthreadnode *tmp=(struct pthreadnode *)arg; pthread_cleanup_push(clean,tmp-pth_mutex);while(1){pthread_mutex_lock(tmp-pth_mutex); //阻塞,等待任务到来 printf(%x do %s\n,pthread_self(),tmp-pth_pt-process_arg);tmp-pth_pt-process_fun(tmp-pth_pt-process_arg); //执行任务 tmp-pth_flag=0;sem_post(sem3); //任务完成,可以空出一个等待任务栏 sem_post(sem1); //线程使用完毕,空出一个线程 }pthread_cleanup_pop(1);pthread_exit(0);}/*------------线程执行函数---------*/ void *lake_goon(void *p){int i=0;struct lake *T=(struct lake *)p;while(1){sem_wait(sem1); //等待线程是否有空的,无就等待 sem_wait(sem2); //等待是否有要执行的函数,无就等待 for(i=0;ilake_pthno;++i) //当有线程空有执行函数时 ,查找空闲线程 {if((T-lake_pthhead+i)-pth_flag == 0){break;}}(T-lake_pthhead+i)-pth_flag=1; //找到空闲线程并置1 (T-lake_pthhead+i)-pth_pt=(T-lake_prohead+T-r); //线程任务 指向任务排队列表,r所在任务 ++ (T-r); //任务下个 if(T-r == T-prono) {