訊號量實現生產者消費者(程式碼邏輯有問題,不適合多個消費者,不常用)

ajdjirkdnkfk發表於2020-11-26

訊號量實現生產者消費者(程式碼邏輯有問題,不適合多個消費者,不常用)

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>
#include <sys/types.h>
#include <string.h>
#include <semaphore.h>
struct Node{
	int val;
	Node* next;
};
sem_t sem_producer; // 生產者的訊號量
sem_t sem_consumer; // 消費者的訊號量
Node* head = NULL;
void* producer(void* argv){
	while(1){
		Node* p = (Node*)malloc(sizeof(Node));
		p->val = rand() % 1000;
        // sem_wait做--操作,初始化的時候我們給與他值,意思是他可以生產幾個,當生產滿了訊號量變為0,阻塞
		sem_wait(&sem_producer);
		printf("sheng chan le %d\n", p->val);
		p->next = head;
		head = p;
        // sem_post做++操作,通知消費者有東西可以拿
		sem_post(&sem_consumer);
        
		sleep(1);
	}
}

void* consumer(void* argv){
	Node* p = NULL; 
	while(1){
        // sem_consumer為0的時候阻塞,被通知的時候做--操作去拿
		sem_wait(&sem_consumer);
		printf("xiao fei le %d\n", head->val);
		p = head;
		head = head->next;
        // 拿完之後告訴生產者可以繼續生產了
		sem_post(&sem_producer);
        
		free(p);
		p = NULL;
	}
}
int main()
{
    //訊號量初始化/初始化的訊號量/線上程中使用/初始化個數
	sem_init(&sem_producer, 0, 5); 
	sem_init(&sem_consumer, 0, 0);

	pthread_t thread1;
	pthread_create(&thread1, NULL, consumer, NULL);

	pthread_t thread2;
	pthread_create(&thread2, NULL, producer, NULL);

	pthread_join(thread1, NULL);
	pthread_join(thread2, NULL);
    //銷燬訊號量
	sem_destroy(&sem_producer);
	sem_destroy(&sem_consumer);
	return 0;
}

相關文章