多线程编程作为程序员必须掌握的一项技能,看到一篇好的博客,适合初学者入门。以一个简单的、著名的生产者/消费者问题实例,比较清晰地阐述了linux线程的控制和通信。一组生产者线程与一组消费者线程通过缓冲区发生联系。生产者线程将生产的产品送入缓冲区,消费者线程则从中取出产品。缓冲区有N个,是一个环形的缓冲池。原文链接如下(链接),代码如下,看懂这部分代码就算入门Linux多线程了。
#include <stdio.h>
#include <pthread.h>
#define BUFFER_SIZE 16//缓冲区数量
struct prodcons
{
//缓冲区相关数据结构
int buffer[BUFFER_SIZE];//实际数据存放的数组
pthread_mutex_t lock;//互斥体lock用于对缓冲区的互斥操作
int readpos,writepos;//读写指针
pthread_cond_t notempty;//缓冲区非空的条件变量
pthread_cond_t notfull;//缓冲区未满的条件变量
};
//初始化缓冲区结构
void init(struct prodcons *b)
{
pthread_mutex_init(&b->lock,NULL);
pthread_cond_init(&b->notempty,NULL);
pthread_cond_init(&b->notfull,NULL);
b->readpos = 0;
b->writepos = 0;
}
//生产者将产品放入缓冲区,这里是存入一个整数
void put(struct prodcons *b,int data)
{
pthread_mutex_lock(&b->lock);
//等待缓冲区未满
if((b->writepos+1)%BUFFER_SIZE == b->readpos)
{
pthread_cond_wait(&b->notfull,&b->lock);//用来等待条件变量被设置
}
//写数据,并移动指针
b->buffer[b->writepos]=data;
b->writepos++;
if(b->writepos >= BUFFER_SIZE)
{
b->writepos = 0;
}
//设置缓冲区非空的条件变量
pthread_cond_signal(&b->notempty);//解除某一个等待线程的阻塞状态
pthread_mutex_unlock(&b->lock);//给互斥变量解锁
}
//消费者从缓冲区取出整数
int get(struct prodcons *b)
{
int data;
pthread_mutex_lock(&b->lock);
//等待缓冲区非空
if(b->writepos == b->readpos)
{
pthread_cond_wait(&b->notempty,&b->lock);
}
//读数据,移动读指针
data = b->buffer[b->readpos];
b->readpos++;
if(b->readpos >= BUFFER_SIZE)
{
b->readpos = 0;
}
//设置缓冲区未满的条件变量
pthread_cond_signal(&b->notfull);
pthread_mutex_unlock(&b->lock);
return data;
}
/*测试:生产者线程将1到10000的整数送入缓冲区,消费者线程从缓冲区中获取整数,两者都打印信息*/
#define OVER (-1)
struct prodcons buffer;
void *producer(void *data)
{
int n;
for(n=0;n<20;n++)
{
printf("%d---->\n",n);
put(&buffer,n);
}
put(&buffer,OVER);
return NULL;
}
void *consumer(void *data)
{
int d;
while(1)
{
d = get(&buffer);
if(d == OVER)
{
break;
}
printf("--->%d\n",d);
}
return NULL;
}
int main(void)
{
pthread_t th_a,th_b;
void *retval;
init(&buffer);
//创建生产者和消费者线程
pthread_create(&th_a,NULL,producer,0);
pthread_create(&th_b,NULL,consumer,0);
//等待两个线程结束
pthread_join(th_a,&retval);
pthread_join(th_b,&retval);
return 0;
}