【C++】实现三个线程顺序打印1 2 3

前言

在线程之间同步的时候，往往要使用锁和信号量机制来完成对资源的绑定。恰巧我在牛客上看到过一个面试题，题目：

创建3个线程，三个线程按顺序分别循环打印 1 2 3

这个题是个生产者消费者模型的变体，每个线程既是生成也是消费者

---

代码

#include<iostream>
#include<mutex>
#include<semaphore.h>
#include<unistd.h>
#include<condition_variable>
#include<thread>

using namespace std;

mutex mtx;//创建一个锁
bool running = true;
condition_variable cv;
int turn = 1;

void print(int val) {
    while (running) {
        unique_lock<mutex> lock(mtx);//创建一个unique锁
        cv.wait(lock, [val] {return turn == val;});//释放锁，阻塞等到满足后面的lambda表达式后再加上锁

        cout<<val<<endl;
        sleep(1);

        turn = (turn % 3) + 1;//作为生产者制作下一个资源
        
        cv.notify_all();//通知消费者
    }
}

int main() {
    thread t1(print, 1);
    thread t2(print, 2);
    thread t3(print, 3);

    sleep(10);
    running = false;


    t1.join();
    t2.join();
    t3.join();
    
    return 0;
}

---

代码分析

在这段代码中，cv.wait从某种程度上是一个消费者，阻塞等待临界区有资源，有资源的限定条件就是wait的第二个参数[val] {return turn == val;}。而在消耗自己的资源（cout<<val<<endl;）后，turn = (turn % 3) + 1;这行代码作为生产者制作的新的资源，并通知所有消费者，其中满足自己条件的消费者就可以再进入临界区。完成了输出 1 2 3 1 2 3的功能