多线程_-读者写者问题

1. #include <windows.h>  
2. #include <iostream>  
3. #include <process.h>  
4. #include <cstdlib>  
5. #include <ctime>  
6.   
7. using namespace std;  
8. DWORD WINAPI Reader(LPVOID lpParameter);  
9. DWORD WINAPI Writer(LPVOID lpParameter);  
10.   
11.   
12. //全局变量  
13. CRITICAL_SECTION ReaderMutex, WriteMutex;//这个变量用来保证输出时不会竞争  
14. CRITICAL_SECTION WritePrior;  
15. int ReaderCount = 0;  
16.   
17. int main(int argc,char *argv[])  
18. {  
19.     HANDLE hthread[5];  
20.     HANDLE hWriter;  
21.     int i;  
22.     int arg[5];  
23.     int count = 5;  
24.     unsigned long retval;  
25.   
26.   
27.     InitializeCriticalSection(&WriteMutex);  
28.     InitializeCriticalSection(&ReaderMutex);  
29.     InitializeCriticalSection(&WritePrior);  
30.   
31.     for(i = 0; i<5;i++)  
32.     {  
33.         arg[i] = i;  
34.         hthread[i] = CreateThread(NULL, 0, Reader, (void*)(arg+i), 0, NULL);  
35.         if( hthread[i] == INVALID_HANDLE_VALUE)//如果线程创建失败返回-1  
36.         {  
37.             cerr << "error while create thread " << i <<endl;  
38.             cerr << "error code : "<< GetLastError() <<endl;  
39.         }  
40.     }  
41.   
42.   
43.     hWriter = CreateThread(NULL, 0, Writer, NULL, 0, NULL);  
44.     //等待所有线程结束  
45.     retval = WaitForMultipleObjects(5,hthread,true,INFINITE);//等待多个线程  
46.     if(retval == WAIT_FAILED)  
47.     {  
48.         cerr<< "wait error,error code: "<<GetLastError()<<endl;  
49.     }  
50.     for(i = 0; i<5;i++)  
51.     {  
52.         if(CloseHandle(hthread[i]) == false)//关闭句柄  
53.         {  
54.             cerr << "error while close thread " <<i<<endl;  
55.             cerr << "error code: "<<GetLastError()<<endl;  
56.         }  
57.     }  
58.   
59.     DeleteCriticalSection( &WriteMutex);  
60.     DeleteCriticalSection( &ReaderMutex);  
61.     DeleteCriticalSection( &WritePrior);  
62.     return 0;  
63. }  
64.   
65.   
66. DWORD WINAPI Reader(LPVOID lpParameter)  
67. {  
68.     int n = ((int *)lpParameter)[0];  
69.     while(true)  
70.     {  
71.         //EnterCriticalSection( &WritePrior);  
72.         EnterCriticalSection( &ReaderMutex);  
73.         if ( ReaderCount == 0 )EnterCriticalSection(&WriteMutex);  
74.         ReaderCount++;  
75.         LeaveCriticalSection( &ReaderMutex);  
76.         cout << "Reader " << n << "is reading " << endl;  
77.         Sleep(15);  
78.   
79.         EnterCriticalSection( &ReaderMutex);  
80.         ReaderCount--;  
81.         if ( ReaderCount == 0 )LeaveCriticalSection(&WriteMutex);  
82.         LeaveCriticalSection( &ReaderMutex);  
83.         //LeaveCriticalSection( &WritePrior);  
84.         Sleep(1);  
85.     }  
86.   
87.     return n;  
88. }  
89.   
90.   
91. DWORD WINAPI Writer(LPVOID lpParameter)  
92. {  
93.   
94.     while(true)  
95.     {  
96.         //EnterCriticalSection( &WritePrior);  
97.         EnterCriticalSection( &WriteMutex);  
98.         cout << "Process write. The number of reader:" << ReaderCount<< endl;  
99.         Sleep(15);  
100.         LeaveCriticalSection( &WriteMutex);  
101.         //LeaveCriticalSection( &WritePrior);  
102.         //Sleep(10);  
103.     }  
104.     return 0;  
105. }