读者写者问题02（关键段＋读写锁）

//实现代码

//读者与写者问题继 读写锁SRWLock  
#include <stdio.h>  
#include <process.h>  
#include <windows.h>  


#include<iostream>
#include<iomanip>
using namespace std;

//设置控制台输出颜色  
//SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), wAttributes);  

const int iReaderNumber = 5;//读者个数
CRITICAL_SECTION csConsole;//互斥使用控制台资源
SRWLOCK rwLock;//读写锁

//读者线程输出函数(变参函数的实现)  
void ReaderInfo(char* lpszFmt, ...){
	va_list pArgLst;

	va_start(pArgLst, lpszFmt);
	EnterCriticalSection(&csConsole);
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED);  
	vfprintf(stdout, lpszFmt, pArgLst);
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);  
	LeaveCriticalSection(&csConsole);
	va_end(pArgLst);
}
//读者线程函数  
UINT WINAPI ReaderProc(PVOID pParam){
	ReaderInfo("读者【%d】进入等待中······\n", GetCurrentThreadId());

	//读者申请读取文件
	AcquireSRWLockShared(&rwLock);

	//读取文件
	ReaderInfo("读者【%d】开始读取文件······\n", GetCurrentThreadId());
	Sleep(rand() % 100);
	ReaderInfo("读者【%d】结束读取文件······\n", GetCurrentThreadId());

	//读者结束读取文件
	ReleaseSRWLockShared(&rwLock);
	
	return 0;
}

//写者线程输出函数  
void WriterInfo(char* lpszMsg){
	EnterCriticalSection(&csConsole);
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN);  
	cout << lpszMsg;
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);  
	LeaveCriticalSection(&csConsole);
}
//写者线程函数  
UINT WINAPI WriterProc(PVOID pParam){
	WriterInfo("作者等待所有读者阅读完毕······\n");

	//写者申请写文件
	AcquireSRWLockExclusive(&rwLock);

		//写文件
	WriterInfo("作者开始创作······\n");
	Sleep(rand() % 100);
	WriterInfo("作者结束创作······\n");

	//写者结束写文件
	ReleaseSRWLockExclusive(&rwLock);
	return 0;
}

void Lock()
{
	//初始化关键段和读写锁
	InitializeCriticalSection(&csConsole);
	InitializeSRWLock(&rwLock);


	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);  


	HANDLE hThreads[iReaderNumber+1];
	for(int i = 0; i < iReaderNumber+1; i++){
		//先启动二个读者线程，然后启动写者线程，最后启动其它读者结程
		if(i == 2){
			hThreads[i] = (HANDLE)_beginthreadex(NULL, 0, WriterProc, NULL, 0, NULL);
			Sleep(50);
		}
		else					hThreads[i] = (HANDLE)_beginthreadex(NULL, 0, ReaderProc, NULL, 0, NULL);
	}
	WaitForMultipleObjects(iReaderNumber+1, hThreads, TRUE, INFINITE);	
	for(int i = 0; i < iReaderNumber+1; i++){
		CloseHandle(hThreads[i]);
	}


	//销毁关键段
	DeleteCriticalSection(&csConsole);
	//没有删除或销毁SRWLOCK的函数，系统会自动清理
}