C++多线程习题:非原创增加注释(02-2)
模型描述:
程序开启3个线程,这3个线程的ID分别为A、B、C。
每个线程将自己的ID在屏幕上打印5遍。
要求输出结果必须按ABC的顺序显示;如:ABCABC….依次递推
#include <iostream>
#include <chrono>
#include <mutex>
#include <thread>
#include <condition_variable>
std::condition_variable cv;
std::mutex mtx;
int cnt = 0;
void work(char id, int num)
{
while (num--)
{
// 休眠1秒模拟线程工作
std::this_thread::sleep_for(std::chrono::seconds(1));
std::unique_lock<std::mutex> lk(mtx);
// 使用条件变量等待正确的线程激活
// 假设首次线程B(t2)先执行,这里谓词不成立,则线程B进行等待状态
// 直到线程A(t1)运行,谓词成立,线程A取得所有权
cv.wait(lk, [&]{ return id == 'A' + cnt; });
std::cout << id;
cnt = (cnt + 1) % 3;
// 注意:当线程被激活后,会自动上锁,因此这里需要释放锁
lk.unlock();
// 发出一次通知,让其它等待同一资源的线程有调度的机会
cv.notify_one();
}
}
int main()
{
std::thread t1(work, 'A', 5);
std::thread t2(work, 'B', 5);
work('C', 5);
t1.join();
t2.join();
return 0;
}