生产者-消费者问题

生产者-消费者问题 是操作系统中经典的同步问题之一
主要为使用PV操作管理临界区，实现进程互斥
使用SP表示生产者是否可把物品放入缓冲区，缓冲区大小就是SP的初始值。SG表示缓冲区是否有物品，初始值为0

书中对该部分伪代码表示为

#include <stdio.h>  
#include <pthread.h>  
#include <semaphore.h>

// 定义信号量，参考书中SP、SG
sem_t sp,sg;
// 定义缓冲区大小
int buffer_size = 6;
// 定义缓冲区内物品数量 
int buffer_count;  
// 定义互斥锁
pthread_mutex_t mutex;
// 定义用于测试的总生产物品数
int test_num = 14;
// 定义生产和消费函数
void *producer(void *arg);
void *consumer(void *arg); 
  

int main() 
{  
    // 初始化 （sp初始值为缓冲区大小）
    if (sem_init(&sp, 0, buffer_size) != 0) {
        printf("初始化信号量失败");
        return -1;
    }
    if (sem_init(&sg, 0, 0) != 0) {
        printf("初始化信号量失败");
        return -1;
    }
    pthread_mutex_init(&mutex, NULL); 

    // 创建生产者和消费者线程
    pthread_t thread_producer, thread_consumer;
    if (pthread_create(&thread_consumer, NULL, consumer, NULL) != 0) {
        printf("创建消费者线程失败");
        return -1; 
    } 
    if (pthread_create(&thread_producer, NULL, producer, NULL) != 0) {
        printf("创建生产者线程失败");
        return -1; 
    }
    if (pthread_join(thread_consumer, NULL) != 0) {
        printf("等待消费者线程结束失败");
        return -1;
    }  
    if (pthread_join(thread_producer, NULL) != 0) {
        printf("等待生产者线程结束失败");
        return -1;
    }  

    // 销毁
    sem_destroy(&sg); 
    sem_destroy(&sp); 
    pthread_mutex_destroy(&mutex);
    return 0;  
}  
  
void *producer(void *arg){  
    for (int i = 0; i < test_num; i++) {
        sem_wait(&sp);  // 测试是否可以把物品放入缓冲区

        pthread_mutex_lock(&mutex);   //加锁  
        buffer_count++; // 将生产物品放入缓冲区
        printf("生产操作，当前缓冲区内物品数量: %d\n", buffer_count);  
        pthread_mutex_unlock(&mutex); //解锁

        sem_post(&sg);  // 发送缓冲区中有物品消息
    }
}  

void *consumer(void *arg ){  
    for (int i = 0; i < test_num; i++) {
        sem_wait(&sg); // 测试缓冲区中是否有物品

        pthread_mutex_lock(&mutex);   //加锁  
        buffer_count--; // 从缓冲区中取出一件物品
        printf("消费操作，当前缓冲区内物品数量: %d\n", buffer_count);
        pthread_mutex_unlock(&mutex); //解锁  

        sem_post(&sp);  // 发送缓冲区中可以再存物品的消息
    }
}