C++线程池的实现

最新推荐文章于 2025-02-03 20:06:36 发布

原创最新推荐文章于 2025-02-03 20:06:36 发布 · 463 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法 #开发语言

深度学习同时被 3 个专栏收录

31 篇文章

订阅专栏

c++

8 篇文章

订阅专栏

linux

8 篇文章

订阅专栏

文章展示了一个使用C++实现的线程池类ThreadPool，它创建指定数量的工作线程来执行任务。线程池接受任务并使用互斥锁和条件变量进行同步，确保在停止时正确处理未完成的任务。在主函数中，创建了一个线程池实例，并向其添加了多个SayHello任务。

#include <iostream>
#include<thread>
#include <queue>
#include <mutex>
#include <chrono>
#include <condition_variable>
#include <functional>
using namespace std;

#define NUM_THREADS 5

class ThreadPool {
public:
    explicit ThreadPool(size_t size):stop_(false) {
        // 创建线程池
        for(size_t i = 0; i < size; i++) {
            threads_.emplace_back([this] {
                while (true) {
                    unique_lock<mutex> lock(mutex_);
                    printf("wait");
                    condition_.wait(lock, [this] {return stop_ || !tasks_.empty();});
                    if(stop_ && tasks_.empty() ) {
                        return;
                    }
                    auto task = move(tasks_.front());
                    tasks_.pop();
                    lock.unlock();
                    task();
                }
            });
        }
    }
    template<class F>
    void addTask(F&& f) {
        printf("add");
        std::unique_lock<mutex> lock(mutex_);
        tasks_.emplace(forward<F>(f));
        condition_.notify_one();
    }
    ~ThreadPool() {
        printf("pool delete");
        std::unique_lock<std::mutex> lock(mutex_);
        stop_ = true;
        condition_.notify_all();
        for (auto& thread : threads_) {
            thread.join();
        }
    }
private:
    bool stop_;
    std::vector<std::thread> threads_;
    std::queue<std::function<void()>> tasks_;
    std::mutex mutex_;
    condition_variable condition_;
};
// 线程的运行函数
void say_hello(int num)
{
    printf("Hello Runoob！%d\n", num);
}
 

 
void SayHello(int num) {
    // std::cout << "Hello from thread " << num << std::endl;
    for(int i = 0; i < num; i ++) {
        cout << "循环" << num << endl;
    }
}
 
int main() {
    ThreadPool pool(16);
    for (int i = 0; i < 800; ++i) {
        pool.addTask([i] { SayHello(i); });
    }
    std::this_thread::sleep_for(std::chrono::seconds(5));
    return 0;
}