qt线程同步之条件等待

本文章使用QWaitCondition实现线程间的同步

#include <QApplication>
#include <QMutex>
#include <QThread>
#include <QSemaphore>
#include <QWaitCondition>
#include <iostream>
using namespace std;

// 需要生产的数据量
const int DATA_SIZE = 100;

// 指定缓冲区的大小
const int BUF_SIZE = 10;

// 存/取数据的缓冲区
int buf[BUF_SIZE] = {0};

QMutex mutex;// 实现线程间的互斥

// 当缓冲区没有填充满时，该条件被触发（反之，当缓冲区被填充满了，该条件会阻塞线程）
QWaitCondition bufIsNotFull;

// 当缓冲区不为空时，该条件被触发（反之，当缓冲区为空时，该条件会阻塞线程）
QWaitCondition bufIsNotEmpty;

// 统计生产的且还没有被消费的数据量
int nUsedSapce = 0;


class CProducer : public QThread
{
protected:
    virtual void run()
    {
        for (int i = 0 ; i < DATA_SIZE ; ++i)
        {
            mutex.lock();
            if (nUsedSapce == BUF_SIZE)// 如果缓冲区已填满，则等待缓冲区中的数据被取走
                bufIsNotFull.wait(&mutex);// 在阻塞线程前，bufIsNotFull会解锁互斥量，然后阻塞线程，等待bufIsNotFull条件对象被触发，在返回前，会重新锁定互斥量，然后返回

            buf[i%BUF_SIZE] = qrand() % 50 + 1;// 保存生产的数据
            cout << "produce data is :" << buf[i%BUF_SIZE] << endl;

            ++nUsedSapce;// 生产了一个数据

            bufIsNotEmpty.wakeAll();// 唤醒所有等待bufIsNotEmpty条件的线程

            mutex.unlock();
        }
    }
};

class CConsumer : public QThread
{
protected:
    virtual void run()
    {
        for (int i = 0 ; i < DATA_SIZE ; ++i)
        {
            mutex.lock();
            if (nUsedSapce == 0)// 如果缓冲区为空，那么等待缓冲区被填充数据
                bufIsNotEmpty.wait(&mutex);// 在阻塞线程前，bufIsNotEmpty会解锁互斥量，然后阻塞线程，等待bufIsNotEmpty条件对象被触发，在返回前，bufIsNotEmpty会重新锁定互斥量，然后返回

            int nData = buf[i%BUF_SIZE];// 取出要消费的数据
            cout << "consume data is: " << nData << endl;

            --nUsedSapce;// 消费了一个数据

            bufIsNotFull.wakeAll();// 唤醒所有等待bufIsNotFull条件的线程

            mutex.unlock();
        }

    }
};


int main(int argc, char *argv[])
{
    QApplication app(argc, argv);

    CProducer producer;
    CConsumer consumer;

    producer.start();
    consumer.start();

    producer.wait();
    consumer.wait();

    return app.exec();
}