linux学习：线程池

目录

原理

初始线程池

运行中的线程池

相关结构体

api

线程池初始化

投送任务

增加活跃线程

删除活跃线程

销毁线程池

例子

thread_pool.h

thread_pool.c

test.c  测试程序

原理

一个进程中的线程就好比是一家公司里的员工，员工的数目应该根据公司的业务多少来 定，太少了忙不过来，但是太多了也浪费资源，最理想的情况是：让进程有一些初始数目的 线程（所谓的线程池），当没有任务的时候这些线程自动进入睡眠，有了任务他们会立即执 行任务，不断循环。进程还应该可以根据自身任务的繁重与否来增删线程的数目，当所有的 任务都完成了之后，所有的线程还能妥当地收官走人，不带走一片云彩

初始线程池

  • 任务队列中刚开始没有任何任务，是一个具有头结点的空链队列
  • 使用互斥锁来保护这个队列
  • 使用条件变量来代表任务队列中的任务个数的变化——将来如果主线程往队列中投 放任务，那么可以通过条件变量来唤醒那些睡着了的线程
  • 通过一个公共开关——shutdown，来控制线程退出，进而销毁整个线程池

运行中的线程池

相关结构体

任务节点，包含需要执行的函数及其参数，通过链表连成一个任务队列

api

线程池初始化

投送任务

增加活跃线程

删除活跃线程

销毁线程池

例子

thread_pool.h

1 #ifndef _THREAD_POOL_H_
2 #define _THREAD_POOL_H_
3
4 #include <stdio.h>
5 #include <stdbool.h>
6 #include <unistd.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <strings.h>
10
11 #include <errno.h>
12 #include <pthread.h>
13
14 #define MAX_WAITING_TASKS 1000 //最大等待任务数
15 #define MAX_ACTIVE_THREADS 20  //最大活跃线程数的常量
16
17 struct task // 任务节点结构体
18 {
19     void *(*task)(void *arg);
20     void *arg;
21
22     struct task *next;
23 };
24
25 typedef struct thread_pool // 线程池结构体
26 {
27     pthread_mutex_t lock;
28     pthread_cond_t cond;
29     struct task *task_list;
30
31     pthread_t *tids;
32
33     unsigned waiting_tasks;
34     unsigned active_threads;
35
36     bool shutdown;
37 }thread_pool;
38
39 // 线程池初始化
40 
41 bool init_pool(thread_pool *pool, 
42         unsigned int threads_number);
43 // 投放任务
44 
45 bool add_task(thread_pool *pool, 
46         void *(*task)(void *arg), 
47             void *arg);
48 // 增加线程
49 
50 int add_thread(thread_pool *pool, 
51         unsigned int additional_threads_number);
52 // 删除线程
53 
54 int remove_thread(thread_pool *pool, 
55         unsigned int removing_threads_number);
56
57 bool destroy_pool(thread_pool *pool); // 销毁线程池
58 void *routine(void *arg); // 线程例程
59
60 #endif

thread_pool.c

