Implementing Producer-Consumer in Linux Environment

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

#define N1 3//Define 3 producers
#define N2 4//Define 4 consumers
#define M 10//Define 10 size buffers

int in = 0;
int out = 0;

int buff[M] = {0};//buffer size is 10

sem_t empty_sem;//Number of empty buffers
sem_t full_sem;//Number of full buffers
pthread_mutex_t mutex;//Mutex access buffer

int product_id = 0;
int consumer_id = 0;

int data;
FILE *fp;

void * product()
{
    int id = ++product_id;
    while(1)
    {
        sleep(1);
        sem_wait(&empty_sem);
        pthread_mutex_lock(&mutex);
        //if(feof(fp) != 0) fseek(fp, 0, SEEK_SET);
        if(fscanf(fp, "%d", &data)==EOF)
        {
            fseek(fp, 0, SEEK_SET);
            fscanf(fp, "%d", &data);
        }
        in = in % M;
        buff[in] = data;
        printf("Producter %d produce %d in position %d\n", id, buff[in], in);
        ++in;
        pthread_mutex_unlock(&mutex);
        sem_post(&full_sem);
    }
}


void *consume()
{
    int id = ++consumer_id;

    while(1)
    {
        sleep(1);
        sem_wait(&full_sem);
        pthread_mutex_lock(&mutex);

        out = out % M;
        printf("Consumer %d take product  %d in position %d\n", id, buff[out], out);
        buff[out] = 0;
        ++out;

        pthread_mutex_unlock(&mutex);
        sem_post(&empty_sem);

    }
}


intmain()
{
    pthread_t id1[N1];//Define producer thread
    pthread_t id2[N2];//Define the consumer thread

    int i;
    int ret1[N1];
    int ret2[N2];

    int ini1 = sem_init(&empty_sem, 0, M); //initialize the empty buffer to M(10)
    int ini2 = sem_init(&full_sem, 0, 0);//initialize full buffer 0
    if(ini1 && ini2 != 0)
    {
        printf("Sem init failed\n");
        exit(1);
    }

    int ini3 = pthread_mutex_init(&mutex, NULL);//Initialize mutex semaphore
    if(ini3 != 0 )
    {
        printf("mutex init failed\n");
        exit(1);
    }
    
    fp = fopen("/.data.txt", "r");//打开文件Data.txt
    if(fp == NULL) exit(1);

    for(i = 0; i < N1; i++)
    {
        ret1[i] = pthread_create(&id1[i], NULL, product, (void*)(&i));//Create a producer thread
        if(ret1[i] != 0)
        {
            printf("product%d creat failed\n", i);
            exit(1);
        }
    }

    for(i = 0; i< N2; i++)
    {
        ret2[i] = pthread_create(&id2[i], NULL, consume, NULL);//Create a consumer thread
        if(ret2[i] != 0)
        {
            printf("consumer%d creat failed\n", i);
            exit(1);
        }
    }


    for(i = 0; i < N1; i++) {pthread_join(id1[i], NULL);}
    for(i = 0; i < N2; i++) {pthread_join(id2[i], NULL);}

    exit(0);


}

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=325652763&siteId=291194637