C++线程池的简单实现方法

 更新时间:2014年09月05日 11:01:12   投稿:shichen2014  
这篇文章主要介绍了C++线程池的简单实现方法,包括了线程操作函数及相关属性的用法,需要的朋友可以参考下

本文以实例形式较为详细的讲述了C++线程池的简单实现方法。分享给大家供大家参考之用。具体方法如下:

一、几个基本的线程函数:

1.线程操纵函数:

int pthread_create(pthread_t *tidp, const pthread_attr_t *attr, (void*)(*start_rtn)(void *), void *arg); //创建
void pthread_exit(void *retval);            //终止自身
int pthread_cancel(pthread_t tid);            //终止其他.发送终止信号后目标线程不一定终止,要调用join函数等待
int pthread_join(pthread_t tid, void **retval);   //阻塞并等待其他线程

2.属性:

int pthread_attr_init(pthread_attr_t *attr);           //初始化属性
int pthread_attr_setdetachstate(pthread_attr_t *attr, int detachstate); //设置分离状态
int pthread_attr_destroy(pthread_attr_t *attr);           //销毁属性

 

3.同步函数
互斥锁

int pthread_mutex_init(pthread_mutex_t *restrict mutex, const pthread_mutexattr_t *restrict attr); //初始化锁
int pthread_mutex_destroy(pthread_mutex_t *mutex); //销毁锁
int pthread_mutex_lock(pthread_mutex_t *mutex); //加锁
int pthread_mutex_trylock(pthread_mutex_t *mutex); //尝试加锁,上面lock的非阻塞版本
int pthread_mutex_unlock(pthread_mutex_t *mutex); //解锁

4.条件变量

int pthread_cond_init(pthread_cond_t *cv, const pthread_condattr_t *cattr); //初始化
int pthread_cond_destroy(pthread_cond_t *cond);                 //销毁 
int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex);     //等待条件
int pthread_cond_signal(pthread_cond_t *cond);                 //通知,唤醒第一个调用pthread_cond_wait()而进入睡眠的线程

5.工具函数

int pthread_equal(pthread_t t1, pthread_t t2); //比较线程ID
int pthread_detach(pthread_t tid);       //分离线程
pthread_t pthread_self(void);            //自身ID

上述代码中,线程的cancel和join,以及最后的工具函数,这些函数的参数都为结构体变量,其他的函数参数都是结构体变量指针;品味一下,参数为指针的,因为都需要改变结构体的内容,而参数为普通变量的,则只需要读内容即可。

二、线程池代码:

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>  //linux环境中多线程的头文件,非C语言标准库,编译时最后要加 -lpthread 调用动态链接库

//工作链表的结构
typedef struct worker {
  void *(*process)(void *arg);  //工作函数
  void *arg;           //函数的参数
  struct worker *next;
}CThread_worker;

//线程池的结构
typedef struct {
  pthread_mutex_t queue_lock;   //互斥锁
  pthread_cond_t queue_ready;  //条件变量/信号量

  CThread_worker *queue_head;   //指向工作链表的头结点,临界区
  int cur_queue_size;       //记录链表中工作的数量,临界区

  int max_thread_num;       //最大线程数
  pthread_t *threadid;      //线程ID

  int shutdown;          //开关
}CThread_pool;

static CThread_pool *pool = NULL;  //一个线程池变量
int pool_add_worker(void *(*process)(void *arg), void *arg);  //负责向工作链表中添加工作
void *thread_routine(void *arg);  //线程例程

//线程池初始化
void pool_init(int max_thread_num)
{
  int i = 0;

  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->cur_queue_size = 0;

  pool->max_thread_num = max_thread_num;
  pool->threadid = (pthread_t *) malloc(max_thread_num * sizeof(pthread_t));
  for (i = 0; i < max_thread_num; i++) {
    pthread_create(&(pool->threadid[i]), NULL, thread_routine, NULL); //创建线程, 参数为线程ID变量地址、属性、例程、参数
  }

  pool->shutdown = 0;
}

//例程,调用具体的工作函数
void *thread_routine(void *arg)
{
  printf("starting thread 0x%x\n", (int)pthread_self());
  while(1) {
    pthread_mutex_lock(&(pool->queue_lock));  //从工作链表中取工作,要先加互斥锁,参数为锁地址

    while(pool->cur_queue_size == 0 && !pool->shutdown) {    //链表为空
      printf("thread 0x%x is waiting\n", (int)pthread_self());
      pthread_cond_wait(&(pool->queue_ready), &(pool->queue_lock));  //等待资源,信号量用于通知。会释放第二个参数的锁,以供添加;函数返回时重新加锁。
    }

    if(pool->shutdown) {
      pthread_mutex_unlock(&(pool->queue_lock));     //结束开关开启,释放锁并退出线程
      printf("thread 0x%x will exit\n", (int)pthread_self());
      pthread_exit(NULL);   //参数为void *
    }

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

    --pool->cur_queue_size;
    CThread_worker *worker = pool->queue_head;
    pool->queue_head = worker->next;

    pthread_mutex_unlock (&(pool->queue_lock));   //获取一个工作后释放锁
    (*(worker->process))(worker->arg);   //做工作
    free(worker);
    worker = NULL;
  }
  pthread_exit(NULL);
}

