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;
}