线程池的简单实现

/*pthreadpool.h*/

#ifndef PTHREADPOOL
2 #define PTHREADPOOL
3 #include<pthread.h>
4
5 struct task_node{
6
7 void *(*run)(void *arg);//一个函数指针
8 void *arg;//函数指针的参数
9 struct task_node *next;
10 };
11
12 typedef struct task_node task_node;
13
14 struct pthreadpool{
15 task_node *head;//任务队列的头指针
16 task_node *top;//任务队列的尾指针
17
18 int max_pthread;//线程池创建线程的最大数
19
20 int pthread_num;//线程的数目
21 int pthread_wait_num;//等待线程的数目
22 int pthread_work_num;//工作线程的数目
23
24 pthread_mutex_t mutex;// 用于保护任务队列的互斥锁
25 pthread_cond_t cond;// 一个条件变量
26 };
27 typedef struct pthreadpool pthreadpool;
28
29 //线程池的初始化
30 void pthreadpool_init(pthreadpool *pool,int maxthread);
31 //向线程池添加任务
32 void pthreadpool_addtask(pthreadpool *pool,void*(*run)(void *arg),voi d *arg);
33 //销毁线程池
34 void pthreadpool_destory(pthreadpool * pool);
35
36

/*pthreadpool.c*/

1 #include"pthreadpool.h"
2 #include<stdlib.h>
3 #include<stdio.h>
4 void *mythread(void *arg)
5 {
6 pthreadpool *pool=(pthreadpool *)arg;
7 while(1)
8 {
9 printf("%uwait.....\n",pthread_self());
10 pool->pthread_wait_num++;
13 {
14 pthread_cond_wait(&pool->cond,&pool->mutex);
15
18
19 //取得任
20 task_node *head=pool->head;
21 pool->head=pool->head->next;
23 pthread_mutex_unlock(&pool->mutex);
24 printf("%uwork....\n",pthread_self());
25 head->run(head->arg);
26 free(head);
27
28
29 }
30 }
31
32
33 void pthreadpool_init(pthreadpool *pool,int maxthread)
34 {
35 pool->head=NULL;
36 pool->top=NULL;
37 pool->max_pthread=maxthread;
40 pool->pthread_work_num=0;
41
42 pthread_mutex_init(&pool->mutex,NULL);
45
46
48 {
49 //创建一个任务
50 task_node *node=malloc(sizeof(task_node));
1 #include"pthreadpool.h"
2 #include<stdlib.h>
3 #include<stdio.h>
4 void *mythread(void *arg)
5 {
6 pthreadpool *pool=(pthreadpool *)arg;
1 #include"pthreadpool.h"
2 #include<stdlib.h>
3 #include<stdio.h>
4 void *mythread(void *arg)
5 {
6 pthreadpool *pool=(pthreadpool *)arg;
7 while(1)
8 {
9 printf("%uwait.....\n",pthread_self());
10 pool->pthread_wait_num++;
11 pthread_mutex_lock(&pool->mutex);
12 while( pool->head==NULL)
13 {
14 pthread_cond_wait(&pool->cond,&pool->mutex);
15
16 }
17 pool->pthread_wait_num--;
18
19 //取得任
20 task_node *head=pool->head;
21 pool->head=pool->head->next;
22 //这里应该解锁因为任务一伙的防止在执行任务的时候其他线程不能获得任务
23 pthread_mutex_unlock(&pool->mutex);
24 printf("%uwork....\n",pthread_self());
25 head->run(head->arg);
26 free(head);
27 sleep(1);
28
29
30 }
31 }
32
33
34 void pthreadpool_init(pthreadpool *pool,int maxthread)
35 {
36 pool->head=NULL;
37 pool->top=NULL;
38 pool->max_pthread=maxthread;
1 #include"pthreadpool.h"
2 #include<stdlib.h>
3 #include<stdio.h>
4 void *mythread(void *arg)
5 {
6 pthreadpool *pool=(pthreadpool *)arg;
7 while(1)
8 {
9 printf("%uwait.....\n",pthread_self());
10 pool->pthread_wait_num++;
11 pthread_mutex_lock(&pool->mutex);
12 while( pool->head==NULL)
13 {
14 pthread_cond_wait(&pool->cond,&pool->mutex);
15
16 }
17 pool->pthread_wait_num--;
18
19 //取得任
20 task_node *head=pool->head;
21 pool->head=pool->head->next;
22 //这里应该解锁因为任务一伙的防止在执行任务的时候其他线程不能获得任务
23 pthread_mutex_unlock(&pool->mutex);
24 printf("%uwork....\n",pthread_self());
25 head->run(head->arg);
26 free(head);
27 sleep(1);
28
29
30 }
31 }
32
33
34 void pthreadpool_init(pthreadpool *pool,int maxthread)
26 free(head);
27 sleep(1);
28
29
30 }
31 }
32
33
34 void pthreadpool_init(pthreadpool *pool,int maxthread)
35 {
36 pool->head=NULL;
37 pool->top=NULL;
38 pool->max_pthread=maxthread;
39 pool->pthread_num=0;
40 pool->pthread_wait_num=0;
41 pool->pthread_work_num=0;
42
43 pthread_mutex_init(&pool->mutex,NULL);
44 pthread_cond_init(&pool->cond,NULL);
45 }
46
47
48 void pthreadpool_addtask(pthreadpool *pool,void*(*run)(void *arg),void *arg)
49 {
50 //创建一个任务
51 task_node *node=malloc(sizeof(task_node));
52 node->run=run;
53 node->arg=arg;
54 node->next=NULL;
55
56 //添加到任务队列
57 pthread_mutex_lock(&pool->mutex);
58
59 if(pool->head==NULL)
60 {
61 pool->head=node;
62 pool->top=node;
63 }else{
64
65 pool->top->next=node;
66 pool->top=node;
67 }
68
69 //添加完任务判断是否有等待线程有唤醒没有创建
70 if(pool->pthread_wait_num>0)
71 {
72 pthread_cond_signal(&pool->cond);
73 }
74
75 if(pool->pthread_wait_num==0&&pool->pthread_num<pool->max_pthread)
76 {
77 printf("cread pthread...\n");
78 pthread_t pid;
79 pthread_create(&pid,NULL,mythread,(void *)pool);
80 pool->pthread_num++;
81 }
82
83 pthread_mutex_unlock(&pool->mutex);
84
85
86 }
87
88 void pthreadpool_destory(pthreadpool * pool)
89 {
90
91
92
93
94
95 }
测试代码

1 #include<stdio.h>
2 #include"pthreadpool.h"
3 #include<stdlib.h>
4 void * run(void *arg)
5 {
6 int num=*(int *)arg;
7 printf("task num %d:",num);
8 printf("hellow world\n");
9 return NULL;
10 }
11 int main()
12 {
13 int i=0;
14 pthreadpool pool;
15 pthreadpool_init(&pool,10);
16 for(int i=0;i<10;i++)
17 {
18 int *arg=malloc(sizeof(int));
19 *arg=i;
20 pthreadpool_addtask(&pool,run,arg);
21 sleep(5);
22 }
23
24
25 while(1);
26
27
28
29
30
31
32
33
34 }