使用互斥锁解决生产者与消费者问题

最新推荐文章于 2022-02-19 12:31:43 发布

WA说

最新推荐文章于 2022-02-19 12:31:43 发布

阅读量2k

点赞数

分类专栏： Linux Learning

本文链接：https://blog.csdn.net/zyw_ym_zone/article/details/10206223

版权

Linux Learning 专栏收录该内容

32 篇文章 0 订阅

订阅专栏

互斥锁:上锁与解锁

互斥锁指代相互排斥，是最基本的同步形式，它可用于保护临界取，以保证任何时刻只有一个线程在执行其中的代码。它可以用来同步一个进程内的各个线程。如果互斥锁存放在多个进程间共享的某个内存区中，它也可以用于这些进程间的同步。它实际上保护的是临界区中被操纵的数据，也就是保护由多个线程或多个进程分享的共享数据

使用方法

lock_the_mutex(...)

临界区

unlock_the_mutex(,,,)

在Posix中，互斥锁声明为具有pthread_mutex_t数据类型的变量

具体实现代码

#include <pthread.h>
#include <sys/types.h>
#include <stdio.h>
#include <stdlib.h>

#define MAXNITEMS  10000
#define MAXNTHREADS 20

int nitems;    //read-only by producer and consumer
struct 
{
	pthread_mutex_t  mutex;
	int buff[MAXNITEMS];
	int nput;
	int nval;
}shared =        //initialized staticly
		{
			PTHREAD_MUTEX_INITIALIZER
		};

void *producer(void *);
void *consumer(void *);
int min(int a, int b) { return a<b ? a : b; }

int main(int argc, char *argv[])
{
	int i, nthreads, count[MAXNTHREADS];
	pthread_t tid_produce[MAXNTHREADS], tid_consume;
	
	if(argc!=3)
	{
		printf("usage: ProCon <#items> <#threads>");
		exit(1);
	}
	nitems = min(atoi(argv[1]), MAXNITEMS);
	nthreads = min(atoi(argv[2]), MAXNTHREADS);
	
	pthread_setconcurrency(nthreads);
	for(i=0; i<nthreads; i++)
	{
		count[i] = 0;
		pthread_create(&tid_produce[i], NULL, producer, &count[i]);
	}
	//wait for all the producer threads
	for(i=0; i<nthreads; i++)
	{
		pthread_join(tid_produce[i], NULL);
		printf("count[%d] = %d\n", i, count[i]);
	}
	//start, then wait for the consumer thread
	pthread_create(&tid_consume, NULL, consumer, NULL);
	pthread_join(tid_consume, NULL);
	return 0;
}

void *producer(void * arg)
{
	for(; ; )
	{
		pthread_mutex_lock(&shared.mutex);
		if(shared.nput >= nitems)
		{
			pthread_mutex_unlock(&shared.mutex);
			return (NULL);		//array is full, we're done
		}
		shared.buff[shared.nput] = shared.nval;
		shared.nput++;
		shared.nval++;
		pthread_mutex_unlock(&shared.mutex);
		*((int *)arg) += 1;
	}
}

void *consumer(void *arg)
{
	int i;
	for(i=0; i<nitems; i++)
	{
		if(shared.buff[i]!=i)
		{	
			printf("buff[%d] = %d\n", i, shared.buff[i]);
		}
	}
	return NULL;
}