信号灯

/*
 *Time:2010/9/27
 *Author:Software Engineering
 */
#include <stdio.h>
#include <string.h>
#include <pthread.h> //包含线程操作的一些函数

#include <semaphore.h> //包含信号灯操作的一些函数


#define N 4 //定义缓冲池的数量

#define P_COUNT 3 //定义生产者的数量

#define C_COUNT 3 //定义消费者的数量

int first = 0; //记录缓冲区的下标

char buffer[N][20]; //缓冲区的定义


pthread_mutex_t mutex; //互斥锁

sem_t full, empty; //分别表示缓冲池中的空区和满区的数量


void print()
{
    int i;
    for(i = 0; i < N; i++) {
        printf("%s\t", buffer[i]);
    }
    printf("\n");
}

void *producer()
{
    char *goods = "鸡蛋";
    int count = 8;
    do {
        sem_wait(&empty); //关灯操作

        pthread_mutex_lock(&mutex);
        strcpy(buffer[first++], goods);
        printf("%u生产者：", pthread_self());
        print();
        sleep(1);
        pthread_mutex_unlock(&mutex);
        sem_post(&full); //开灯操作

    } while(count-- > 0);
}

void *customer()
{
    int count = 8;
    do {
        sem_wait(&full); //关灯操作

        pthread_mutex_lock(&mutex);
        strcpy(buffer[--first], "0");
        printf("%u消费者：", pthread_self());
        print();
        sleep(1);
        pthread_mutex_unlock(&mutex);
        sem_post(&empty); //开灯操作

    } while(count-- > 0);
}

int main()
{
    pthread_t tid1, tid2;
    int i;

    //初始化缓冲池

    for(i = 0; i < N; i++) {
        strcpy(buffer[i], "0");
    }

    //创建信号灯

    sem_init(&empty, 0, N);
    sem_init(&full, 0, 0);

    //创建生产者线程和消费者线程

    for(i = 0; i < P_COUNT; i++) {
        pthread_create(&tid1, NULL, producer, NULL);
    }
    for(i = 0; i < C_COUNT; i++) {
        pthread_create(&tid2, NULL, customer, NULL);
    }

    pthread_exit(0);
}