linux系统编程中的信号量--模拟生产者与消费者

/************************************************************
*FileName    : producer_and_customer.c
*description : This app demonstrates how to use the semaphore
*              solve the problem about the producer and customer.
*version     : 1.0 
*history     : none   
*************************************************************/
/**************************************************************
用多线程、信号量实现生产者和消费者的模拟，仓库容量为10，仓库
中开始有3件产品，消费者每3秒消费一件产品，生产者每两秒生产一个产
品，生产者和消费者不能同时进入仓库（需要互斥）.
***************************************************************/


#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <semaphore.h>
#include <phread.h>


#define msleep(x) usleep(x*1000)
#define PRODUCT_SPEED 3 //生产速度
#define COSTOM_SPEED  1 //消费速度
#define INIT_NUM      3 //仓库原有产品数
#define TOTAL_NUM     10//仓库容量


sem_t p_sem,c_sem,sh_sem;
int num = INIT_NUM;


void product(void)//生产产品
{
    sleep(PRODUCT_SPEED);
}


int add_to_lib()//添加产品到仓库
{
    num++;
    msleep(500);
    return num;
}


void consum()//消费
{
    sleep(CONSUM_SPEED);
}


int sub_from_lib()//从仓库中取产品
{
    num--;
    msleep(500);
    return num;
}


void *productor(void *arg)
{
    while(1)
    {
        sem_wait(&p_sem);//生产信号减一

        product();//生产延时

        sem_wait(&sh_sem);//用来互斥的信号
        printf("push into!tatol_num=%d\n",add_to_lib());
        sem_post(&sh_sem);

        sem_post(&c_sem);//消费者信号量加一
    }
}


void *consumer(void *arg)
{
    while(1)
    {
        sem_wait(&c_sem);//消费者信号量减一

        sem_wait(&sh_sem);
        printf("pop out!tatol_num=%d\n",sub_from_lib());
        sem_post(&sh_sem);

        sem_post(&p_sem);//生产者信号量加一

        consum();//消费延时
    }
}


int main(int argc, char *argv[])
{
    pthread_t tid1,tid2;

    sem_init(&p_sem,0,TOTAL_NUM-INIT_NUM);
    sem_init(&c_sem,0,INIT_NUM);
    sem_init(&sh_sem,0,1);

    pthread_create(&tid1,NULL,productor,NULL);
    pthread_create(&tid2,NULL,consumer,NULL);

    pthread_join(tid1,NULL);
    pthread_join(tid2,NULL);

    return 0;
}