linux下编程学习----- 线程同步之无名信号量

linux下编程学习----- 线程同步之无名信号量

这是一个书上的例子，逻辑是：一个线程生产一组数字1、2、3..... 等等，放入一个循环队列中。另一个线程去处理这些数字，达到的效果是把这些数字求和。还是看看代码吧。

#include <stdio.h>
#include <string.h>
#include <pthread.h>
#include <errno.h>
#include <semaphore.h>
#define BUFSIZE 4
#define NUMBER 8

int sum_of_number=0;
/* 可读 和 可写资源数*/
sem_t write_res_number;
sem_t read_res_number;

/* 循环队列 */
struct recycle_buffer{
int buffer[BUFSIZE];
int head,tail;
}re_buf;

/* 用于实现临界区的互斥锁，我们对其初始化*/
pthread_mutex_t buffer_mutex=PTHREAD_MUTEX_INITIALIZER;

static void *producer(void * arg)
{
int i;
for(i=0;i<=NUMBER;i++)
{
/* 减少可写的资源数 */
sem_wait(&write_res_number);
/* 进入互斥区 */
pthread_mutex_lock(&buffer_mutex);
/*将数据复制到缓冲区的尾部*/
re_buf.buffer[re_buf.tail]=i;
re_buf.tail=(re_buf.tail+1)%BUFSIZE;
printf("procuder %d write %d./n",pthread_self(),i);
/*离开互斥区*/
pthread_mutex_unlock(&buffer_mutex);
/*增加可读资源数*/
sem_post(&read_res_number);
}

/* 线程终止，如果有线程等待它们结束，则把NULL作为等待其结果的返回值*/
return NULL;
}

static void * consumer(void * arg)
{
int i,num;

for(i=0;i<=NUMBER;i++)
{
/* 减少可读资源数 */
sem_wait(&read_res_number);
/* 进入互斥区*/
pthread_mutex_lock(&buffer_mutex);
/* 从缓冲区的头部获取数据*/
num = re_buf.buffer[re_buf.head];
re_buf.head = (re_buf.head+1)%BUFSIZE;
printf("consumer %d read %d./n",pthread_self(),num);
/* 离开互斥区*/
pthread_mutex_unlock(&buffer_mutex);
sum_of_number+=num;
/* 增加客写资源数*/
sem_post(&write_res_number);
}
/* 线程终止，如果有线程等待它们结束，则把NULL作为等待其结果的返回值*/
return NULL;
}

int main(int argc,char ** argv)
{
/* 用于保存线程的线程号 */
pthread_t p_tid;
pthread_t c_tid;
int i;

re_buf.head=0;
re_buf.tail=0;
for(i=0;i<BUFSIZE;i++)
re_buf.buffer[i] =0;

/* 初始化可写资源数为循环队列的单元数 */
sem_init(&write_res_number,0,BUFSIZE); // 这里限定了可写的bufsize，当写线程写满buf时，会阻塞，等待读线程读取
/* 初始化可读资源数为0 */
sem_init(&read_res_number,0,0);

/* 创建两个线程，线程函数分别是 producer 和 consumer */
/* 这两个线程将使用系统的缺省的线程设置，如线程的堆栈大小、线程调度策略和相应的优先级等等*/
pthread_create(&p_tid,NULL,producer,NULL);
pthread_create(&c_tid,NULL,consumer,NULL);

/*等待两个线程完成退出*/
pthread_join(p_tid,NULL);
pthread_join(c_tid,NULL);
printf("The sun of number is %d/n",sum_of_number);
}

编译命令：gcc -g -o threadtest threadtest.c -lpthread

引用：http://shake863.javaeye.com/blog/181250