//销毁线程池
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);   //主线程等待所有线程退出,只有join第一个参数不是指针,第二个参数类型是void **,接收exit的返回值,需要强制转换
  free(pool->threadid);
  CThread_worker *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;
}

void *myprocess(void *arg)
{
  printf("threadid is 0x%x, working on task %d\n", (int)pthread_self(), *(int*)arg);
  sleep (1);
  return NULL;
}

//添加工作
int pool_add_worker(void *(*process)(void *arg), void *arg)
{
  CThread_worker *newworker = (CThread_worker *) malloc(sizeof(CThread_worker));
  newworker->process = process;  //具体的工作函数
  newworker->arg = arg;
  newworker->next = NULL;

  pthread_mutex_lock( &(pool->queue_lock) );   //加锁

  CThread_worker *member = pool->queue_head;   //插入链表尾部
  if( member != NULL ) {
    while( member->next != NULL )
      member = member->next;
    member->next = newworker;
  }
  else {
    pool->queue_head = newworker;
  }
  ++pool->cur_queue_size;

  pthread_mutex_unlock( &(pool->queue_lock) );  //解锁

  pthread_cond_signal( &(pool->queue_ready) );  //通知一个等待的线程
  return 0;
}

int main(int argc, char **argv)
{
  pool_init(3);  //主线程创建线程池,3个线程

  int *workingnum = (int *) malloc(sizeof(int) * 10);
  int i;
  for(i = 0; i < 10; ++i) {
    workingnum[i] = i;
    pool_add_worker(myprocess, &workingnum[i]);   //主线程负责添加工作,10个工作
  }

  sleep (5);
  pool_destroy();   //销毁线程池
  free (workingnum);

  return 0;
}

希望本文所述对大家的C++程序设计有所帮助。

相关文章

  • C语言之实现辗转相除法的两种方式

    C语言之实现辗转相除法的两种方式

    这篇文章主要介绍了C语言之实现辗转相除法的两种方式,具有很好的参考价值,希望对大家有所帮助,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-08-08
  • C++ Vector用法深入剖析

    C++ Vector用法深入剖析

    C++ Vector应用方式是比较特殊的,我们将会在这篇文章中针对于它的应用方式进行一个详细的介绍,希望大家能充分掌握这一应用技巧
    2014-08-08
  • C++如何调用opencv完成运动目标捕捉详解

    C++如何调用opencv完成运动目标捕捉详解

    OpenCV作为机器视觉开源库,使用起来非常不错,这篇文章主要给大家介绍了关于C++如何调用opencv完成运动目标捕捉的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-05-05
  • C++获取当前系统时间的方法总结

    C++获取当前系统时间的方法总结

    这篇文章主要介绍了C++获取当前系统时间的方法,实例总结了四个获取系统时间的方法,非常具有实用价值,需要的朋友可以参考下
    2015-04-04
  • Java C++ 算法leetcode828统计子串中唯一字符乘法原理

    Java C++ 算法leetcode828统计子串中唯一字符乘法原理

    这篇文章主要为大家介绍了Java C++ 算法leetcode828统计子串中唯一字符乘法原理详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-09-09
  • Lua和C/C++互相调用实例分析

    Lua和C/C++互相调用实例分析

    今天小编就为大家分享一篇关于Lua和C/C++互相调用实例分析,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-01-01
  • STL list链表的用法详细解析

    STL list链表的用法详细解析

    这篇文章是关于C++语言的一个新的扩展——标准模板库的(Standard Template Library),也叫STL
    2013-09-09
  • 基于C++ list中erase与remove函数的使用详解

    基于C++ list中erase与remove函数的使用详解

    本篇文章是对C++ list中erase与remove函数的使用进行了详细的分析介绍,需要的朋友参考下
    2013-05-05
  • C语言实现大学生考勤管理系统

    C语言实现大学生考勤管理系统

    这篇文章主要为大家详细介绍了C语言实现大学生考勤管理系统,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2020-12-12
  • C++ Thread实现简单的socket多线程通信

    C++ Thread实现简单的socket多线程通信

    本文主要介绍了C++ Thread实现简单的socket多线程通信,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-07-07

最新评论