c中的一个消费者多个生产者在完全缓冲后恢复时阻止竞争

时间:2012-05-28 11:17:29

标签: c mutex race-condition producer-consumer circular-buffer

我创建了一个循环缓冲区,其中多个客户端将不同长度的消息写入缓冲区。服务器将它们读出来。它基于代码和消费者/生产者问题。 问题是当缓冲区已满并且服务器从缓冲区中删除所有数据时,客户端会发出信号以恢复其写入操作,而是另一个客户端(在另一个线程中)开始在缓冲区中写入消息。我希望在缓冲区已满之前已经写入的客户端恢复它的操作,以便消息不会无序到达。

这是我的代码(我删除了很多测试代码)

#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include <pthread.h>
#include <unistd.h>

#define BUFFER_SIZE 8
#define NUM_THREADS 4

struct cBuf{
    char    *buf; 
    int     size;   
    int     start;
    int     end;   
    pthread_mutex_t mutex;
    pthread_cond_t  buffer_full;
    pthread_cond_t  buffer_empty;
};

struct cBuf cb;


void buf_Init(struct cBuf *cb, int size) {
    int i;
    cb->size  = size + 1; 
    cb->start = 0;
    cb->end   = 0; 
    cb->buf = (char *)calloc(cb->size, sizeof(char)); 
    for (i=0;i<size;i++) cb->buf[i]='_';

}

void buf_Free(struct cBuf *cb) {
    free(cb->buf);
}

int buf_IsFull(struct cBuf *cb) {
    return (cb->end + 1) % cb->size == cb->start; 
}

int buf_IsEmpty(struct cBuf *cb) {
    return cb->end == cb->start; 
}

int buf_Insert(struct cBuf *cb, char *elem) {

    int i,j;

    pthread_mutex_lock(&(cb->mutex));
    for (i=0; i < strlen(elem); ++ i){
        if (buf_IsFull(cb)==1) printf("\nProducer (buf_Insert) is waiting because of full buffer");
        while(buf_IsFull(cb)){  
            pthread_cond_signal(&(cb->buffer_full));            
            pthread_cond_wait(&(cb->buffer_empty),&(cb->mutex));
        } 
        cb->buf[cb->end] = elem[i]; 
        cb->end = (cb->end + 1) % cb->size;     
        printf("%c [INPUT]",elem[i]);
    }

    pthread_cond_signal(&(cb->buffer_full));
    pthread_mutex_unlock(&(cb->mutex));     
    return 0;       
}

int buf_Read(struct cBuf *cb, char *out) {
    int i,j;

    pthread_mutex_lock(&(cb->mutex));
    if (buf_IsEmpty(cb))printf("\nConsumer (buf_Read) is waiting because of empty buffer\n");
    while(buf_IsEmpty(cb)){
        pthread_cond_wait(&(cb->buffer_full),&(cb->mutex));
    }
    for (i=0;i<BUFFER_SIZE-1;i++){
    printf("\n");
        if (cb->start == cb->end) break;        
        out[i] = cb->buf[cb->start];
        cb->buf[cb->start] = '_';
        cb->start = (cb->start + 1) % cb->size; 
        printf("%c [OUTPUT]",out[i]);
    }
    pthread_cond_signal(&(cb->buffer_empty));
    pthread_mutex_unlock(&(cb->mutex)); 
    return 0;
}

void * client(void *cb){

    pthread_detach(pthread_self());

    struct cBuf *myData;
    myData = (struct cBuf*) cb;
    char input[]="Hello World!";
    if (buf_Insert(myData, input)){
        //succes on return 0
        printf("\n");
    } 

    return 0;
}

int main(void) {
    char out[60];
    pthread_t thread;
    int i;
    /* Initialise conditioners*/
    pthread_cond_init(&(cb.buffer_full),NULL);
    pthread_cond_init(&(cb.buffer_empty),NULL);

    buf_Init(&cb, BUFFER_SIZE);

    for (i = 0; i<NUM_THREADS; i++){
            if(pthread_create (&thread,NULL, client, (void *) &cb) !=0){
        } else {

        }
    }

    while (1){
        if (buf_Read(&cb,out)){
        } 
    }

    //empty the buffer; free the allocated memory
    buf_Free(&cb);
    return 0;
}

1 个答案:

答案 0 :(得分:3)

我已在Producer/consumer seems to be in deadlock when buffer is smaller than input from producer的评论中解释过,但这些是评论,所以这里的答案如下:

你永远不应该在队列中有部分消息。确保你永远不会写一个。

您可以在开始编写消息之前检查是否有足够的空间,如果没有,则可以立即等待buffer_empty,或者您可以更改队列以向分配的数据发送共享指针(将所有权传递给使用者或引用计数) )或其他东西,所以每条消息只占用队列中的一个槽并为其余的分配内存。什么是最好的将取决于您的消息的确切性质。只要没有部分消息,任何事情都会发生。

虽然可以记录哪个特定的作者需要完成一条消息并且只是唤醒它,但它会非常复杂。同步很难实现,不要通过在其上添加额外的要求而使其变得更难。

事实上,除非这是一个家庭作业(从某种意义上说,你要学习同步是如何工作的),只需寻找现成的消息队列。数据报模式下的SysV-IPC或者unix-domain套接字是我想到的两个选项,或者寻找一些可以实现的库。