1 #include "thread_pool.h" 
2 // 函数在被取消时调用，用于释放互斥锁
3 void handler(void *arg)
4 {
5     // 响应取消请求之后自动处理的例程：释放互斥锁
6     pthread_mutex_unlock((pthread_mutex_t *)arg);
7 }
8 // 函数是线程的执行例程，不断地从任务队列中取出任务并执行
9 void *routine(void *arg)
10 {
11     thread_pool *pool = (thread_pool *)arg;
12     struct task *p;
13
14     while(1)
15     {
16         // 访问任务队列前加锁，为防止取消后死锁，注册处理例程 handler
17         pthread_cleanup_push(handler, (void *)&pool->lock);
18         pthread_mutex_lock(&pool->lock);
19
20         // 若当前没有任务，且线程池未关闭，则进入条件变量等待队列睡眠
21         while(pool->waiting_tasks == 0 && !pool->shutdown)
22         {
23             pthread_cond_wait(&pool->cond, &pool->lock);
24         }
25
26         // 若当前没有任务，且线程池关闭标识为真，则立即释放互斥锁并退出
27         if(pool->waiting_tasks == 0 && pool->shutdown == true)
28         {
29             pthread_mutex_unlock(&pool->lock);
30             pthread_exit(NULL);
31         }
32
33         // 若当前有任务，则消费任务队列中的任务
34         p = pool->task_list->next;
35         pool->task_list->next = p->next;
36         pool->waiting_tasks--;
37
38         // 释放互斥锁，并弹栈 handler（但不执行他）
39         pthread_mutex_unlock(&pool->lock);
40         pthread_cleanup_pop(0);
41
42         // 执行任务，并且在此期间禁止响应取消请求
43         pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
44         (p->task)(p->arg);
45         pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
46
47         free(p);
48     }
49
50     pthread_exit(NULL);
51 }
52 // 函数用于初始化线程池，包括初始化互斥锁、条件变量，分配任务队列和线程数组，并创建指定数量的线程
53 bool init_pool(thread_pool *pool, unsigned int threads_number)
54 {
55     pthread_mutex_init(&pool->lock, NULL);
56     pthread_cond_init(&pool->cond, NULL);
57
58     pool->shutdown = false; // 关闭销毁线程池标识
59     pool->task_list = malloc(sizeof(struct task)); // 任务队列头结点
60     pool->tids = malloc(sizeof(pthread_t) * MAX_ACTIVE_THREADS);
61
62     if(pool->task_list == NULL || pool->tids == NULL)
63     {
64         perror("allocate memory error");
65         return false;
66     }
67
68     pool->task_list->next = NULL;
69
70     pool->waiting_tasks = 0;
71     pool->active_threads = threads_number;
72
73     int i;
74     for(i=0; i<pool->active_threads; i++) // 创建指定数目线程
75     {
76         if(pthread_create(&((pool->tids)[i]), NULL,
77             routine, (void *)pool) != 0)
78         {
79             perror("create threads error");
80             return false;
81         }
82     }
83
84     return true;
85 }
86 // 函数用于向线程池中添加任务，将任务添加到任务队列中，并唤醒一个等待中的线程来执行任务
87 bool add_task(thread_pool *pool, 88 void *(*task)(void *arg), void *arg)
89 {
90     struct task *new_task = malloc(sizeof(struct task)); // 新任务节点
91     if(new_task == NULL)
92     {
93         perror("allocate memory error");
94         return false;
95     }
96     new_task->task = task;
97     new_task->arg = arg;
98     new_task->next = NULL;
99
100     // 访问任务队列前获取互斥锁，此处无需注册取消处理例程
101     pthread_mutex_lock(&pool->lock);
102     if(pool->waiting_tasks >= MAX_WAITING_TASKS)
103     {
104         pthread_mutex_unlock(&pool->lock);
105
106         fprintf(stderr, "too many tasks.\n");
107         free(new_task);
108
109         return false;
110     }
111
112     struct task *tmp = pool->task_list;
113     while(tmp->next != NULL)
114         tmp = tmp->next;
115
116     tmp->next = new_task; // 添加新的任务节点
117     pool->waiting_tasks++;
118
119     // 释放互斥锁，并唤醒其中一个阻塞在条件变量上的线程
120     pthread_mutex_unlock(&pool->lock);
121     pthread_cond_signal(&pool->cond);
122
123     return true;
124 }
125 // 函数用于动态增加线程，根据传入的参数增加指定数量的线程
126 int add_thread(thread_pool *pool, unsigned additional_threads)
127 {
128     if(additional_threads == 0)
129         return 0;
130
131     unsigned total_threads =
132         pool->active_threads + additional_threads;
133
134     int i, actual_increment = 0;
135     for(i = pool->active_threads; // 循环地创建若干指定数目的线程
136         i < total_threads && i < MAX_ACTIVE_THREADS;
137             i++)
138     {
139         if(pthread_create(&((pool->tids)[i]), 140 NULL, routine, (void *)pool) != 0)
141         {
142             perror("add threads error");
143
144             if(actual_increment == 0)
145                 return -1;
146
147             break;
148         }
149         actual_increment++;
150     }
151
152     pool->active_threads += actual_increment;
153     return actual_increment;
154 }
155 // 函数用于动态减少线程，根据传入的参数取消掉指定数量的线程
156 int remove_thread(thread_pool *pool, unsigned int removing_threads)
157 {
158     if(removing_threads == 0)
159         return pool->active_threads;
160
161     int remain_threads = pool->active_threads - removing_threads;
162     remain_threads = remain_threads > 0 ? remain_threads : 1;
163
164     int i; // 循环地取消掉指定数目的线程
165     for(i=pool->active_threads-1; i>remain_threads-1; i--)
166     {
167         errno = pthread_cancel(pool->tids[i]);
168         if(errno != 0)
169             break;
170     }
171
172     if(i == pool->active_threads-1)
173         return -1;
174     else
175     {
176         pool->active_threads = i+1;
177         return i+1;
178     }
179 }
180 // 函数用于销毁线程池，将线程池的shutdown标志设为true，然后广播条件变量以唤醒所有等待中的线程，最后等待所有线程退出并释放资源
181 bool destroy_pool(thread_pool *pool)
182 {
183
184     pool->shutdown = true;
185     pthread_cond_broadcast(&pool->cond);
186
187     int i;
188     for(i=0; i<pool->active_threads; i++)
189     {
190         errno = pthread_join(pool->tids[i], NULL);
191         if(errno != 0)
192         {
193             printf("join tids[%d] error: %s\n", 194 i, strerror(errno));
195         }
196         else
197             printf("[%u] is joined\n", (unsigned)pool->tids[i]);
198
199     }
200
201     free(pool->task_list);
202     free(pool->tids);
203     free(pool);
204
205     return true;
206 }

