请帮我给这些代码加上
注释 #pragma once
#include <thread>
#include <
mutex>
#include <vector>
#include <queue>
#include <atomic>
#include <functional>
#include <condition_variable>
#include <map>
#include <future>
using namespace std;
class ThreadPool
{
public:
ThreadPool
(int min = 4, int max = thread::hardware_concurrency
());
~ThreadPool
();
void addTask
(function<void
()> f
);
template<typename F, typename... Args>
auto addTask
(F&& f, Args&&... args
) -> future<typename result_of<F
(Args...
)>::type>
{
using returnType = typename result_of<F
(Args...
)>::type;
auto task = make_shared<packaged_task<returnType
()>>
(
bind
(forward<F>
(f
), forward<Args>
(args
)...
)
);
future<returnType> res = task->get_future
();
{
unique_lock<
mutex> lock
(m_queue
Mutex);
m_tasks.emplace
([task]
() {
(*task
)(); }
);
}
m_condition.notify_one
();
return res;
}
private:
void manager
();
void worker
();
private:
thread* m_manager;
map<thread::id, thread> m_workers;
vector<thread::id> m_ids;
int m_minThreads;
int m_maxThreads;
atomic<bool> m_stop;
atomic<int> m_curThreads;
atomic<int> m_idleThreads;
atomic<int> m_exitNumber;
queue<function<void
()>> m_tasks;
mutex m_ids
Mutex;
mutex m_queue
Mutex;
condition_variable m_condition;
}; #include "threadpool.h"
#include <iostream>
ThreadPool::ThreadPool
(int min, int max
) : m_maxThreads
(max
),
m_minThreads
(min
), m_stop
(false
), m_exitNumber
(0
)
{
//m_idleThreads = m_curThreads = max / 2;
m_idleThreads = m_curThreads = min;
cout << "线程数量: " << m_curThreads << endl;
m_manager = new thread
(&ThreadPool::manager, this
);
for
(int i = 0; i < m_curThreads; ++i
)
{
thread t
(&ThreadPool::worker, this
);
m_workers.insert
(make_pair
(t.get_id
(), move
(t
)));
}
}
ThreadPool::~ThreadPool
()
{
m_stop = true;
m_condition.notify_all
();
for
(auto& it : m_workers
)
{
thread& t = it.second;
if
(t.joinable
())
{
cout << "******** 线程 " << t.get_id
() << " 将要退出了..." << endl;
t.join
();
}
}
if
(m_manager->joinable
())
{
m_manager->join
();
}
delete m_manager;
}
void ThreadPool::addTask
(function<void
()> f
)
{
{
lock_guard<
mutex> locker
(m_queue
Mutex);
m_tasks.emplace
(f
);
}
m_condition.notify_one
();
}
void ThreadPool::manager
()
{
while
(!m_stop.load
())
{
this_thread::sleep_for
(chrono::seconds
(2
));
int idle = m_idleThreads.load
();
int current = m_curThreads.load
();
if
(idle > current / 2 && current > m_minThreads
)
{
m_exitNumber.store
(2
);
m_condition.notify_all
();
unique_lock<
mutex> lck
(m_ids
Mutex);
for
(const auto& id : m_ids
)
{
auto it = m_workers.find
(id
);
if
(it != m_workers.end
())
{
cout << "############## 线程 " <<
(*it
).first << "被销毁了...." << endl;
(*it
).second.join
();
m_workers.erase
(it
);
}
}
m_ids.clear
();
}
else if
(idle == 0 && current < m_maxThreads
)
{
thread t
(&ThreadPool::worker, this
);
cout << "+++++++++++++++ 添加了一个线程, id: " << t.get_id
() << endl;
m_workers.insert
(make_pair
(t.get_id
(), move
(t
)));
m_curThreads++;
m_idleThreads++;
}
}
}
void ThreadPool::worker
()
{
while
(!m_stop.load
())
{
function<void
()> task = nullptr;
{
unique_lock<
mutex> locker
(m_queue
Mutex);
while
(!m_stop && m_tasks.empty
())
{
m_condition.wait
(locker
);
if
(m_exitNumber.load
() > 0
)
{
cout << "----------------- 线程任务结束, ID: " << this_thread::get_id
() << endl;
m_exitNumber--;
m_curThreads--;
m_idleThreads--;
unique_lock<
mutex> lck
(m_ids
Mutex);
m_ids.emplace_back
(this_thread::get_id
());
return;
}
}
if
(!m_tasks.empty
())
{
cout << "取出一个任务..." << endl;
task = move
(m_tasks.front
());
m_tasks.pop
();
}
}
if
(task
)
{
m_idleThreads--;
task
();
m_idleThreads++;
}
}
}
扫一扫
5921
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
