与上一篇《秒杀多线程第十篇 生产者消费者问题》的生产者消费者问题一样,读者写者也是一个非常著名的同步问题。读者写者问题描述非常简单,有一个写者很多读者,多个读者可以同时读文件,但写者在写文件时不允许有读者在读文件,同样有读者在读文件时写者也不去能写文件。
上面是读者写者问题示意图,类似于生产者消费者问题的分析过程,首先来找找哪些是属于“等待”情况。
第一.写者要等到没有读者时才能去写文件。
第二.所有读者要等待写者完成写文件后才能去读文件。
找完“等待”情况后,再看看有没有要互斥访问的资源。由于只有一个写者而读者们是可以共享的读文件,所以按题目要求并没有需要互斥访问的资源。类似于上一篇中美观的彩色输出,我们对生产者输出代码进行了颜色设置(在控制台输出颜色设置参见《VC 控制台颜色设置》)。因此在这里要加个互斥访问,不然很有可能在写者线程将控制台颜色设置还原之前,读者线程就已经有输出了。所以要对输出语句作个互斥访问处理,修改后的读者及写者的输出函数如下所示:
-
- void ReaderPrintf(char *pszFormat, ...)
- {
- va_list pArgList;
- va_start(pArgList, pszFormat);
- EnterCriticalSection(&g_cs);
- vfprintf(stdout, pszFormat, pArgList);
- LeaveCriticalSection(&g_cs);
- va_end(pArgList);
- }
-
- void WriterPrintf(char *pszStr)
- {
- EnterCriticalSection(&g_cs);
- SetConsoleColor(FOREGROUND_GREEN);
- printf(" %s\n", pszStr);
- SetConsoleColor(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
- LeaveCriticalSection(&g_cs);
- }
读者线程输出函数所使用的可变参数详见《C,C++中使用可变参数》。
解决了互斥输出问题,接下来再考虑如何实现同步问题。可以设置一个变量来记录正在读文件的读者个数,第一个开始读文件的读者要负责将关闭允许写者进入的标志,最后一个结束读文件的读者要负责打开允许写者进入的标志。这样第一种“等待”情况就解决了。第二种“等待”情况是有写者进入时所以读者不能进入,使用一个事件就可以完成这个任务了——所有读者都要等待这个事件而写者负责触发事件和设置事件为未触发。详细见代码中注释:
-
- #include <stdio.h>
- #include <process.h>
- #include <windows.h>
-
- BOOL SetConsoleColor(WORD wAttributes)
- {
- HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
- if (hConsole == INVALID_HANDLE_VALUE)
- return FALSE;
-
- return SetConsoleTextAttribute(hConsole, wAttributes);
- }
- const int READER_NUM = 5;
-
- CRITICAL_SECTION g_cs, g_cs_writer_count;
- HANDLE g_hEventWriter, g_hEventNoReader;
- int g_nReaderCount;
-
- void ReaderPrintf(char *pszFormat, ...)
- {
- va_list pArgList;
-
- va_start(pArgList, pszFormat);
- EnterCriticalSection(&g_cs);
- vfprintf(stdout, pszFormat, pArgList);
- LeaveCriticalSection(&g_cs);
- va_end(pArgList);
- }
-
- unsigned int __stdcall ReaderThreadFun(PVOID pM)
- {
- ReaderPrintf(" 编号为%d的读者进入等待中...\n", GetCurrentThreadId());
-
- WaitForSingleObject(g_hEventWriter, INFINITE);
-
-
- EnterCriticalSection(&g_cs_writer_count);
- g_nReaderCount++;
- if (g_nReaderCount == 1)
- ResetEvent(g_hEventNoReader);
- LeaveCriticalSection(&g_cs_writer_count);
-
-
- ReaderPrintf("编号为%d的读者开始读取文件...\n", GetCurrentThreadId());
-
- Sleep(rand() % 100);
-
-
- ReaderPrintf(" 编号为%d的读者结束读取文件\n", GetCurrentThreadId());
-
-
- EnterCriticalSection(&g_cs_writer_count);
- g_nReaderCount--;
- if (g_nReaderCount == 0)
- SetEvent(g_hEventNoReader);
- LeaveCriticalSection(&g_cs_writer_count);
-
- return 0;
- }
-
- void WriterPrintf(char *pszStr)
- {
- EnterCriticalSection(&g_cs);
- SetConsoleColor(FOREGROUND_GREEN);
- printf(" %s\n", pszStr);
- SetConsoleColor(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
- LeaveCriticalSection(&g_cs);
- }
-
- unsigned int __stdcall WriterThreadFun(PVOID pM)
- {
- WriterPrintf("写者线程进入等待中...");
-
- WaitForSingleObject(g_hEventNoReader, INFINITE);
-
- ResetEvent(g_hEventWriter);
-
-
- WriterPrintf(" 写者开始写文件.....");
- Sleep(rand() % 100);
- WriterPrintf(" 写者结束写文件");
-
-
- SetEvent(g_hEventWriter);
- return 0;
- }
- int main()
- {
- printf(" 读者写者问题\n");
- printf(" -- by MoreWindows( http://blog.csdn.net/MoreWindows ) --\n\n");
-
-
- InitializeCriticalSection(&g_cs);
- InitializeCriticalSection(&g_cs_writer_count);
-
-
- g_hEventWriter = CreateEvent(NULL, TRUE, TRUE, NULL);
- g_hEventNoReader = CreateEvent(NULL, FALSE, TRUE, NULL);
- g_nReaderCount = 0;
-
- int i;
- HANDLE hThread[READER_NUM + 1];
-
- for (i = 1; i <= 2; i++)
- hThread[i] = (HANDLE)_beginthreadex(NULL, 0, ReaderThreadFun, NULL, 0, NULL);
-
- hThread[0] = (HANDLE)_beginthreadex(NULL, 0, WriterThreadFun, NULL, 0, NULL);
- Sleep(50);
-
- for ( ; i <= READER_NUM; i++)
- hThread[i] = (HANDLE)_beginthreadex(NULL, 0, ReaderThreadFun, NULL, 0, NULL);
- WaitForMultipleObjects(READER_NUM + 1, hThread, TRUE, INFINITE);
- for (i = 0; i < READER_NUM + 1; i++)
- CloseHandle(hThread[i]);
-
-
- CloseHandle(g_hEventWriter);
- CloseHandle(g_hEventNoReader);
- DeleteCriticalSection(&g_cs);
- DeleteCriticalSection(&g_cs_writer_count);
- return 0;
- }
运行结果如下所示:
根据结果可以看出当有读者在读文件时,写者线程会进入等待状态中。当写者线程在写文件时,读者线程也会排队等待,说明读者和写者已经完成了同步。
本系列通过经典线程同步问题来列举线程同步手段的关键段、事件、互斥量、信号量,并作对这四种方法进行了总结。然后又通过二个著名的线程同步实例——生产者消费者问题和读者写者问题来强化对多线程同步互斥的理解与运用。希望读者们能够熟练掌握,从而在笔试面试中能够顺利的“秒杀”多线程的相关试题,获得自己满意的offer。
从《秒杀多线程第十篇生产者消费者问题》到《秒杀多线程第十一篇读者写者问题》可以得出多线程问题的关键在于找到所有“等待”情况和判断有无需要互斥访问的资源。那么如何从实际问题中更好更快更全面的找出这些了?请看《秒杀多线程第十二篇多线程同步内功心法——PV操作上》和《秒杀多线程第十三篇多线程同步内功心法——PV操作下》这二篇以加强解决多线程同步问题的“内功”。
---------------------------------------------------------------------------------------------------------------
在《秒杀多线程第十一篇读者写者问题》文章中我们使用事件和一个记录读者个数的变量来解决读者写者问题。问题虽然得到了解决,但代码有点复杂。本篇将介绍一种新方法——读写锁SRWLock来解决这一问题。读写锁在对资源进行保护的同时,还能区分想要读取资源值的线程(读取者线程)和想要更新资源的线程(写入者线程)。对于读取者线程,读写锁会允许他们并发的执行。当有写入者线程在占有资源时,读写锁会让其它写入者线程和读取者线程等待。因此用读写锁来解决读者写者问题会使代码非常清晰和简洁。
下面就来看看如何使用读写锁,要注意编译读写锁程序需要VS2008,运行读写锁程序要在Vista或Windows Server2008系统(比这两个更高级的系统也可以)。读写锁的主要函数就五个,分为初始化函数,写入者线程申请和释放函数,读取者线程申请和释放函数,以下是详细的函数使用说明:
第一个 InitializeSRWLock
函数功能:初始化读写锁
函数原型:VOID InitializeSRWLock(PSRWLOCK SRWLock);
函数说明:初始化(没有删除或销毁SRWLOCK的函数,系统会自动清理)
第二个 AcquireSRWLockExclusive
函数功能:写入者线程申请写资源。
函数原型:VOID AcquireSRWLockExclusive(PSRWLOCK SRWLock);
第三个 ReleaseSRWLockExclusive
函数功能:写入者线程写资源完毕,释放对资源的占用。
函数原型:VOID ReleaseSRWLockExclusive(PSRWLOCK SRWLock);
第四个 AcquireSRWLockShared
函数功能:读取者线程申请读资源。
函数原型:VOID AcquireSRWLockShared(PSRWLOCK SRWLock);
第五个 ReleaseSRWLockShared
函数功能:读取者线程结束读取资源,释放对资源的占用。
函数原型:VOID ReleaseSRWLockShared(PSRWLOCK SRWLock);
注意一个线程仅能锁定资源一次,不能多次锁定资源。
使用读写锁精简后的代码如下(代码中变参函数的实现请参阅《C,C++中使用可变参数》,控制台颜色设置请参阅《VC 控制台颜色设置》):
-
- #include <stdio.h>
- #include <process.h>
- #include <windows.h>
-
- BOOL SetConsoleColor(WORD wAttributes)
- {
- HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
- if (hConsole == INVALID_HANDLE_VALUE)
- return FALSE;
- return SetConsoleTextAttribute(hConsole, wAttributes);
- }
- const int READER_NUM = 5;
-
- CRITICAL_SECTION g_cs;
- SRWLOCK g_srwLock;
-
- void ReaderPrintf(char *pszFormat, ...)
- {
- va_list pArgList;
- va_start(pArgList, pszFormat);
- EnterCriticalSection(&g_cs);
- vfprintf(stdout, pszFormat, pArgList);
- LeaveCriticalSection(&g_cs);
- va_end(pArgList);
- }
-
- unsigned int __stdcall ReaderThreadFun(PVOID pM)
- {
- ReaderPrintf(" 编号为%d的读者进入等待中...\n", GetCurrentThreadId());
-
- AcquireSRWLockShared(&g_srwLock);
-
-
- ReaderPrintf("编号为%d的读者开始读取文件...\n", GetCurrentThreadId());
- Sleep(rand() % 100);
- ReaderPrintf(" 编号为%d的读者结束读取文件\n", GetCurrentThreadId());
-
-
- ReleaseSRWLockShared(&g_srwLock);
- return 0;
- }
-
- void WriterPrintf(char *pszStr)
- {
- EnterCriticalSection(&g_cs);
- SetConsoleColor(FOREGROUND_GREEN);
- printf(" %s\n", pszStr);
- SetConsoleColor(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
- LeaveCriticalSection(&g_cs);
- }
-
- unsigned int __stdcall WriterThreadFun(PVOID pM)
- {
- WriterPrintf("写者线程进入等待中...");
-
- AcquireSRWLockExclusive(&g_srwLock);
-
-
- WriterPrintf(" 写者开始写文件.....");
- Sleep(rand() % 100);
- WriterPrintf(" 写者结束写文件");
-
-
- ReleaseSRWLockExclusive(&g_srwLock);
- return 0;
- }
- int main()
- {
- printf(" 读者写者问题继 读写锁SRWLock\n");
- printf(" -- by MoreWindows( http://blog.csdn.net/MoreWindows ) --\n\n");
-
-
- InitializeCriticalSection(&g_cs);
- InitializeSRWLock(&g_srwLock);
-
- HANDLE hThread[READER_NUM + 1];
- int i;
-
- for (i = 1; i <= 2; i++)
- hThread[i] = (HANDLE)_beginthreadex(NULL, 0, ReaderThreadFun, NULL, 0, NULL);
-
- hThread[0] = (HANDLE)_beginthreadex(NULL, 0, WriterThreadFun, NULL, 0, NULL);
- Sleep(50);
-
- for ( ; i <= READER_NUM; i++)
- hThread[i] = (HANDLE)_beginthreadex(NULL, 0, ReaderThreadFun, NULL, 0, NULL);
- WaitForMultipleObjects(READER_NUM + 1, hThread, TRUE, INFINITE);
- for (i = 0; i < READER_NUM + 1; i++)
- CloseHandle(hThread[i]);
-
-
- DeleteCriticalSection(&g_cs);
- return 0;
- }
对比下《秒杀多线程第十一篇读者写者问题》中的代码就可以发现这份代码确实清爽许多了。这个程序用VS2008编译可以通过,但在XP系统下运行会导致报错。
在Win7系统下能够正确的运行,结果如图所示:
最后总结一下读写锁SRWLock
1.读写锁声明后要初始化,但不用销毁,系统会自动清理读写锁。
2.读取者和写入者分别调用不同的申请函数和释放函数。