50 lines
936 B
C++
50 lines
936 B
C++
|
#include "ThreadPool.h"
|
||
|
|
#include "Thread.h"
|
||
|
|
#include <QDebug>
|
||
|
|
#include <algorithm>
|
||
|
|
|
||
|
|
ThreadPool::ThreadPool(size_t threads_num, QObject *parent)
|
||
|
|
: QObject(parent)
|
||
|
|
{
|
||
|
|
for (size_t i = 0; i < threads_num; i++) {
|
||
|
|
m_threads.push_back(QSharedPointer<Thread>(new Thread(i, this)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
ThreadPool::~ThreadPool()
|
||
|
|
{
|
||
|
|
{
|
||
|
|
QMutexLocker lock(&m_mutex);
|
||
|
|
is_running = false;
|
||
|
|
}
|
||
|
|
m_cond.wakeAll();
|
||
|
|
// 等待线程退出
|
||
|
|
for (auto thread : m_threads) {
|
||
|
|
thread->stop();
|
||
|
|
}
|
||
|
|
m_threads.clear(); // 智能指针自动释放内存
|
||
|
|
}
|
||
|
|
|
||
|
|
void ThreadPool::enqueue(Task &t)
|
||
|
|
{
|
||
|
|
QMutexLocker lock(&m_mutex);
|
||
|
|
if (!is_running) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
m_tasks.push(std::move(t));
|
||
|
|
m_cond.wakeOne();
|
||
|
|
}
|
||
|
|
|
||
|
|
void ThreadPool::start()
|
||
|
|
{
|
||
|
|
is_running = true;
|
||
|
|
for (auto thread : m_threads) {
|
||
|
|
thread->start();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int ThreadPool::tasksCount()
|
||
|
|
{
|
||
|
|
return m_tasks.size();
|
||
|
|
}
|