test.c  测试程序

1 #include "thread_pool.h" 
2 // 模拟了一个需要执行一段时间的任务。任务接受一个参数n，表示任务执行的时间（秒）
3 void *mytask(void *arg)
4 {
5     int n = (int)arg;
6
7     printf("[%u][%s] ==> job will be done in %d sec...\n", 
8         (unsigned)pthread_self(), __FUNCTION__, n);
9
10     sleep(n);
11
12     printf("[%u][%s] ==> job done!\n", 13 (unsigned)pthread_self(), __FUNCTION__);
14
15     return NULL;
16 }
17 // 函数是一个辅助函数，每隔一秒打印一个计时器，用来帮助观察任务执行情况
18 void *count_time(void *arg)
19 {
20     int i = 0;
21     while(1)
22     {
23         sleep(1);
24         printf("sec: %d\n", ++i);
25     }
26 }
27 
28 int main(void)
29 {
30     pthread_t a;
31     pthread_create(&a, NULL, count_time, NULL);// 创建一个辅助线程a，用来计时
32
33     // 1, 初始化一个带有 2 条线程的线程池
34     thread_pool *pool = malloc(sizeof(thread_pool));
35     init_pool(pool, 2);
36
37     // 2, 投入 3 个任务 每个任务执行时间为0到9秒不等
38     printf("throwing 3 tasks...\n");
39     add_task(pool, mytask, (void *)(rand()%10));
40     add_task(pool, mytask, (void *)(rand()%10));
41     add_task(pool, mytask, (void *)(rand()%10));
42
43     // 3, 显示当前有多少条线程
44     printf("current thread number: %d\n", 45 remove_thread(pool, 0));
46     sleep(9);
47
48     // 4, 再投入 2 个任务
49     printf("throwing another 2 tasks...\n");
50     add_task(pool, mytask, (void *)(rand()%10));
51     add_task(pool, mytask, (void *)(rand()%10));
52
53     // 5, 增加 2 条线程
54     add_thread(pool, 2);
55
56     sleep(5);
57
58     // 6, 删除 3 条线程
59     printf("remove 3 threads from the pool, " 
60         "current thread number: %d\n", 61 remove_thread(pool, 3));
62
63     // 7, 销毁线程池
64     destroy_pool(pool);
65     return 0;
66 }

需将上述测试代码中的 mytask 函数改成你需要实现的功能函数即可

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.rhkb.cn/news/323014.html

如若内容造成侵权/违法违规/事实不符，请联系长河编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

【qt】容器的用法

【qt】容器的用法

容器目录 一.QVertor1.应用场景2.增加数据3.删除数据4.修改数据5.查询数据6.是否包含7.数据个数8.交换数据9.移动数据10.嵌套使用 二.QList1.应用场景2.QStringList 三.QLinkedList1.应用场景2.特殊点3.用迭代器来变量 四.QStack1.应用场景2.基本用法 五.QQueue1.应用场景2.基本…
阅读更多...
LeetCode例题讲解:快乐数

LeetCode例题讲解:快乐数

