从生产者-消费者模型到线程池:用pthread手把手实现Linux C并发编程核心模式
在Linux系统编程领域,多线程开发一直是提升程序性能的关键手段。POSIX线程(pthread)作为Unix-like系统的标准线程API,为开发者提供了强大的工具集。本文将深入探讨两个经典并发模式:生产者-消费者模型和线程池的实现,通过完整代码示例展示如何用pthread构建稳健的并发程序。
1. 生产者-消费者模型实现
生产者-消费者问题是并发编程中的经典案例,它模拟了数据生产和消费的速度不匹配场景。我们使用pthread_mutex_t和pthread_cond_t来实现线程安全的缓冲区和线程间通信。
1.1 核心数据结构设计
首先定义共享缓冲区结构,这是生产者和消费者线程交互的媒介:
#define BUFFER_SIZE 10
typedef struct {
int buffer[BUFFER_SIZE];
int count;
int in;
int out;
pthread_mutex_t mutex;
pthread_cond_t not_empty;
pthread_cond_t not_full;
} shared_buffer_t;
这个结构包含:
buffer: 固定大小的循环缓冲区count: 当前缓冲区中的项目数in/out: 生产/消费位置的索引mutex: 保护共享状态的互斥锁not_empty/not_full: 用于线程同步的条件变量
1.2 初始化与销毁函数
缓冲区需要正确的初始化和清理:
void buffer_init(shared_buffer_t *b) {
b->count = 0;
b->in = 0;
b->out = 0;
pthread_mutex_init(&b->mutex, NULL);
pthread_cond_init(&b->not_empty, NULL);
pthread_cond_init(&b->not_full, NULL);
}
void buffer_destroy(shared_buffer_t *b) {
pthread_mutex_destroy(&b->mutex);
pthread_cond_destroy(&b->not_empty);
pthread_cond_destroy(&b->not_full);
}
1.3 生产者线程实现
生产者线程的核心逻辑是向缓冲区添加数据:
void* producer(void *arg) {
shared_buffer_t *b = (shared_buffer_t*)arg;
for (int i = 0; i < 20; i++) {
pthread_mutex_lock(&b->mutex);
// 等待缓冲区不满
while (b->count == BUFFER_SIZE) {
pthread_cond_wait(&b->not_full, &b->mutex);
}
// 生产数据
b->buffer[b->in] = i;
b->in = (b->in + 1) % BUFFER_SIZE;
b->count++;
printf("Produced: %d\n", i);
// 通知消费者
pthread_cond_signal(&b->not_empty);
pthread_mutex_unlock(&b->mutex);
usleep(rand() % 100000); // 模拟生产耗时
}
return NULL;
}
关键点说明:
- 使用
pthread_mutex_lock保护临界区 - 当缓冲区满时,通过
pthread_cond_wait等待not_full条件 - 生产数据后,用
pthread_cond_signal唤醒可能等待的消费者
1.4 消费者线程实现
消费者线程从缓冲区获取并处理数据:
void* consumer(void *arg) {
shared_buffer_t *b = (shared_buffer_t*)arg;
for (int i = 0; i < 20; i++) {
pthread_mutex_lock(&b->mutex);
// 等待缓冲区不空
while (b->count == 0) {
pthread_cond_wait(&b->not_empty, &b->mutex);
}
// 消费数据
int item = b->buffer[b->out];
b->out = (b->out + 1) % BUFFER_SIZE;
b->count--;
printf("Consumed: %d\n", item);
// 通知生产者
pthread_cond_signal(&b->

488

被折叠的 条评论
为什么被折叠?



