3

我正在尝试做一个简单的多线程消费者/生产者，其中多个读取器和写入器线程从文件读取到缓冲区，然后从缓冲区读取回文件。它应该是线程安全的。但是，它并没有像我预期的那样表现。它在中途停止，但每次都在不同的线路上？请帮助我理解我做错了什么？！？

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
//TODO Define global data structures to be used
#define BUF_SIZE 5
FILE *fr;
FILE *to;            /* declare the file pointer */

struct _data {
    pthread_mutex_t mutex;
    pthread_cond_t cond_read;
    pthread_cond_t cond_write;
    int condition;
    char buffer[BUF_SIZE];
    int datainbuffer;
}dc1 = {
    PTHREAD_MUTEX_INITIALIZER,PTHREAD_COND_INITIALIZER,PTHREAD_COND_INITIALIZER,0,{0},0
};


void *reader_thread(void *arg) {
    //TODO: Define set-up required
    struct _data *d = (struct _data *)arg;
    int killreaders = 0;
    while(1) {
        //TODO: Define data extraction (queue) and processing

        pthread_mutex_lock(&d->mutex);

        while (d->condition == 0 || d->datainbuffer<=0){
            pthread_cond_wait( &d->cond_read, &d->mutex );
            if(killreaders == 1){
                pthread_mutex_unlock(&d->mutex);
                pthread_cond_signal(&d->cond_read);
                pthread_cond_signal(&d->cond_write);
                return NULL;
            }

        }
        d->condition = 0;

        int i;
        char res;
        //if the buffer is not full, that means the end of file is reached and it time to kill the threads remaining.
        if(d->datainbuffer!=BUF_SIZE)
            killreaders = 1;

        for (i=0; i<(sizeof d->datainbuffer); i++) {
            res = d->buffer[i];
            printf("to file:%c",res);
            fputc(res, to);
        }
        d->datainbuffer = 0;


        pthread_mutex_unlock(&d->mutex);
        pthread_cond_signal( &d->cond_write );


    }

    return NULL;
}

void *writer_thread(void *arg) {
    //TODO: Define set-up required
    struct _data *d = (struct _data *)arg;
    char * pChar;
    int killwriters = 0;

    while(1){
        pthread_mutex_lock(&d->mutex);
        while( d->condition == 1 || d->datainbuffer>0){
            pthread_cond_wait( &d->cond_write, &d->mutex );
            if(killwriters==1){
                pthread_mutex_unlock(&d->mutex);
                pthread_cond_signal(&d->cond_write);
                pthread_cond_signal(&d->cond_read);
                return NULL;

            }
        }
        d->condition = 1;
        int i;
        char rc;
        for (i = 0; i < BUF_SIZE; i++){
            if((rc = getc(fr)) == EOF){
                killwriters = 1;
                pthread_mutex_unlock(&d->mutex);
                pthread_cond_signal(&d->cond_read);

                return NULL;
            }
            d->datainbuffer = i+1;
            d->buffer[i] = rc;
            printf("%c",rc);
        }

        int m = 0;

        pthread_mutex_unlock(&d->mutex);
        pthread_cond_signal(&d->cond_read);


    }


    return NULL;
}


#define M 10
#define N 20
int main(int argc, char **argv) {
    struct _data dc=dc1;

    fr = fopen ("from.txt", "rt");  /* open the file for reading */
    if (fr == NULL)
    {
        printf("Could not open file!");
        return 1;
    }
    to = fopen("to.txt", "wt");


    int i;
    pthread_t readers[N];
    pthread_t writers[M];


    for(i = 0; i < N; i++) { 
        pthread_create(&readers[i], NULL, reader_thread, (void*)&dc);
    }

    for(i = 0; i < M; i++) { 
        pthread_create(&writers[i], NULL, writer_thread, (void*)&dc);
    }
    fclose(fr);
    fclose(to);

    return 0;   
}

任何建议表示赞赏！

4

3 回答 3

3

您的线程正在读取和写入您在 main 中打开和关闭的文件。但是 main 在关闭这些文件之前并没有明确地等待线程完成。

于 2012-01-04T02:59:27.460 回答
3

除了 Scott Hunter 指出的问题之外，您的读者和作者在持有互斥锁的同时完成了所有“真正的工作”，从而打破了拥有多个线程的观点。

读者操作如下：

1) 获取互斥锁。
2) 阻塞条件变量，直到工作可用。
3）从队列中删除工作，可能是信号条件变量。
4) 释放互斥锁。
5) 处理工作。
6) 转到步骤 1。

编写器应按以下方式操作：

1）获取我们需要写的信息。
2) 获取互斥锁。
3）阻塞条件变量，直到队列中有空间。
4) 将信息放入队列，可能是信号条件变量。
5) 释放互斥锁。
6) 转到步骤 1。

注意两个线程都在不持有互斥锁的情况下做“真正的工作”？否则，如果一次只有一个线程可以工作，为什么还要有多个线程呢？

于 2012-01-04T03:38:42.467 回答
0

我不确定我的回答是否会对您有所帮助..但我会尽力为您提供一些参考代码。

我写了一个类似的程序（除了它不写入文件，而是在 中显示 queue-/produced-/consumed- 项目stdout）。它可以在这里找到 - https://github.com/sangeeths/pc。我已将命令行处理和队列逻辑分离到单独的文件中。

希望这可以帮助！

于 2012-01-04T05:13:32.373 回答