编写一个算法来判断一个数 n 是不是快乐数。 「快乐数」 定义为&#xff1a; 对于一个正整数&#xff0c;每一次将该数替换为它每个位置上的数字的平方和。然后重复这个过程直到这个数变为 1&#xff0c;也可能是 无限循环 但始终变不到 1。如果这个过程 结果为 1&#xff0c…
阅读更多...
[uniapp 地图组件] 小坑：translateMarker的回调函数，会调用2次

[uniapp 地图组件] 小坑：translateMarker的回调函数，会调用2次

大概率是因为旋转和移动是两个动画&#xff0c;动画结束后都会分别调用此函数 即使你配置了 【不旋转】它还是会调用两次&#xff0c; 所以此处应该是官方的bug
阅读更多...
JavaWeb之Servlet(上)

JavaWeb之Servlet(上)

前言 1. 什么是Servlet (1) Servlet介绍 (2) Servlet运行于支持Java的应用服务器中。 (3) Servlet工作模式&#xff1a; 2. Servlet API 3. 第一个Servlet (1) 创建一个类实现Servlet接口,重写方法。或继承HttpServlet亦可 (2) 在web.xml文档中配置映射关系 标签的执行…
阅读更多...
新能源汽车充电站智慧充电电能服务综合解决方案

新能源汽车充电站智慧充电电能服务综合解决方案

安科瑞薛瑶瑶18701709087/17343930412 ★解决方案 ✔目的地充电-EMS微电网平台 基于EMS解决方案从设备运维的角度解决本地充电的能量管理及运维问题&#xff0c;与充电管理平台打通数据&#xff0c;为企业微电网提供源、网、荷、储、充一体化解决方案。 ✔运营场站--电能服务…
阅读更多...
Burp插件之BurpSuitFake

Burp插件之BurpSuitFake

在最近的测试中发现一个问题 某些高校在在登陆教务平台登陆的时候没有设置验证码 而且重点是 他写的很清楚 密码是身份证后六位 以下是我的一些思路和过程中解决的办法 part1信息搜集 学生的学号是很有规律的 例如我的学号 就是入学年份专业代码班级学号 每个大学学号规则不…
阅读更多...
AlphaFold3: Google DeepMind的的新突破

AlphaFold3: Google DeepMind的的新突破

AlphaFold 3的论文今天在Nature期刊发表啦!这可是AI在生物领域最厉害的突破的最新版本。AlphaFold-3的新招就是用扩散模型去"画出"分子的结构。它一开始先从一团模模糊糊的原子云下手,然后慢慢透过去噪把分子变得越来越清楚。 Alphafold3 我们活在一个从Llama和Sora那…
阅读更多...
69、oak和华为atlas 200dk A2进行编解码测试

69、oak和华为atlas 200dk A2进行编解码测试

基本思想：将oak深度相机与atlas 200dk A2进行结合，测试其dvpp的编解码能力 cmakelist.txt cmake_minimum_required(VERSION 3.16) project(untitled10) set(CMAKE_CXX_FLAGS "-std=c++11") set(CMAKE_CXX_STANDARD 11) add_definitions(-DENABLE_DVPP_INTERFACE)i…
阅读更多...
小红书释放被封手机号 无限注册

小红书释放被封手机号 无限注册

前几年抖音也可以释放被封手机号 那时候都不重视 导致现在被封手机号想释放 基本不可能的 或者就是最少几百块 有专业的人帮你通过某些信息差释放 本教程是拆解 小红书被封手机号怎么释放&#xff0c;从今年开始&#xff0c;被封的手机号无法注销了 所以很困扰 那么本教程来…
阅读更多...
【ytb数据采集器】按关键词批量爬取视频数据，界面软件更适合文科生！

【ytb数据采集器】按关键词批量爬取视频数据，界面软件更适合文科生！

一、背景介绍 1.1 爬取目标 用Python独立开发的爬虫工具&#xff0c;作用是&#xff1a;通过搜索关键词采集油管的搜索结果&#xff0c;包含14个关键字段&#xff1a;关键词,页码,视频标题,视频id,视频链接,发布时间,视频时长,频道名称,频道id,频道链接,播放数,点赞数,评论数…
阅读更多...
nacos下载安装和nacos启动报错

nacos下载安装和nacos启动报错

nacos简介: Nacos /nɑ:kəʊs/ 是 Dynamic Naming and Configuration Service的首字母简称&#xff0c;一个更易于构建云原生应用的动态服务发现、配置管理和服务管理平台。 Nacos 致力于帮助您发现、配置和管理微服务。Nacos 提供了一组简单易用的特性集&#xff0c;帮助您…
阅读更多...
关于线程池，它的扩展问题你知道吗？（自己总结）

