C++多线程入门：完善线程池

2024-04-08

字数统计: 539字 | 阅读时长≈ 2分

上一章我们制作了一个简单的线程池，本次将使用智能指针来管理创建的对象：各种任务和创建的线程

使用shared_ptr来管理任务和线程

首先对线程池中进行修改：

将BaseTask*替换为std::shared_ptr< BaseTask >
将线程池中的std::thread*替换为std::shared_ptr< std::thread >
对AddTask的逻辑稍加修改

具体改动：

//void AddTask(BaseTask* Task);
void AddTask(std::shared_ptr<BaseTask> Task);

//BaseTask* GetTask();
std::shared_ptr<BaseTask> GetTask();

//std::vector<std::thread*> ThreadPool;
std::vector<std::shared_ptr<std::thread>> ThreadPool;

//std::list<BaseTask*> TaskList;
std::list<std::shared_ptr<BaseTask>> TaskList;

void Start()
{
    //...
    //auto th = new std::thread(&BaseThreadPool::Run,this);	
    auto th = std::make_shared<std::thread>(&BaseThreadPool::Run,this);
    //...
}

void Exit()
{
    //...
    //智能指针可以帮助我们自动释放资源，所以这里就不在需要我们手动释放
    //std::unique_lock<std::mutex> Lock(mtx);
	//for (auto& th : ThreadPool) {
	//	delete th;
	//	th=nullptr;
	//}
}

在其它地方因为我们使用了auto自动类型推导，所以并不需要有太大变化。

希望异步获取任务的运行结果

对BaseTask类做出修改，增加promise变量以及Set和Get函数：

#include <future>
class BaseTask
{
public:
    //...

    void SetValue(int val)
    {
        p.set_value(val);
    }

    auto GetValue()->int
    {
        return p.get_future().get();
    }

private:
    std::promise<int> p;
};

在main中进行测试

int main()
{
    BaseThreadPool pool;
	pool.Init(4);
	pool.Start();

	// 使用智能指针管理对象
	auto t = make_shared<TestTask>();
	t->name="Name01";
	pool.AddTask(t);
	auto t2 = make_shared<TestTask>();
	t2->name = "Name02";
	pool.AddTask(t2);
	auto t3 = make_shared<TestTask>();
	t3->name = "Name03";
	pool.AddTask(t3);
	auto t4 = make_shared<TestTask>();
	t4->name = "Name04";
	pool.AddTask(t4);
	std::this_thread::sleep_for(3000ms);
	cout << "Task1 ret = " << t.get()->GetValue() << endl;
	cout << "Task2 ret = " << t2.get()->GetValue() << endl;
	
	pool.Exit();

	cout << "Task3 ret = " << t3.get()->GetValue() << endl;
	cout << "Task4 ret = " << t4.get()->GetValue() << endl;
	auto t5 = make_shared<TestTask>();
	pool.AddTask(t2);
	getchar();
	return 0;
}

总结

至此就完成了一个功能相对完善的线程池并尽可能的减少了内存泄漏的问题。

本文作者： KongXinQing
本文链接： https://13114987559.github.io/2024/04/08/note/C++多线程入门：完善线程池/
版权声明： 本博客所有文章除特别声明外，均采用 MIT 许可协议。转载请注明出处！