首页
/ BS::thread_pool 中硬件并发数获取的优化思考

BS::thread_pool 中硬件并发数获取的优化思考

2025-06-30 09:11:59作者:姚月梅Lane

在C++多线程编程中,线程池的大小设置是一个关键性能因素。BS::thread_pool作为一款优秀的C++线程池库,其默认使用std::thread::hardware_concurrency()来获取硬件支持的线程并发数。然而,在实际生产环境中,特别是在Linux系统上,这种方式存在一定的局限性。

问题背景

std::thread::hardware_concurrency()返回的是底层硬件支持的线程并发能力,而不是当前进程实际可用的核心数。在以下场景中会出现问题:

  1. 使用taskset命令限制进程运行在特定核心上时
  2. 在Slurm等批处理系统中运行作业时
  3. 通过cgroups限制CPU资源时

例如,在一台20核机器上运行taskset -c 1 myprogram,线程池会创建20个工作线程,但这些线程都被限制在单个核心上运行,导致性能下降。

技术分析

在Linux系统中,更准确的方法是使用sched_getaffinity()系统调用,它可以获取当前进程的CPU亲和性掩码。Folly库就采用了这种实现方式:

unsigned int hardware_concurrency() noexcept {
#if defined(__linux__) && !defined(__ANDROID__)
  cpu_set_t cpuset;
  if (!sched_getaffinity(0, sizeof(cpuset), &cpuset)) {
    auto count = CPU_COUNT(&cpuset);
    if (count != 0) {
      return count;
    }
  }
#endif
  return std::thread::hardware_concurrency();
}

BS::thread_pool的解决方案

BS::thread_pool的作者考虑到了跨平台兼容性问题,在v5.0.0版本中引入了原生扩展功能,提供了BS::get_os_process_affinity()函数。这个函数在Windows和Linux上都能返回当前进程的CPU亲和性信息。

用户可以通过以下方式获取实际可用的线程数:

#define BS_THREAD_POOL_NATIVE_EXTENSIONS
#include "BS_thread_pool.hpp"

int main()
{
    const std::optional<std::vector<bool>> affinity = BS::get_os_process_affinity();
    const std::size_t num_threads = affinity ? std::ranges::count(*affinity, true) : std::thread::hardware_concurrency();
    BS::thread_pool pool(num_threads);
    std::cout << "Number of available threads: " << pool.get_thread_count() << '\n';
}

未来发展方向

作者计划在v5.1.0版本中进一步优化这一功能:当启用原生扩展且创建线程池时未指定线程数时,自动使用进程亲和性信息来确定最佳线程数。这将在保持跨平台兼容性的同时,提供更智能的默认行为。

最佳实践建议

对于需要精确控制线程数的应用场景,建议:

  1. 启用BS::thread_pool的原生扩展功能
  2. 显式检查进程亲和性设置
  3. 根据实际可用核心数创建线程池
  4. 在容器化环境中特别注意CPU资源限制

这种设计既保持了库的跨平台特性,又为特定平台提供了优化空间,体现了良好的软件工程实践。

登录后查看全文

热门内容推荐

项目优选

收起
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
604
424
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
51
15
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
128
209
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
90
146
cherry-studiocherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
479
39
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
106
255
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
299
1.03 K
MateChatMateChat
前端智能化场景解决方案UI库,轻松构建你的AI应用,我们将持续完善更新,欢迎你的使用与建议。 官网地址:https://matechat.gitcode.com
693
92
markdown4cjmarkdown4cj
一个markdown解析和展示的库
Cangjie
33
4
JeecgBootJeecgBoot
🔥企业级低代码平台集成了AI应用平台,帮助企业快速实现低代码开发和构建AI应用!前后端分离架构 SpringBoot,SpringCloud、Mybatis,Ant Design4、 Vue3.0、TS+vite!强大的代码生成器让前后端代码一键生成,无需写任何代码! 引领AI低代码开发模式: AI生成->OnlineCoding-> 代码生成-> 手工MERGE,显著的提高效率,又不失灵活~
Java
96
17