c++ 简单线程池

蜉蝣之翼❉

已于 2024-07-03 17:29:52 修改

阅读量1.1k

点赞数 21

CC 4.0 BY-SA版权

分类专栏： C++ 文章标签： c++ 开发语言

于 2024-07-03 17:20:35 首次发布

本文链接：https://blog.csdn.net/fuyouzhiyi/article/details/140153321

头文件

#pragma once
#include <iostream>
#include <vector>
#include <queue>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <functional>
#include <future>

class ThreadPool {
   
   
public:
    ThreadPool(size_t numThreads);
    ~ThreadPool();

    template<class F, class... Args>
    auto enqueue(F&& f, Args&&... args)->std::future<typename std::result_of<F(Args...)>::type>;
private:
    void workerThread();
private:
    std::vector<std::thread> workers;
    std::queue<std::function<void()>> tasks;

    std::mutex queueMutex;
    std::condition_variable condition;
    bool stop;
};

cpp

 #include "threadPool.h"

ThreadPool::ThreadPool(size_t numThreads) : stop(false)
{
   
   
    for (size_t i = 0; i < numThreads; ++i)
    {
   
   
        workers.emplace_back(&ThreadPool::workerThread, this);
    }
}

ThreadPool::~ThreadPool() {
   
   
    {
   
   
        std::unique_lock<std::mutex> lock(queueMutex);
        stop = true;
    }
    condition.notify_all();
    for (std::thread& worker : workers) {
   
   
        worker.join();
    }
}

void ThreadPool::workerThread() {
   
   
    while (true) {
   
   
        std::function<void()> task;
        {
   
   
            std::unique_lock<std::mutex> lock(queueMutex);
            condition.wait(lock, [this] {
   
    return stop || !tasks.empty(); });
            if (stop