pthread是否支持任务队列

avatar
作者
筋斗云
阅读量:0

pthread(POSIX Threads)本身并不直接支持任务队列

以下是一个简单的C语言示例,展示了如何使用pthread和队列实现一个简单的线程池:

#include<stdio.h> #include <stdlib.h> #include <pthread.h> #include <stdatomic.h>  #define QUEUE_SIZE 10 #define NUM_THREADS 4  typedef struct Task {     void (*func)(void *);     void *arg; } Task;  typedef struct TaskQueue {     Task queue[QUEUE_SIZE];     atomic_int head;     atomic_int tail;     pthread_mutex_t lock;     pthread_cond_t cond; } TaskQueue;  typedef struct ThreadPool {     pthread_t threads[NUM_THREADS];     TaskQueue task_queue; } ThreadPool;  void task_queue_init(TaskQueue *task_queue) {     task_queue->head = ATOMIC_VAR_INIT(0);     task_queue->tail = ATOMIC_VAR_INIT(0);     pthread_mutex_init(&task_queue->lock, NULL);     pthread_cond_init(&task_queue->cond, NULL); }  void task_queue_push(TaskQueue *task_queue, Task task) {     pthread_mutex_lock(&task_queue->lock);     while ((task_queue->tail + 1) % QUEUE_SIZE == task_queue->head) {         pthread_cond_wait(&task_queue->cond, &task_queue->lock);     }     task_queue->queue[task_queue->tail] = task;     task_queue->tail = (task_queue->tail + 1) % QUEUE_SIZE;     pthread_cond_signal(&task_queue->cond);     pthread_mutex_unlock(&task_queue->lock); }  Task task_queue_pop(TaskQueue *task_queue) {     pthread_mutex_lock(&task_queue->lock);     while (task_queue->tail == task_queue->head) {         pthread_cond_wait(&task_queue->cond, &task_queue->lock);     }     Task task = task_queue->queue[task_queue->head];     task_queue->head = (task_queue->head + 1) % QUEUE_SIZE;     pthread_cond_signal(&task_queue->cond);     pthread_mutex_unlock(&task_queue->lock);     return task; }  void *thread_pool_worker(void *arg) {     ThreadPool *thread_pool = (ThreadPool *)arg;     while (1) {         Task task = task_queue_pop(&thread_pool->task_queue);         task.func(task.arg);     }     return NULL; }  void thread_pool_init(ThreadPool *thread_pool) {     task_queue_init(&thread_pool->task_queue);     for (int i = 0; i < NUM_THREADS; i++) {         pthread_create(&thread_pool->threads[i], NULL, thread_pool_worker, thread_pool);     } }  void thread_pool_add_task(ThreadPool *thread_pool, void (*func)(void *), void *arg) {     Task task = {.func = func, .arg = arg};     task_queue_push(&thread_pool->task_queue, task); }  int main() {     ThreadPool thread_pool;     thread_pool_init(&thread_pool);      // 添加任务到线程池     for (int i = 0; i < 20; i++) {         thread_pool_add_task(&thread_pool, my_function, (void *)(intptr_t)i);     }      // 等待所有任务完成     // ...      return 0; } 

这个示例中,我们创建了一个线程池,其中包含一个任务队列和四个工作线程。我们可以向线程池添加任务,这些任务将被工作线程从队列中取出并执行。注意,这个示例仅用于演示目的,实际应用中可能需要考虑更多细节,例如关闭线程池、处理错误等。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!