103 lines
2.5 KiB
C++
103 lines
2.5 KiB
C++
|
#ifndef THREADPOOL_HPP
|
|||
|
|
#define THREADPOOL_HPP
|
|||
|
|
|
|||
|
|
#include <atomic>
|
|||
|
|
#include <condition_variable>
|
|||
|
|
#include <functional>
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <memory> //unique_ptr
|
|||
|
|
#include <queue>
|
|||
|
|
#include <stdexcept>
|
|||
|
|
#include <thread>
|
|||
|
|
#include <vector>
|
|||
|
|
|
|||
|
|
typedef std::function<void(void)> Task;
|
|||
|
|
|
|||
|
|
class ThreadPool {
|
|||
|
|
public:
|
|||
|
|
/**
|
|||
|
|
* @brief ThreadPool
|
|||
|
|
* @param number[in]线程数 *默认开一个线程
|
|||
|
|
* @param
|
|||
|
|
* emptyQuit[in]空线程退出，默认false。如果为true，需要先添加任务，再start，然后waite完成
|
|||
|
|
*/
|
|||
|
|
ThreadPool(int number = 1, bool emptyQuit = false);
|
|||
|
|
~ThreadPool();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief append 往请求队列＜task_queue＞中添加任务<T *>
|
|||
|
|
* @param task
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
bool append(Task task);
|
|||
|
|
/**
|
|||
|
|
* @brief start 启动
|
|||
|
|
*/
|
|||
|
|
void start();
|
|||
|
|
/**
|
|||
|
|
* @brief stop 停止
|
|||
|
|
*/
|
|||
|
|
void stop();
|
|||
|
|
/**
|
|||
|
|
* @brief waiteFinish 等待所有任务结束,设置为任务为空退出时调用
|
|||
|
|
*/
|
|||
|
|
void waiteFinish();
|
|||
|
|
/**
|
|||
|
|
* @brief 任务数
|
|||
|
|
*/
|
|||
|
|
int taskNum();
|
|||
|
|
private:
|
|||
|
|
/**
|
|||
|
|
* @brief worker 线程回调函数
|
|||
|
|
* @param arg
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
static void *worker(void *arg);
|
|||
|
|
/**
|
|||
|
|
* @brief run 工作线程需要运行的函数,不断的从任务队列中取出并执行
|
|||
|
|
*/
|
|||
|
|
void run();
|
|||
|
|
/**
|
|||
|
|
* @brief stopThreadGroup 停止线程组
|
|||
|
|
*/
|
|||
|
|
void stopThreadGroup();
|
|||
|
|
/**
|
|||
|
|
* @brief startThread 启动线程
|
|||
|
|
*/
|
|||
|
|
void startThread();
|
|||
|
|
/**
|
|||
|
|
* @brief waiteThreadFinish 等待线程结束
|
|||
|
|
*/
|
|||
|
|
void waiteThreadFinish();
|
|||
|
|
/**
|
|||
|
|
* @brief notEmpty 是否空
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
bool notEmpty();
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
std::vector<std::thread *> m_WorkThreads; /*工作线程*/
|
|||
|
|
std::queue<Task> m_TasksQueue; /*任务队列*/
|
|||
|
|
|
|||
|
|
std::mutex m_QueueMutex;
|
|||
|
|
std::condition_variable m_Condition; /*必须与unique_lock配合使用*/
|
|||
|
|
|
|||
|
|
std::recursive_mutex m_DataMutex; //数据锁
|
|||
|
|
|
|||
|
|
std::atomic_bool m_StopFlag; //是否停止标志
|
|||
|
|
std::once_flag m_CallStopSlag;
|
|||
|
|
std::once_flag m_CallStartSlag;
|
|||
|
|
std::once_flag m_CallWaiteFinisFlag;
|
|||
|
|
int m_ThreadNum; //线程数
|
|||
|
|
|
|||
|
|
std::atomic_bool m_EmptyQuit; //无任务退出模式,改模式先添加任务，再启动
|
|||
|
|
std::atomic_bool m_JoinFlag; //线程是否Join
|
|||
|
|
|
|||
|
|
std::atomic_bool m_EmptyQuitWaite;//开始退出等待
|
|||
|
|
std::atomic_int m_QuitNum;//退出的线程计数
|
|||
|
|
|
|||
|
|
};
|
|||
|
|
typedef std::shared_ptr<ThreadPool> ThreadPoolPtr;
|
|||
|
|
|
|||
|
|
#endif // THREADPOOL_HPP
|