# ThreadPool.hpp ```cpp #pragma once #include #include #include #include #include #include #include #include #include class ThreadAdapter { protected: bool stop = false; std::shared_ptr m_t; /** * @brief 子线程阻塞锁 * */ std::mutex m_mutex; public: enum class ThreadStatus { WAIT, RUNNING, }; std::atomic status = ThreadStatus::WAIT; explicit ThreadAdapter(std::queue> &fq, std::condition_variable &cv, std::mutex &fqM) { m_t = std::make_shared([this, &q = fq, &c = cv, &m = fqM]() { while (!stop) { std::function fun; while (true) { { std::unique_lock l(m); if (!q.empty()) { fun = q.front(); q.pop(); break; } } std::unique_lock lock(m_mutex); c.wait(lock); } this->status = ThreadStatus::RUNNING; if (fun) { fun(); } this->status = ThreadStatus::WAIT; c.notify_all(); } }); } void Detach() { if (this->status != ThreadStatus::WAIT && m_t->joinable()) { this->stop = true; m_t->detach(); } } ~ThreadAdapter() { if (this->status != ThreadStatus::WAIT && m_t->joinable()) { this->stop = true; m_t->join(); } } }; class ThreadPool { protected: std::queue> fq; std::mutex fqM; std::condition_variable cv; std::vector> t; public: explicit ThreadPool(int s) { for (int i = 0; i < s; ++i) { auto p = std::make_shared(fq, cv, fqM); t.emplace_back(std::move(p)); } } /** * @brief 主要核心思想就这一个模板函数，用于将原始函数封装成 * * @tparam F 目标函数类型 * @tparam Args 函数的参数类型 * @param func 目标函数 * @param args 目标参数 * @return decltype(auto) 自动推导返回值 */ template decltype(auto) AddTask(F &&func, Args &&...args) { //推导最后的返回值类型 using res_type = typename std::result_of::type; //封装为packaged_task的智能指针类型 auto task = std::make_shared>( std::bind(std::forward(func), std::forward(args)...)); //获取用于异步获取返回值的future std::future res = task->get_future(); { { std::unique_lock lock(fqM); fq.emplace([task]() { (*task)(); }); } cv.notify_one(); } return res; } }; ``` ## test.cpp ```cpp #include #include #include "ThreadPool.hpp" #include using namespace std; int main() { ThreadPool threadPool(10); mutex m; std::atomic at = 0; for (int i = 0; i < 100000; ++i) { threadPool.AddTask([&a = at]() { for (int j = 0; j < 1000; ++j) { a++; } }); } for (int i = 0; i < 100; ++i) { cout << at << endl; std::this_thread::sleep_for(std::chrono::milliseconds (1)); } std::this_thread::sleep_for(std::chrono::minutes(100)); return 0; } ```

最多访问

最近更新

标签

最新评论