C实现一个简单的线程池

C实现一个简单的线程池2013-11-16

//threadpool.h#ifndef __THREADPOOL_H__#define __THREADPOOL_H__#include<pthread.h>typedef void* （*task_fun）（void*）;//用链表来维护等待任务typedef struct threadtask{//任务的执行函数task_fun task;//执行函数的参数void* arg;//下一节点struct threadtask* next;}THREAD_TASK;void* thr_fun（void* arg）;//每个线程执行的函数int init_task（void）;//初始化等待任务链表int init_pool（void）;//初始化线程池//向任务链表中添加任务int add_task（task_fun task, void* arg）;int destroy_poll（void）;#endif

//threadpool.c//对于基本不出现的错误不进行检测#include<stdio.h>#include<stdlib.h>#include<unistd.h>#include<string.h>#include"threadpool.h"//线程池中最多允许的线程数#define THREADPOOLSIZE 3//当前等待任务的个数static unsigned int taskwaittingnum=0;//taskwaittingnum的锁pthread_mutex_t g_nummutex= PTHREAD_MUTEX_INITIALIZER;//等待任务队列的锁pthread_mutex_t g_taskmutex=PTHREAD_MUTEX_INITIALIZER;THREAD_TASK* g_pTask=NULL;pthread_t tid[THREADPOOLSIZE];//是否销毁线程池static int isShut=0;void* thr_fun（void* arg）{task_fun task;void* funarg=NULL;THREAD_TASK* ptmp=NULL;while（1）{//如果要销毁线程池，跳出while循环if（ 1==isShut ） {break;//跳出while循环}pthread_mutex_lock（&g_nummutex）;//如果当前没有任务需要调度，休眠5000微妙if（ 0==taskwaittingnum ）{pthread_mutex_unlock（&g_nummutex）;usleep（5000）;continue;}//当前有任务调度else{//需要写入链表，所以要锁定链表pthread_mutex_lock（&g_taskmutex）;task=g_pTask->next->task;funarg=g_pTask->next->arg;ptmp=g_pTask->next;g_pTask->next=ptmp->next;taskwaittingnum--;pthread_mutex_unlock（&g_taskmutex）;pthread_mutex_unlock（&g_nummutex）;free（ptmp）;ptmp=NULL;（*task）（funarg）;}}pthread_exit（NULL）;}//初始化任务链表，首个节点不用来存储任务int init_task（void）{g_pTask=（THREAD_TASK*）malloc（sizeof（THREAD_TASK））;if（ NULL==g_pTask ）{printf（"init_task malloc fails./n"）;return -1;}memset（g_pTask, 0, sizeof（THREAD_TASK））;g_pTask->next=NULL;return 0;}//初始化线程池int init_pool（void）{int ret;int i;for（ i=0 ; i<THREADPOOLSIZE ; i++ ）{ret=pthread_create（tid+i, NULL, thr_fun, NULL）;if（ ret！=0 ）{printf（"init_pool pthread_create:/n%s/n", strerror（ret））;return -1;}}return 0; }int add_task（task_fun task, void* arg）{THREAD_TASK* ptmp=NULL;pthread_mutex_lock（&g_nummutex）;pthread_mutex_lock（&g_taskmutex）;ptmp=g_pTask;while（ ptmp->next！=NULL ）{ptmp=ptmp->next;}ptmp->next=（THREAD_TASK*）malloc（sizeof（THREAD_TASK））;if（ NULL==ptmp->next ）{printf（"add_task malloc fails/n"）;return -1;}ptmp=ptmp->next;ptmp->task=task;ptmp->arg=arg;ptmp->next=NULL;taskwaittingnum++;pthread_mutex_unlock（&g_nummutex）;pthread_mutex_unlock（&g_taskmutex）;return 0;}int destroy_pool（void）{isShut=1;pthread_mutex_destroy（&g_nummutex）;pthread_mutex_destroy（&g_taskmutex）;return 0;}

//用来测试的main.c#include<stdio.h>#include"threadpool.h"void* task1（void* arg）{printf（"task1正在运行！/n"）;sleep（10）;printf（"task1运行结束！/n"）;return NULL;}void* task2（void* arg）{printf（"task2正在运行！/n"）;sleep（10）;printf（"task2运行结束！/n"）;return NULL;}void* task3（void* arg）{printf（"task3正在运行！/n"）;printf（"task3运行结束！/n"）;return NULL;}void* task4（void* arg）{printf（"task4正在运行！/n"）;printf（"task4运行结束！/n"）;return NULL;}int main（int argc, char *argv[]）{init_task（）;init_pool（）;add_task（task1, NULL）;add_task（task2, NULL）;add_task（task3, NULL）;add_task（task4, NULL）;sleep（30）;destroy_pool（）;return 0;}