线程池实例

/************************************
文件名称：pthread_pool.h
文件功能：线程池的变量定义和函数声明
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/
#ifndef _THREAD_H_
#define _THREAD_H_
#include <stdio.h> 
#include <stdlib.h> 
#include <unistd.h> 
#include <sys/types.h> 
#include <pthread.h> 
typedef struct task 
{ 
    void *(*process) (void* arg); /*线程的任务函数*/
    void * arg;    /* 传入任务函数的参数 */
    struct task *next; /*任务链表的指针域*/
} Cthread_task; 




/*线程池结构*/ 
typedef struct 
{ 
    pthread_mutex_t queue_lock; /*线程互斥锁(防止同时访问共享资源) */
    pthread_cond_t queue_ready; /*线程条件变量(线程同步机制)*/
    Cthread_task *queue_head;    /*链表结构，线程池中所有等待任务*/ 
  
    int shutdown;   /*是否销毁线程池*/ 
    pthread_t *threadid; 
    int max_thread_num;   /*线程池中线程数目*/ 
    
    int cur_task_size; /*当前等待的任务数*/ 
} Cthread_pool; 




static Cthread_pool *pool = NULL; 
/*任务处理函数*/
void *thread_routine (void *arg); 


void * rut_process(int arg);
/*线程池的创建和初始化*/
void pool_init (int max_thread_num);
/*往线程池中添加任务*/
int pool_add_task (void *(*process) (int arg), int arg);
int pool_destroy();
#endif




/************************************
文件名称：pthread_pool.c
文件功能：头文件函数pthread_pool.h实现
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/


#include "thread.h"
/************************************
函数名称：pool_init()
函数功能：线程池的创建并初始化
参    数：
 max_thread_num：可容纳的最大线程数
返回  值：void
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/
void pool_init (int max_thread_num) 
{ 
    int i = 0;
    /*分配 Cthread_pool 结构体空间*/
    pool = (Cthread_pool *) malloc (sizeof (Cthread_pool)); 


    pthread_mutex_init (&(pool->queue_lock), NULL); 
    /*初始化条件变量*/
    pthread_cond_init (&(pool->queue_ready), NULL); 
    
    pool->queue_head = NULL; 


     /*线程池中线程数目*/ 
    pool->max_thread_num = max_thread_num; 
    /*当前等待的任务数*/ 
    pool->cur_task_size = 0; 


   /*是否销毁线程池*/ 
    pool->shutdown = 0; 


    /*动态的分配长度为max_thread_num * sizeof (pthread_t)内存块，并将其强制转换为(pthread_t *)*/
    pool->threadid = (pthread_t *) malloc (max_thread_num * sizeof (pthread_t)); 


    //创建max_thread_num线程
    for (i = 0; i < max_thread_num; i++) 
    {  
        pthread_create (&(pool->threadid[i]), NULL, thread_routine, NULL); 
    } 
} 
/************************************
函数名称：pool_add_task()
函数功能：往线程池中添加任务
参    数：
 process：线程的任务函数
 arg：任务函数携带的参数
返回  值：0
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/


int pool_add_task (void *(*process) (void*), void* arg) 
{ 
    /*构造一个新任务*/ 
    Cthread_task *task = (Cthread_task *) malloc (sizeof (Cthread_task)); 
    /*给Cthread_task结构体赋值*/
    task->process = process; 
    task->arg = arg; 
    task->next = NULL;


     /*处理共享资源时需要枷锁*/
    pthread_mutex_lock (&(pool->queue_lock)); 
    /*将任务加入到等待队列中*/ 
    Cthread_task *member = pool->queue_head; 
    if (member != NULL) 
    { 
        while (member->next != NULL) 
            member = member->next; 
        member->next = task; 
    } 
    else 
    { 
        pool->queue_head = task; 
    } 




    pool->cur_task_size++; 
    pthread_mutex_unlock (&(pool->queue_lock)); 
    
    pthread_cond_signal (&(pool->queue_ready)); 
    
    return 0; 
} 
/************************************
函数名称：pool_destroy()
函数功能：销毁线程池，等待队列中的任务不会再被执行，
 但是正在运行的线程会一直 把任务运行完后再退出
参    数：void
返回  值：0
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/


int pool_destroy () 
{ 
    if (pool->shutdown) 
        return -1;/*防止两次调用*/ 
    pool->shutdown = 1; 




    /*唤醒所有等待线程，线程池要销毁了*/ 
    pthread_cond_broadcast (&(pool->queue_ready)); 




    /*阻塞等待线程退出，否则就成僵尸了*/ 
    int i; 
    for (i = 0; i < pool->max_thread_num; i++) 
        pthread_join (pool->threadid[i], NULL); 
    free (pool->threadid); 




    /*销毁等待队列*/ 
    Cthread_task *head = NULL; 
    while (pool->queue_head != NULL) 
    { 
        head = pool->queue_head; 
        pool->queue_head = pool->queue_head->next; 
        free (head); 
    } 
    /*条件变量和互斥量也别忘了销毁*/ 
    pthread_mutex_destroy(&(pool->queue_lock)); 
    pthread_cond_destroy(&(pool->queue_ready)); 
     
    free (pool); 
    /*销毁后指针置空是个好习惯*/ 
    pool=NULL; 
    return 0; 
} 
/************************************
函数名称：thread_routine()
函数功能：线程的处理函数
参    数：void
返回  值：0
作    者：王治博
创建日期：2016-12-23
修改日期：无
修改原因：无
********************************************/
void * thread_routine (void *arg) 
{ 
    printf ("starting thread 0x%x\n", pthread_self ()); 
    while (1) 
    { 
        pthread_mutex_lock (&(pool->queue_lock)); 




        while (pool->cur_task_size == 0 && !pool->shutdown) 
        { 
            printf ("thread 0x%x is waiting\n", pthread_self ()); 
            pthread_cond_wait (&(pool->queue_ready), &(pool->queue_lock)); 
        } 




        /*线程池要销毁了*/ 
        if (pool->shutdown) 
        { 
            /*遇到break,continue,return等跳转语句，千万不要忘记先解锁*/ 
            pthread_mutex_unlock (&(pool->queue_lock)); 
            printf ("thread 0x%x will exit\n", pthread_self ()); 
            pthread_exit (NULL); 
        } 




        printf ("thread 0x%x is starting to work\n", pthread_self ()); 




         
        /*待处理任务减1，并取出链表中的头元素*/ 
        pool->cur_task_size--; 
        Cthread_task *task = pool->queue_head; 
        pool->queue_head = task->next; 
        pthread_mutex_unlock (&(pool->queue_lock)); 




        /*调用回调函数，执行任务*/ 
        (*(task->process)) (task->arg); 
        free (task); 
        task = NULL; 
    } 
    /*这一句应该是不可达的*/ 
    pthread_exit (NULL); 
}