C设计实现一个线程池(thread pool)

本文介绍了如何在Ubuntu Linux环境下,使用C语言和pthread库实现一个线程池。线程池遵循先进先出原则,通过链表管理任务,不涉及线程同步。文中详细阐述了线程池的结构、任务结构、线程创建与销毁、任务添加和线程运行函数。实验证明,线程池能实现任务的并行执行,提高了执行效率。当不使用循环执行任务时,可能会导致任务执行不完全。
摘要由CSDN通过智能技术生成

C设计实现一个线程池

环境

在ubuntu的linux操作系统下,使用C语言,使用pthread创建和管理线程。任务满足先进先出的原则,因此使用链表。
由于课程暂未学习到线程之间的互斥和锁的问题,所以不考虑线程的问题。
以链表的形式创建任务,导入线程池然后执行,再销毁线程池,即为实验的流程。

思路

下面开始讲解我关于线程池的构造。因为还没有学习到互斥锁的知识,所以暂时不考虑调用线程切换的时候可能出现的互斥问题。而是着重关心线程池中任务以及线程的构造。

对于任务,我在结构里定义了三个变量:任务函数(其实是输出一行字符串的函数),传入任务的参数,以及指向下一个任务节点的指针。我用链表构造了一个任务串,这样可以按照需求实现FIFO。

而对于线程池的结构,我也定义了三个变量:int类型的最大线程数,用指针表示的线程ID数组,以及任务链表的链表头部。

然后线程池的函数主要有四个:线程池的创建,线程池的销毁,往任务链表里添加任务(添加到链表尾部),以及线程运行函数routine。

对于线程池的创建,先是对线程池的结构以及线程ID数组进行开辟空间,然后初始化线程池的变量:最大线程数置0,链表头部置零。然后创建若干个线程,每个线程对应着独一无二的TID,以及联系着线程运行函数的地址。

线程池的销毁跟它的创建对应。我们需要释放三个空间。在等待子线程执行完routine函数之后,释放线程ID数组的空间,一个一个的释放链表节点的空间,最后是释放整个线程池的空间。

而添加任务的函数,思路来源于往链表的结尾添加节点——可以保证访问时可以从头访问到尾部,保证了FIFO的原则。先创建一个任务结构work,开辟空间后,填充参数,然后往链表里添置节点。如果链表为空,则work为链表头节点。

而线程运行函数则是使用了while循环,因为如果不使用for循环,每一个线程只能执行一个任务就结束了,从而完成不了所有的任务(假如任务数量超过线程数量的话)。

调试的时候发现线程池的并行运行可能跟openmp有相似之处——每个线程不会执行重复的任务,任务会对所有线程进行均匀的分配,使得执行速度为最快。

主函数文件先是设置了一个函数——关于任务的执行函数(输出一行字符串)。然后主函数分别编写线程池的创建,往线程池的链表扔任务,销毁线程池。

部分执行结果如下。

在这里插入图片描述

在这里插入图片描述
可以看到,线程的执行是并行的。即for循环里的pthread_join()让父线程等待所有子线程的执行,而在执行for循环的第一次迭代时,输出“wait for thread 0 exit ”后,即出现五个”finished”,再出现其余四个本应该跟在”wait for thread 0 exit”的输出,说明了,pthread_join()等待子线程0执行时,所有线程都已经在并行执行了。

如果在线程的执行任务routine函数里不采用while,只采用if的话,则会出现任务执行不完的情况。我设置了一个标记,即在销毁线程池函数中,如果销毁链表节点的时候发现有节点存在,说明任务还没执行(执行了的任务都当场被free掉了),就会输出一行”not destroyed!”字符串。如下所示:

在这里插入图片描述

代码

main.c

 
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include "pool.c"
#include "pool.h"
#include <pthread.h>

void *do_task(void *arg)
{
   
    printf("thread %d\n", (int)arg);
    printf("begin to sleep for 1 sec\n");
    sleep(1);
    return NULL;
}
 