关于线程池，它的扩展问题你知道吗？（自己总结）

专门想一下为什么线程池不用Excutors&#xff0c;之前的印象是错的&#xff0c;居然还拿来面试里讲&#xff0c;惭愧&#xff0c;这里暂时整理俩小问题&#xff0c;其他的后续可能会更新。。 线程池是创建的越大越好嘛 #线程池创建的越大越好吗 Tip&#xff1a;2024-04-10 更…
阅读更多...
阵痛中的乳业产业，何时才能成为下一个啤酒产业？

阵痛中的乳业产业，何时才能成为下一个啤酒产业？

说起饮品&#xff0c;近年来中国啤酒业中各大品牌齐齐聚焦高端化的趋势绝对值得一提。然而&#xff0c;与之相反&#xff0c;国内乳业却是仍未进入高端化阶段&#xff0c;甚至陷入了周期底部中。 图源&#xff1a;中国圣牧财报 增收降利 牧企承受巨大的供需缺口压力 从产业链…
阅读更多...
字节跳动（社招）四面算法原题

字节跳动（社招）四面算法原题

TikTok 进展 又是一期定时汇报 TikTok 进展的推文。 上周&#xff0c;美国总统拜登签署了价值 950 亿美元的一揽子对外援助法案。 该法案涉及强制字节跳动剥离旗下应用 TikTok 美国业务&#xff0c;即 针对 TikTok 非卖即禁的"强抢行为"开始进入九个月&#xff08;27…
阅读更多...
五大自动化测试的 Python 框架

五大自动化测试的 Python 框架

&#x1f525; 交流讨论&#xff1a;欢迎加入我们一起学习&#xff01; &#x1f525; 资源分享&#xff1a;耗时200小时精选的「软件测试」资料包 &#x1f525; 教程推荐&#xff1a;火遍全网的《软件测试》教程 &#x1f4e2;欢迎点赞 &#x1f44d; 收藏 ⭐留言 &#x1…
阅读更多...
学习经验分享【36】论文投稿写作（非理工科文章）

学习经验分享【36】论文投稿写作（非理工科文章）

业务进一步扩展&#xff0c;可辅导非理工科偏文科性质的论文辅导&#xff0c;有需要评职称但没有时间精力研究的或者其他相关需求的朋友可咨询了解。 人工智能技术在各领域的发展和思考&#xff0c;类似这种主题的文章。
阅读更多...
Adobe系列软件安装

Adobe系列软件安装

双击解压 先运行Creative_Cloud_Set_Up.exe。 完毕后&#xff0c;运行AdobeGenP.exe 先Path&#xff0c;选路径&#xff0c;如 C:\Program Files\Adobe 后Search 最后Patch。 关闭软件&#xff0c;修图&#xff01;
阅读更多...
17 内核开发-内核内部内联汇编学习

17 内核开发-内核内部内联汇编学习

​ 17 内核开发-内核内部内联汇编学习 课程简介&#xff1a; Linux内核开发入门是一门旨在帮助学习者从最基本的知识开始学习Linux内核开发的入门课程。该课程旨在为对Linux内核开发感兴趣的初学者提供一个扎实的基础&#xff0c;让他们能够理解和参与到Linux内核的开发过程中…
阅读更多...
Jenkins--自动化构建和部署SpringBoot项目

Jenkins--自动化构建和部署SpringBoot项目

一、实现目标 通过在Jenkins中创建流水线任务&#xff0c;编写流水线脚本以实现自动化构建和部署SpringBoot项目。流水线脚本主要实现以下几个步骤&#xff1a; Preparation&#xff1a;从gitee上拉取远程仓库的SpringBoot项目代码。Build&#xff1a;使用Maven对拉取的代码进…
阅读更多...
浅析扩散模型与图像生成【应用篇】(二十二)——DreamBooth

浅析扩散模型与图像生成【应用篇】(二十二)——DreamBooth

21. DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generation 本文提出一种根据少量样例图片来对文生图模型进行微调的方法&#xff0c;从而可以生成包含样例物体&#xff0c;但风格、姿态、背景都可以任意修改的图片。现有的文生图模型都是需要…
阅读更多...
最新文章
推荐文章

Copyright @ 2022~2023