众所周知,在多线程程序里面使用cout输出,很容易造成混乱。例如下面这段代码:
// cout_thread.cpp : 定义控制台应用程序的入口点。// #include "stdafx.h"#include #include #include #include #include using namespace std; bool g_runFlag = true;int g_cnt = 0; void outputThread1(){cout << "outputThread1 start" << endl;while (g_runFlag){cout << "outputThread1 " << ++g_cnt << endl;this_thread::yield();cout << "yield " << "this_thread1 ";cout << ++g_cnt << endl;Sleep(500);}cout << "outputThread1 end" << endl;} void outputThread2(){cout << "outputThread2 start" << endl; while (g_runFlag){cout << "outputThread2 " << ++g_cnt << endl;this_thread::yield();cout << "yield " << "this_thread2 ";cout << ++g_cnt << endl;Sleep(500);} cout << "outputThread2 end" << endl;} int _tmain(int argc, _TCHAR* argv[]){thread t1(outputThread1);thread t2(outputThread2); Sleep(5000);g_runFlag = false; t1.join();t2.join(); cout << "exit" << endl; return 0;}
输出是这样的:
有什么办法让cout输出不混乱呢?
一个很容易想到的方法是加锁,在cout执行前加锁,执行后解锁。例如这样:
mutex mutexOutput;mutexOutput.lock();cout << "exit" << endl;mutexOutput.unlock();
但是如果cout语句比较多,这样写有点烦人。
有没有简捷点的办法呢?
有的,可以使用宏帮我们简化代码,让编译器在cout之前自动加锁,在endl后自动解锁。
首先,我们要把加锁和解锁的代码封装成两个函数outputLock和outputUnLock,函数返回类型是bool
然后我们写两个宏,在cout之前调用outputLock,在endl后调用outputUnLock
代码如下:
bool outputLock(){g_mutexOutput.lock(); return true;} bool outputUnLock(){g_mutexOutput.unlock(); return true;}#define cout outputLock() && cout#define endl endl && outputUnLock()
是不是这样就可以了呢? 不是的,启动程序运行会发现输出虽然正常了,但没跑一会就出现异常
查了一下,原来是同一个线程重复调用mutex的lock函数引起的。
原因是有一些代码只调用了cout,但最后没有用endl做为结尾
如outputThread1第7~8行
而我们用宏让编译器在cout之前都添加了outputLock函数调用,导致出现死锁
怎么办呢? 能不能用递归锁recursive_mutex代替呢?
答案是不行的,递归锁可以解决重复lock的问题,但是解决不了cout和endl不匹配的问题导致lock与unlock不匹配的问题,也容易导致死锁
经常出现cout调用两次,而只调用了endl一次的情况。
这个问题我是这样解决的:
增加一个全局变量,标记是否已经上锁,再增加一个全局变量,标记当前获取到锁的线程id
调用outputLock时,先判断mutex是否已经上锁,如果已经上锁并且获取到锁的线程id与本线程id一致,则直接返回即可。
完整代码如下:
// cout_thread.cpp : 定义控制台应用程序的入口点。// #include "stdafx.h"#include #include #include #include #include using namespace std; bool g_runFlag = true;int g_cnt = 0; mutex g_mutexOutput;bool g_lockFlag = false;thread::id g_lockTid; bool outputLock(){if (g_lockFlag && g_lockTid == this_thread::get_id()){return true;}g_mutexOutput.lock(); g_lockFlag = true;g_lockTid = this_thread::get_id();return true;} bool outputUnLock(){g_lockFlag = false;g_lockTid = thread::id ();g_mutexOutput.unlock(); return true;} #define cout outputLock() && cout#define endl endl && outputUnLock() void outputThread1(){cout << "outputThread1 start" << endl;while (g_runFlag){cout << "outputThread1 " << ++g_cnt << endl;this_thread::yield();cout << "yield " << "this_thread1 ";cout << ++g_cnt << endl;Sleep(500);}cout << "outputThread1 end" << endl;} void outputThread2(){cout << "outputThread2 start" << endl; while (g_runFlag){cout << "outputThread2 " << ++g_cnt << endl;this_thread::yield();cout << "yield " << "this_thread2 ";cout << ++g_cnt << endl;Sleep(500);} cout << "outputThread2 end" << endl;} int _tmain(int argc, _TCHAR* argv[]){thread t1(outputThread1);thread t2(outputThread2); Sleep(5000);g_runFlag = false; t1.join();t2.join(); cout << "exit" << endl; return 0;}
这样就可以了:
还能玩什么花样?
其实我们还可以使用宏让编译器在cout之前很多操作,例如这样:
#define cout outputLock() && cout << "[" << strrchr(__FILE__, '') + 1 << ':' << __LINE__ << "] " << __FUNCTION__ << ":" && cout
输出:
在每一行输出前增加打印源文件和时间信息,可以大大方便调试