int main(int arg, char **argv)
{
   
    if (tpool_create(5) != 0)//创建成功之后返回的值是0
    {
   
        printf("tpool_create failed\n");
        exit(
下面是一个用 C 语言实现的简单动态线程池的示例: ```c #include <stdio.h> #include <stdlib.h> #include <pthread.h> typedef struct { void (*task_func)(void*); // 任务函数指针 void* task_arg; // 任务参数 } Task; typedef struct { Task* tasks; // 任务数组 int capacity; // 任务数组容量 int size; // 当前任务数量 int front; // 队列头索引 int rear; // 队列尾索引 pthread_mutex_t lock; // 互斥锁 pthread_cond_t not_empty; // 非空条件变量 pthread_cond_t not_full; // 非满条件变量 } ThreadPool; void task_queue_init(ThreadPool* pool, int capacity) { pool->tasks = (Task*)malloc(sizeof(Task) * capacity); pool->capacity = capacity; pool->size = 0; pool->front = 0; pool->rear = 0; pthread_mutex_init(&pool->lock, NULL); pthread_cond_init(&pool->not_empty, NULL); pthread_cond_init(&pool->not_full, NULL); } void task_queue_destroy(ThreadPool* pool) { free(pool->tasks); pthread_mutex_destroy(&pool->lock); pthread_cond_destroy(&pool->not_empty); pthread_cond_destroy(&pool->not_full); } void task_queue_push(ThreadPool* pool, Task task) { pthread_mutex_lock(&pool->lock); while (pool->size == pool->capacity) { pthread_cond_wait(&pool->not_full, &pool->lock); } pool->tasks[pool->rear] = task; pool->rear = (pool->rear + 1) % pool->capacity; pool->size++; pthread_cond_signal(&pool->not_empty); pthread_mutex_unlock(&pool->lock); } Task task_queue_pop(ThreadPool* pool) { pthread_mutex_lock(&pool->lock); while (pool->size == 0) { pthread_cond_wait(&pool->not_empty, &pool->lock); } Task task = pool->tasks[pool->front]; pool->front = (pool->front + 1) % pool->capacity; pool->size--; pthread_cond_signal(&pool->not_full); pthread_mutex_unlock(&pool->lock); return task; } void* worker_thread(void* arg) { ThreadPool* pool = (ThreadPool*)arg; while (1) { Task task = task_queue_pop(pool); task.task_func(task.task_arg); } return NULL; } void thread_pool_init(ThreadPool* pool, int capacity, int num_workers) { task_queue_init(pool, capacity); pthread_t* threads = (pthread_t*)malloc(sizeof(pthread_t) * num_workers); for (int i = 0; i < num_workers; i++) { pthread_create(&threads[i], NULL, worker_thread, pool); } } void thread_pool_destroy(ThreadPool* pool, int num_workers) { for (int i = 0; i < num_workers; i++) { task_queue_push(pool, (Task){NULL, NULL}); // 发送结束任务给所有工作线程 } pthread_t* threads = (pthread_t*)malloc(sizeof(pthread_t) * num_workers); for (int i = 0; i < num_workers; i++) { pthread_join(threads[i], NULL); } task_queue_destroy(pool); } // 示例任务函数 void my_task(void* arg) { printf("Hello from task\n"); } int main() { ThreadPool pool; thread_pool_init(&pool, 10, 5); for (int i = 0; i < 10; i++) { task_queue_push(&pool, (Task){my_task, NULL}); } thread_pool_destroy(&pool, 5); return 0; } ``` 上述示例使用了一个循环队列来实现任务队列,通过互斥锁和条件变量来保证线程安全。`ThreadPool` 结构体表示线程池,包含了任务队列和相关的属性。`task_queue_push` 和 `task_queue_pop` 分别用于将任务放入队列和从队列中取出任务。 `worker_thread` 函数是工作线程的入口函数,它会不断从任务队列中取出任务并执行。在 `thread_pool_init` 函数中创建了指定数量的工作线程,并开始执行任务。 注意,这只是一个简单的示例实现,实际使用时可能需要更多的错误处理和线程安全的考虑。另外,该示例没有实现线程池的动态调整功能,可以根据实际需求进行扩展。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值