C++11 简单实现线程池的方法

#pragma once
#include <thread>
#include <vector>
#include <queue>
#include <condition_variable>
#include <mutex>
/*
* 抽象一个任务 根据自己的需求扩展
*/
class AbsTask 
{
public:
 AbsTask() = default;
 virtual ~AbsTask() = default;
public:
 virtual void run() = 0;
};
template<class _Ty>
class ThreadPoolManage
{
public:
 ThreadPoolManage(unsigned int nMaxThread)
 :mMaxThreadNum(nMaxThread)
 , mThreadStatus(false)
 {
 //启动的时候就要创建线程
 auto maxNum = std::thread::hardware_concurrency();
 //获取当前操作系统中CPU的核心数量 根据核心数量来设置 最大工作线程的并发数量
 mMaxThreadNum = mMaxThreadNum > maxNum ? maxNum : mMaxThreadNum;
 //创建工作线程池
 for (auto i = 0; i < mMaxThreadNum; i++)
 {
  mWorkers.emplace_back([this] {
  while (true)
  {
   std::unique_lock<std::mutex> lock(this->mQueue_mutex);
   this->mCondition.wait(lock, [this]() {return this->mThreadStatus || !this->mTasks.empty(); });
   if (this->mThreadStatus && this->mTasks.empty()) 
   { 
   return; 
   }
   //获取队列头部的任务
   auto task = std::move(this->mTasks.front());
   //任务出队
   this->mTasks.pop();
   //执行工作
   task.run();
  }
  });
 }
 }
 ~ThreadPoolManage()
 {

 {
  std::unique_lock<std::mutex> lock(this->mQueue_mutex);
  this->mThreadStatus = true;
 }
 //通知所有线程起来工作 然后退出
 this->mCondition.notify_all();
 //等待所有线程工作完毕
 for (std::thread& worker : this->mWorkers) 
 {
  if (worker.joinable()) {
  worker.join();
  }
 }
 }
 /*
 * 添加任务到任务队列
 */
 void addTask(_Ty& task)
 {
 std::unique_lock<std::mutex> lock(this->mQueue_mutex);
 if (mThreadStatus) {
  throw std::runtime_error("workers stop");
 }
 mTasks.emplace(std::move(task));
 mCondition.notify_one();
 }
private:
 /*
 * 工作线程池
 */
 std::vector<std::thread> mWorkers;
 /*
 * 任务队列
 */
 std::queue<_Ty> mTasks;
 /*
 工作线程的最大并发数量
 */
 unsigned int mMaxThreadNum;
 /*
 条件变量 控制线程池中线程的工作状态
 */
 std::condition_variable mCondition;
 /*
 * 工作线程锁
 */
 std::mutex mQueue_mutex;
 /*
 * 控制线程的开关 false 继续工作 true 退出线程
 */
 bool mThreadStatus;

};

#include <iostream>
#include <chrono>
#include "ThreadPool.hpp"
class Task :public AbsTask
{
public:
 void run() override
 {
 std::this_thread::sleep_for(std::chrono::seconds(1));
 std::cout << "works ...... " << std::this_thread::get_id() << std::endl;
 }
};
int main()
{
 ThreadPoolManage<Task> ThreadPool(8);
 for (size_t i = 0; i < 256; i++)
 {
 Task task;
 ThreadPool.addTask(task);
 }
 std::cin.get();
 system("pause");
}