C++实现简易线程池

#include <iostream>
#include <vector>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <queue>
#include <functional>
using namespace std;

class ThreadPool {
private:
	typedef function<void()> Task;
public:
	ThreadPool(int);
	~ThreadPool();
	bool start();
	void put(Task task);
	queue<Task> q;
private:
	Task take();
	int threadnum;
	vector<thread> workers;
	mutex mtx;
	condition_variable cv;
	void workfunc();
	bool stop;
};

ThreadPool::ThreadPool(int num)
	:threadnum(num),
	stop(false)
{}

ThreadPool::~ThreadPool() {
	stop = true;
	cv.notify_all();
	for (int i = 0; i < threadnum; ++i) {
		workers[i].join();
	}
}
bool ThreadPool::start() {
	stop = false;
	for (int i = 0; i < threadnum; ++i) {
		workers.push_back(thread(&ThreadPool::workfunc, this));
	}
	return true;
}

void ThreadPool::put(Task task) {
	unique_lock<mutex> ul(mtx);
	q.emplace(task);
	cv.notify_one();
}

ThreadPool::Task ThreadPool::take() {
	unique_lock<mutex> ul(mtx);
	cv.wait(ul, [this]() {
		return !q.empty() || stop;
	});
	if (stop) return nullptr;
	Task ret(move(q.front()));
	q.pop();
	return move(ret);
}

void ThreadPool::workfunc() {
	while (!stop) {
		Task t = take();
		t();
	}
}

void func1() {
	for (int i = 0; i < 1000; ++i) cout << this_thread::get_id() << "第" << i << "次执行方法1" << endl;
}
void func2() {
	for (int i = 0; i < 1000; ++i) cout << this_thread::get_id() << "第" << i << "次执行方法2" << endl;
}
void func3() {
	for (int i = 0; i < 1000; ++i) cout << this_thread::get_id() << "第" << i << "次执行方法3" << endl;
}
void func4() {
	for (int i = 0; i < 1000; ++i) cout << this_thread::get_id() << "第" << i << "次执行方法4" << endl;
}
void func5() {
	for (int i = 0; i < 1000; ++i) cout << this_thread::get_id() << "第" << i << "次执行方法5" << endl;
}

int main() {
	ThreadPool test(2);
	test.start();
	test.put(func1);
	test.put(func2);
	test.put(func3);
	test.put(func4);
	test.put(func5);

	while (true);

	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值