C++11 多线程编程 使用lambda创建std::thread (生产/消费者模式)

要写个tcp server / client的博客，想着先写个c++11多线程程序。方便后面写博客使用。

目前c++11中写多线程已经很方便了，不用再像之前的pthread_create，c++11中已经有了std::thread库可以方便使用。

 

直接看代码(100个任务, 多个线程处理)：

#include <iostream>
#include <thread>
#include <chrono>
#include <vector>
#include <mutex>
 
class Task{
public:
    Task(int x, std::shared_ptr<std::mutex> mutex)
            :_x(x), _mutex(mutex){ }
    
    void handle(){
        int task_id = 0;
        while(true){
            //获取任务, 尽早释放锁
            if (_x > 0){ 
                std::lock_guard<std::mutex> lock(*_mutex);
                if (_x > 0){ task_id = _x; --_x; }
                else { _x = 0; }
            }   
            else { return ; } 

            //do task
            std::cout << "do task id: " << task_id << std::endl;
            std::this_thread::sleep_for(std::chrono::seconds(1));
        }   
    }   
private:
    int _x; 
    std::shared_ptr<std::mutex> _mutex;
};

int main()
{
    int x = 0;
    const int THREAD_NUM = 7;
    const int TASK_NUM = 100;
    std::vector<std::thread> threads;

    //shared_ptr 主线程与子线程可以共用一把锁.
    //方便后面扩展程序(生产/消费者)
    std::shared_ptr<std::mutex> mutex = std::make_shared<std::mutex>();
    Task t(TASK_NUM, mutex);

    //新建线程, std::thread支持使用lambda
    for (int i = 0; i < THREAD_NUM; ++i){
        threads.emplace_back(std::thread(
            [&t] { t.handle(); })
        );
    }

    //等待线程结束
    for(auto &thread : threads){ thread.join(); }
    return 0;
}

编译、执行:

g++ --std=c++11 -pthread thread.cpp