Linux应用开发（4）：POSIX条件变量: pthread_cond_t

1. 简述

        条件变量是多线程编程中用于线程同步的一种机制，它允许线程在某个条件不满足时挂起，直到该条件变为真。在这种情况下，条件变量起着线程间同步的作用。

        在Linux系统中，默认提供了POSIX线程库pthread。 在POSIX线程（pthread）库中，条件变量通过pthread_cond_t类型表示。

2. 工作原理

        条件变量的工作机制依赖于两个关键的概念：互斥锁（mutex）和条件等待/通知。互斥锁用于保护共享资源，防止多个线程同时访问造成数据不一致。条件变量则用于在特定条件下挂起线程，当条件满足时再唤醒线程。

        线程在使用条件变量挂起之前，必须先获取与之关联的互斥锁。这样做是为了防止在线程挂起和唤醒过程中条件状态发生改变，从而造成竞态条件。在条件变量控制进入等待（休眠）之前，会释放互斥锁，这样其他线程才有可能获得该锁并修改状态。当条件变量被信号唤醒时，线程会自动重新获取互斥锁，然后再次检查条件是否满足。如果不满足，线程可以再次挂起等待。

3. pthread_cond_t关联的API

（1）初始化条件变量

int pthread_cond_init(pthread_cond_t *cond, const pthread_condattr_t *attr);

（2）销毁条件变量

int pthread_cond_destroy(pthread_cond_t *cond);

（3）等待条件变量

int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex);

（4）唤醒等待条件变量的一个线程

int pthread_cond_signal(pthread_cond_t *cond);

（5）唤醒所有等待条件变量的线程

int pthread_cond_broadcast(pthread_cond_t *cond);

4. 例程

        在这个历程中，我们创建了一种生产者和消费者的供需关系，当生产者生产出来了数据，通知消费者使用数据。当没有数据时，消费者进入等待。

#include <stdio.h>

#include <stdlib.h>

#include <pthread.h>


#define BUFFER_SIZE 10

pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; ///< 初始化互斥锁

pthread_cond_t cond = PTHREAD_COND_INITIALIZER; ///< 初始化条件变量

int buffer[BUFFER_SIZE];

int count = 0; ///< 记录缓冲区中的元素数量

/** 生产者线程. */

void* producer(void* arg) {

    int id = *(int*)arg;

    for (int i = 0; i < 20; ++i) {

        pthread_mutex_lock(&mutex);

        while (count == BUFFER_SIZE) {

            pthread_cond_wait(&cond, &mutex);

        }

        buffer[id % BUFFER_SIZE] = i;

        count++;

        printf("Producer %d: produced %d\n", id, i);

        pthread_cond_signal(&cond); ///< 通知消费者

        pthread_mutex_unlock(&mutex);

        sleep(1); ///< 模拟生产时间

    }

    return NULL;

}

/** 消费者线程. */

void* consumer(void* arg) {

    int id = *(int*)arg;

    for (int i = 0; i < 20; ++i) {

        pthread_mutex_lock(&mutex);

        while (count == 0) {

            pthread_cond_wait(&cond, &mutex);

        }

        int data = buffer[id % BUFFER_SIZE];

        count--;

        printf("Consumer %d: consumed %d\n", id, data);

        pthread_cond_signal(&cond); // 通知生产者

        pthread_mutex_unlock(&mutex);

        sleep(1); // 模拟消费时间

    }

    return NULL;

}

int main(int argc, char* argv[])

{

    pthread_t p1, p2, c1, c2;

    int ids[] = {0, 1};

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

    pthread_create(&p1, NULL, producer, &ids[0]);

    pthread_create(&p2, NULL, producer, &ids[1]);

    pthread_create(&c1, NULL, consumer, &ids[0]);

    pthread_create(&c2, NULL, consumer, &ids[1]);

    /** 等待线程结束,回收资源. */

    pthread_join(p1, NULL);

    pthread_join(p2, NULL);

    pthread_join(c1, NULL);

    pthread_join(c2, NULL);

    return 0;

}