Modern C++ 一个例子学习条件变量

目录

问题程序

施魔法让BUG浮出水面

条件变量注意事项

修改程序

---

问题程序

今天无意中看到一篇帖子，关于条件变量的，不过仔细看看发现它并达不到原本的目的。

程序如下，读者可以先想想他的本意，以及有没有问题：

#include <iostream>
#include <thread>
#include <condition_variable>
#include <mutex>
#include <unistd.h>
using namespace std;
//全局条件变量
condition_variable cond;
mutex _mutex;
int count = 0;
 
void fun1(){
    while(1)
    {
        count++;
        unique_lock<mutex>lock(_mutex);
        if(count%5 == 0)
        {
            cond.notify_one();
        }
        else
        {
            cout<<"this is fun1,count="<<count<<endl;
        }
        lock.unlock();
        sleep(1);
    }
}
 
void fun2()
{
    while(1)
    {
        unique_lock<mutex>lock(_mutex);
        cond.wait(lock);
        cout<<"this is fun2,count="<<count<<endl;
        lock.unlock();
        sleep(2);
    }
}
 
int main()
{
    thread t1(fun1);
    thread t2(fun2);
    t1.join();
    t2.join();
    return 0;
}

OK，本意显然是：

1. 从1开始打印整数
2. 线程t1, 打印非5的倍数
3. 线程t2, 打印5的倍数

编译执行，运行的还不错，符合预期，但这都是sleep的功劳。

施魔法让BUG浮出水面

把fun1中的sleep去掉，fun2中的sleep放到cond.wait(lock)后，它BUG的面目就暴露出来了：

void fun1(){
    while(1)
    {
        count++;
        unique_lock<mutex>lock(_mutex);
        if(count%5 == 0)
        {
            cond.notify_one();
        }
        else
        {
            cout<<"this is fun1,count="<<count<<endl;
        }
        lock.unlock();
    }
}

void fun2()
{
    while(1)
    {
        unique_lock<mutex>lock(_mutex);
        cond.wait(lock);
        sleep(2);
        cout<<"this is fun2,count="<<count<<endl;
        lock.unlock();
    }
}

[mzhai@lock]$ ./a.out
this is fun1,count=1
this is fun1,count=2
this is fun1,count=3
this is fun1,count=4
this is fun2,count=6
this is fun1,count=6
this is fun1,count=7
this is fun1,count=8
this is fun1,count=9
this is fun1,count=11
this is fun1,count=12
this is fun1,count=13
this is fun1,count=14
this is fun1,count=16
this is fun1,count=17
this is fun1,count=18
this is fun1,count=19
this is fun1,count=21

多线程结果不能因随机加了几个sleep就不同，加sleep仅仅是模拟线程调度不大一样了。

再回过头来看看代码哪些地方有问题：

1. cond.notify_one(); count是5的倍数时，t1会通过notify_one通知t2做事，但并不会阻止t1继续执行。想想一下如果t1执行的很快而t2一直没得到调度，则t1会打印1，2，3，4，6，7，8，9，11...
2. cond.wait(lock); 可能会假唤醒，此时t1并没有通知它。

那"this is fun2,count=6" 是怎么回事哪？不应该是5吗？一种可能性是（可以通过GDB调试来模拟）：

条件变量注意事项

1. 条件变量不擅长单打独斗，一般要和flag变量与锁同时使用。
2. notify对方线程并不代表调度让给了对方线程。

修改程序

说了那么多，怎么改哪？

这是一个典型的你等我我等你的例子，对于这个例子都是一方干完事情另一方才能继续，完全串休化的任务，直接写到一个线程里即可。如果说我为了练习线程同步技巧非要整两个线程，那也行，condition_variable官方文档上就有一个例子实现了main线程等待worker_thread完成任务：

#include <condition_variable>
#include <iostream>
#include <mutex>
#include <string>
#include <thread>
 
std::mutex m;
std::condition_variable cv;
std::string data;
bool ready = false;
bool processed = false;
 
void worker_thread()
{
    // Wait until main() sends data
    std::unique_lock lk(m);
    cv.wait(lk, []{ return ready; });
 
    // after the wait, we own the lock.
    std::cout << "Worker thread is processing data\n";
    data += " after processing";
 
    // Send data back to main()
    processed = true;
    std::cout << "Worker thread signals data processing completed\n";
 
    // Manual unlocking is done before notifying, to avoid waking up
    // the waiting thread only to block again (see notify_one for details)
    lk.unlock();
    cv.notify_one();
}
 
int main()
{
    std::thread worker(worker_thread);
 
    data = "Example data";
    // send data to the worker thread
    {
        std::lock_guard lk(m);
        ready = true;
        std::cout << "main() signals data ready for processing\n";
    }
    cv.notify_one();
 
    // wait for the worker
    {
        std::unique_lock lk(m);
        cv.wait(lk, []{ return processed; });
    }
    std::cout << "Back in main(), data = " << data << '\n';
 
    worker.join();
}

我们依样画葫芦：

#include <iostream>
#include <thread>
#include <condition_variable>
#include <mutex>
#include <unistd.h>
using namespace std;
//全局条件变量
condition_variable cond;
mutex _mutex;
bool ready = false;
bool processed = false;

int count = 0;

void fun1(){
    while(1)
    {
        count++;
        unique_lock<mutex> lock1(_mutex);
        if(count%5 == 0)
        {
            ready = true;
            processed = false;
            lock1.unlock();
            cond.notify_one();
            lock1.lock();
            cond.wait(lock1, []{ return processed; });
        }
        else
        {
            cout<<"this is fun1,count="<<count<<endl;
        }
        lock1.unlock();
    }
}

void fun2()
{
    while(1)
    {
        unique_lock<mutex> lock1(_mutex);
        cond.wait(lock1, []{ return ready; });
        cout<<"this is fun2,count="<<count<<endl;
        processed = true;
        ready = false;
        lock1.unlock();
        cond.notify_one();
    }
}

int main()
{
    thread t1(fun1);
    thread t2(fun2);
    t1.join();
    t2.join();
    return 0;
}

结果符合预期，感兴趣的读者可以到处插入sleep测试一下。

啰嗦几句多线程程序的测试

多线程程序架构设计很重要，因为它很难测试，很难穷尽负面测试用例。几种可行的测试办法：

1. 随机加sleep。需要改程序。参考上面。
2. GDB调试。模拟和正常运行不同的调度策略。参考《GDB调试技巧实战--多线程&弱鸡条件变量-CSDN博客》
3. strace、bpftrace、bcc把快速运行的程序降慢。目的是降低一个线程的速度，另一个保持原来的速度或者也降低。原来两者都是100迈前进，那一个100另一个50会不会出问题？或者10 20哪？组合就很多了。