#ifndef THREADPOOL_H
|
#define THREADPOOL_H
|
|
#include <vector>
|
#include <thread>
|
#include <queue>
|
#include <functional>
|
#include <future>
|
#include <mutex>
|
#include <condition_variable>
|
#include <iostream>
|
|
class ThreadPool {
|
public:
|
explicit ThreadPool(size_t thread_count);
|
~ThreadPool();
|
|
// 提交任务
|
template<class F>
|
void enqueue(F&& task) {
|
{
|
std::unique_lock<std::mutex> lock(m_queue_mutex);
|
m_tasks.emplace(std::forward<F>(task));
|
}
|
m_cv.notify_one();
|
}
|
|
private:
|
void worker_thread();
|
|
std::vector<std::thread> m_threads;
|
std::queue<std::function<void()>> m_tasks;
|
std::mutex m_queue_mutex;
|
std::condition_variable m_cv;
|
bool m_stop;
|
};
|
|
#endif // THREADPOOL_H
|
