C++手写线程池80行搞定(加详细注释)

#include <cstdio>
#include<iostream>
#include<mutex>
#include<string>
#include<thread>
#include<future>
#include<queue>//任务队列,生产者
#include<vector>
#include<functional>
using namespace std;

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(move(task));
        }
        condition.notify_one();
    }

private:
    vector<thread> threads;//工作的线程数组
    queue < std::function<void()>>tasks;//任务队列(属于函数类型)
    std::mutex mtx;//控制线程进出(需要一个互斥锁+)
    std::condition_variable condition;//条件变量
    bool stop;
};


int main()
{
    Threadpool mypool(5);//线程池里面开五个线程
    for (int i = 0; i < 10; i++)
    {
        mypool.enqueue([i] {
            std::cout << "task:" << i << "is runing" << std::endl;
            std::this_thread::sleep_for(std::chrono::seconds(1));
            std::cout << "task:" << i << "is done" << std::endl;
            });

    }


    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值