#pragma once
#include <thread>
#include <vector>
#include <queue>
#include <condition_variable>
#include <mutex>
/*
* 抽象一個任務(wù) 根據(jù)自己的需求擴展
*/
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)
 {
 //啟動的時候就要創(chuàng)建線程
 auto maxNum = std::thread::hardware_concurrency();
 //獲取當前操作系統(tǒng)中CPU的核心數(shù)量 根據(jù)核心數(shù)量來設(shè)置 最大工作線程的并發(fā)數(shù)量
 mMaxThreadNum = mMaxThreadNum > maxNum ? maxNum : mMaxThreadNum;
 //創(chuàng)建工作線程池
 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; 
   }
   //獲取隊列頭部的任務(wù)
   auto task = std::move(this->mTasks.front());
   //任務(wù)出隊
   this->mTasks.pop();
   //執(zhí)行工作
   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();
  }
 }
 }
 /*
 * 添加任務(wù)到任務(wù)隊列
 */
 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;
 /*
 * 任務(wù)隊列
 */
 std::queue<_Ty> mTasks;
 /*
 工作線程的最大并發(fā)數(shù)量
 */
 unsigned int mMaxThreadNum;
 /*
 條件變量 控制線程池中線程的工作狀態(tài)
 */
 std::condition_variable mCondition;
 /*
 * 工作線程鎖
 */
 std::mutex mQueue_mutex;
 /*
 * 控制線程的開關(guān) false 繼續(xù)工作 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");
}