使用c++实现简易线程池

线程池其实就是一堆处理任务的线程和 一个任务队列 ，处理线程不断地从这个任务队列中拿出任务进行处理。
不过需要注意的是 对于这个任务队列需要保证线程安全

一个简易的线程池需要
1，一个向任务队列中添加任务的接口
2，一个从任务队列中取出任务的接口
3，一个处理任务的方法；
4，提前创建一批线程
线程安全的任务队列使用信号量和互斥锁来实现

1.创建线程
void start(int threadCount);
循环创建 threadcount 个线程
绑定处理任务的方法

2.添加任务的接口
向任务队列中添加任务 ，添加完成之后唤醒一个线程

3.从队列中取出任务
判断当队列中元素个数为0的时候使用wait方法，阻塞线程

如果元素个数不为0，获取一个任务，将这个任务移除队列，线程处理任务。

完整代码

#ifndef MYTHREADPOOL_H
#define MYTHREADPOOL_H

#include
#include
#include
#include
#include
#include
#include
using namespace std;
class MyThreadPool {
public:
	//定义一个void类型的函数类型
	typedef std::function<void()> Task;

	~MyThreadPool();
	static MyThreadPool* GetInstance();

	//启动线程池，线程池进入等待状态
	void start(int thread_count);

	//关闭线程池，如果没有任务了就直接推出所有线程，如果有则等待完成后全部推出
	void stop();

	//添加一个任务
	void addTask(const Task& ta);

private:
	MyThreadPool();
	void runTask();
	Task takeTask();

	vector<thread>m_threads;//线程
	queue<Task>m_taskqueue;//任务队列
	condition_variable m_cond;//条件变量
	mutex m_mutex;//互斥锁
	mutex m_con_mutex;
	int m_thread_count;//线程数
	int max_task;
	bool over;//是否退出
};

#include"MyThreadPool.h"
#include

MyThreadPool::MyThreadPool() {
	over = false;
}

MyThreadPool::~MyThreadPool() {

}

MyThreadPool* MyThreadPool::GetInstance()
{
	static MyThreadPool instance;
	return &instance;
}
void MyThreadPool::start(int threadCount) {
	m_thread_count = threadCount;
	over = false;
	for (int i = 0; i < m_thread_count; ++i) {
		m_threads.push_back(thread(&MyThreadPool::runTask, this));
		m_threads[i].detach();
	}
}

void MyThreadPool::stop(){
	unique_lock<mutex> lck(m_mutex);
	over = true;
	m_cond.notify_all();

	for (int i = 0; i < m_thread_count; ++i)
	{
		m_threads[i].detach();
	}

	m_threads.clear();
}

void MyThreadPool::addTask(const Task& task) {
	unique_lock<mutex> lck(m_mutex);
	m_taskqueue.push(task);
	m_cond.notify_one();
}

void MyThreadPool::runTask() {
	unique_lock<mutex> lck(m_con_mutex);
	while (!over || !m_taskqueue.empty()) {
		Task task = takeTask();
		if (task)
		{
			std::cout << "任务队列大小：" << m_taskqueue.size() << "线程id" << this_thread::get_id() << endl;
			task();
		}
	}
}

MyThreadPool::Task MyThreadPool::takeTask() {
	unique_lock<mutex> lck(m_mutex);
	while (m_taskqueue.empty() && !over) {
		m_cond.wait(lck);
	}
	Task task;
	if (!m_taskqueue.empty()) {                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                             
		task = m_taskqueue.front();
		m_taskqueue.pop();
	}
	return task;
}


#include 
#include
#include"MyThreadPool.h"

//线程池 就是一个阻塞队列 和一堆线程
void func(int i) {
    cout << "task finish" << "-----" << i << "线程id" << this_thread::get_id() << endl;
}
int main()
{
    std::cout << "Hello World!\n";
    MyThreadPool *pool =  MyThreadPool::GetInstance();
    pool->start(5);
    int i = 0;
    while (1)
    {
        i++;
        auto task = bind(func, i);
        pool->addTask(task);
    }

    pool->stop();
    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137

测试截图