c++线程池

最新推荐文章于 2025-07-03 17:22:49 发布

原创最新推荐文章于 2025-07-03 17:22:49 发布 · 471 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法 #开发语言

本文介绍了如何使用C++实现一个简单的线程池ThreadPool，通过线程池管理并发任务，展示了如何创建线程、任务队列的使用以及停止线程的操作。

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


class ThreadPool
{
public:
	ThreadPool(int numThreads) : stop(false)
	{
		for (int i = 0; i < numThreads ;i++)
		{
			threads.emplace_back
								(
									[this]()
									{
										while (1)
										{
											std::unique_lock<std::mutex> lock(mtx);
											condition.wait(lock,
												[this]()
												{
													return !tasks.empty() || stop;
												}
											);

											if (stop && tasks.empty()) return;

											std::function<void()> task(std::move(tasks.front()));
											tasks.pop();
											lock.unlock();

											task();
										}

									}
								);
		}
	}

	~ThreadPool()
	{
		{
			std::unique_lock<std::mutex> lock(mtx);
			stop = true;
		}
		condition.notify_all();

		for (auto& t : threads)
		{
			t.join();
		}

	}

	template<class F, class ... Args>
	void enqueue(F&& f, Args&&... args)
	{
		std::function<void()> task = std::bind(std::forward<F>(f), std::forward<Args> (args) ...);

		std::unique_lock<std::mutex> lock(mtx);
		tasks.emplace(std::move(task));
		condition.notify_one();
	}

private:
	std::vector<std::thread> threads;
	std::queue < std::function<void()>> tasks;

	std::mutex mtx;
	std::condition_variable condition;

	bool stop;
};


int main()
{
	ThreadPool pool(4);
	for (size_t i = 0; i < 10; i++)
	{
		pool.enqueue(
			[i]()
			{
				std::cout << "task:" << i << "is running" << std::endl;
				std::this_thread::sleep_for(std::chrono::seconds(1));
				std::cout << "task:" << i << "is done" << std::endl;

			}
		);
	}

	return 0;
}