多线程编程5. 线程池（1）

前置基础：
多线程编程1. 线程库的基本用法
多线程编程2. 多线程数据共享
多线程编程4. std::condition_variable

1. 线程池简介

原理

任务创建者不断创建任务添加到任务队列中，线程池中的工作线程会不断地从任务队列中获取任务，并执行任务。
其中的每一个任务都是一个可执行对象，可以是一个函数。

线程池的优点

1. 减少资源的消耗：较少线程创建和销毁造成的消耗
2. 提高响应速度：当任务到达的时候，就能够立刻执行，减少了线程创建和销毁的时间
3. 能够更好地管理线程

2. 实现代码

下面一个示例，根据线程池原理实现的简易线程池的实现代码

#include <iostream>
#include <thread>
#include <mutex>
#include <vector>
#include <queue>
#include <functional>

class ThreadPool
{
public:
	ThreadPool(size_t nums);
	~ThreadPool();
	void add_task(const std::function<void()>& task);

private:
	std::vector<std::thread>  m_threads;  // 线程池中所有的工作线程
	std::queue<std::function<void()>> m_tasks;   // 任务队列中的当前状态下的任务
	std::condition_variable  m_condition_variable;   // 条件变量，当任务队列为空时，控制工作线程进行等待，当有任务添加到任务队列中时，则取消等待
	std::mutex  m_mutex;  // 用于操作ThreadPool 中的变量时所用的互斥量
};

ThreadPool::ThreadPool(size_t nums) {
	for (size_t i = 0; i < nums; ++i) {
		m_threads.emplace_back([this]() {
			while (true) { // 让工作线程创建好后，一直处于运行的状态
				std::unique_lock<std::mutex> lock(m_mutex);
				// 条件变量，如果任务队列为空，则进行等待
				m_condition_variable.wait(lock, [this]() {
					return !m_tasks.empty();
					});

				std::function<void()> task = m_tasks.front();
				m_tasks.pop();
				lock.unlock();  // 执行任务时，已经不需要再对线程池的变量进行加锁
				
				task();  // 执行任务
			}
		});
	}
}

ThreadPool::~ThreadPool()
{
	m_condition_variable.notify_all();

	// 析构之前一定要等待所有的线程都已经执行完毕了
	for (std::thread& thrad : m_threads) {  // 必须得是&， thread不支持拷贝，只支持移动
		thrad.join();
	}
}

void ThreadPool::add_task(const std::function<void()>& task)
{
	std::unique_lock<std::mutex> lock(m_mutex);
	m_tasks.push(std::move(task));
	// 每次添加一个任务就通知一个工作线程来取任务
	m_condition_variable.notify_one();
}

int main() {
	ThreadPool pool(4);
	for (int i = 0; i < 8; ++i) {
		pool.add_task([i]() {
			std::cout << "Task " << i << " is running in thread " << std::this_thread::get_id() << std::endl;
			std::this_thread::sleep_for(std::chrono::seconds(1));
			std::cout << "Task " << i << " is done" << std::endl;
			});
	}
	return 0;
}

3. 线程池进阶

一般来说，用户的任务函数，可能有很多参数，每个参数的类型也不一样，有些任务甚至还需要得到返回值。因此就需要对用户的任务函数做进一步的泛化。
关于进阶版的线程池见 多线程编程7. 线程池（2）