标签：std thread list C++ ThreadPool _. 线程 NewTask 简单

// thread_pool.h
#pragma once
#include <vector>
#include <deque>
#include <thread>
#include <functional>
#include <condition_variable>

class ThreadPool {
    using Task = std::function<void()>;
    using TaskList = std::deque<Task>;
    using WorkThreadQueue = std::vector<std::thread*>;

public:
    ThreadPool();
    ~ThreadPool();

public:
    bool Start(uint16_t ThreadNum = 1);
    void Stop();
    void AddTask(const Task&);

private:
    void ThreadLoop();
    Task AcceptTask();

private:
    uint16_t thread_num_;
    bool is_started_;
    WorkThreadQueue work_thread_list_;
    TaskList task_list_;
    std::mutex thread_pool_mutex_;
    std::condition_variable condition_variable_;
};

// thread_pool.cpp
#include "thread_pool.h"

ThreadPool::ThreadPool() : thread_num_(1), is_started_(false) {}

ThreadPool::~ThreadPool() {
    if (true == is_started_) {
        Stop();
    }
}

bool ThreadPool::Start(uint16_t ThreadNum) {
    thread_num_ = ThreadNum;
    if (false == work_thread_list_.empty()) {
        return false;
    }
    is_started_ = true;
    //预先创建线程
    work_thread_list_.reserve(thread_num_);
    for (uint16_t i = 0; i < thread_num_; ++i) {
        work_thread_list_.push_back(new std::thread(std::bind(&ThreadPool::ThreadLoop, this)));
    }
    return true;
}

void ThreadPool::Stop() {
    std::lock_guard<std::mutex> Lock(thread_pool_mutex_);
    is_started_ = false;
    condition_variable_.notify_all();
    for (WorkThreadQueue::iterator it = work_thread_list_.begin(); it != work_thread_list_.end(); ++it) {
        (*it)->join();
        delete *it;
    }
    work_thread_list_.clear();
}

void ThreadPool::ThreadLoop() {
    while (true == is_started_) {
        Task NewTask = AcceptTask();
        if (NewTask) {
            NewTask();
        }
    }
}

void ThreadPool::AddTask(const Task& NewTask) {
    std::lock_guard<std::mutex> Lock(thread_pool_mutex_);
    task_list_.push_back(NewTask);
    condition_variable_.notify_one();
}

ThreadPool::Task ThreadPool::AcceptTask() {
    std::unique_lock<std::mutex> Lock(thread_pool_mutex_);
    // always use a while-loop, due to spurious wakeup
    while (task_list_.empty() && is_started_) {
        condition_variable_.wait(Lock);
    }
    Task NewTask;
    TaskList::size_type size = task_list_.size();
    if (!task_list_.empty() && is_started_) {
        NewTask = task_list_.front();
        task_list_.pop_front();
    }
    return NewTask;
}

标签：std,thread,list,C++,ThreadPool,_.,线程,NewTask,简单
来源： https://www.cnblogs.com/umichan0621/p/16533311.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。