线程池通过复用线程减少开销，核心组件包括工作线程、任务队列、同步机制和控制状态；使用线程安全队列管理任务，配合互斥锁与条件变量实现等待唤醒；每个线程循环从队列取任务执行，支持优雅关闭；通过enqueue提交任务，可结合future获取结果，优化方向包括无锁队列、优先级调度和动态调整线程数。
在C++高性能并发编程中，线程池的核心目标是复用线程、减少频繁创建销毁的开销，并有效管理任务调度。实现一个高效的线程池需要解决任务队列、线程同步、负载均衡和资源释放等问题。下面是一个简洁、实用且线程安全的线程池设计与实现方案。
线程池的基本结构
一个典型的线程池包含以下几个核心组件：
- 工作线程集合：固定数量的后台线程，持续从任务队列中取任务执行。
- 任务队列：存放待执行的任务（通常为函数对象），使用线程安全的队列。
- 同步机制：使用互斥锁（std::mutex）和条件变量（std::condition_variable）协调线程间的等待与唤醒。
- 控制状态：标记线程池是否正在运行，用于优雅关闭。
任务队列与线程安全封装
任务队列需要支持多线程下的入队和出队操作。可以封装一个线程安全的队列：
template<typename T>
class ThreadSafeQueue {
private:
std::queue<T> queue_;
mutable std::mutex mtx_;
std::condition_variable cv_;
public:
void push(T task) {
std::lock_guard<std::mutex> lock(mtx_);
queue_.push(std::move(task));
cv_.notify_one();
}
bool try_pop(T& task) {
std::lock_guard<std::mutex> lock(mtx_);
if (queue_.empty()) return false;
task = std::move(queue_.front());
queue_.pop();
return true;
}
void wait_and_pop(T& task) {
std::unique_lock<std::mutex> lock(mtx_);
cv_.wait(lock, [this] { return !queue_.empty(); });
task = std::move(queue_.front());
queue_.pop();
}
};
线程池的实现
基于上述队列，构建线程池类：
立即学习“C++免费学习笔记（深入）”；
class ThreadPool {
private:
std::vector<std::thread> workers_;
ThreadSafeQueue<std::function<void()>> task_queue_;
std::atomic<bool> stop_{false};
void worker_thread() {
while (!stop_) {
std::function<void()> task;
task_queue_.wait_and_pop(task);
if (task) task();
}
}
public:
explicit ThreadPool(size_t num_threads) {
for (size_t i = 0; i < num_threads; ++i) {
workers_.emplace_back(&ThreadPool::worker_thread, this);
}
}
~ThreadPool() {
stop_ = true;
// 唤醒所有等待线程
// 可以通过向队列中放入空任务来触发退出
for (auto& t : workers_) {
if (t.joinable()) t.join();
}
}
template<typename F>
void enqueue(F&& f) {
task_queue_.push(std::forward<F>(f));
}
};
这个实现中，每个工作线程在worker_thread中循环等待任务。通过wait_and_pop阻塞等待新任务，一旦收到任务立即执行。
使用示例与性能建议
使用线程池提交任务非常简单：
int main() {
ThreadPool pool(4); // 创建4个线程
for (int i = 0; i < 8; ++i) {
pool.enqueue([i]() {
std::cout << "Task " << i << " running on thread "
<< std::this_thread::get_id() << '\n';
std::this_thread::sleep_for(std::chrono::milliseconds(100));
});
}
// 主线程等待任务完成（实际项目中可加屏障或future）
std::this_thread::sleep_for(std::chrono::seconds(2));
return 0;
}
为了进一步提升性能，可以考虑以下优化：
- 使用无锁队列（如基于CAS的SPSC/MPSC队列）减少锁竞争。
- 为任务添加优先级支持，使用std::priority_queue配合自定义比较器。
- 结合std::packaged_task返回std::future，方便获取执行结果。
- 动态调整线程数（根据负载），但需权衡复杂性与收益。
