今天晚上總算有點收獲,寫服務器,要用到(多讀/單寫)互斥,也就是MutiRead/SingleWrite互斥的實現,采用用非信號量來實現,用戶模式,非內核模式,實現高效的線程調度
源代碼如下:
// ThreadTest.cpp : 定義控制臺應用程序的入口點。
//
#include "stdafx.h"
#include <process.h>
#include <windows.h>
#include <stdio.h>
//關鍵區鎖
class MemLock {
CRITICAL_SECTION _crisection;
public:
MemLock(){
InitializeCriticalSection( &_crisection );
}
~MemLock(){
DeleteCriticalSection( &_crisection );
}
void lock() {
EnterCriticalSection( &_crisection );
}
void unlock() {
LeaveCriticalSection( &_crisection );
}
};
//共用資源
class ShareSource
{
MemLock mlock;
volatile LONG g_bResourceUse;
volatile LONG g_bResourceUse1;
int j;
public:
ShareSource()
{
g_bResourceUse = 0;
g_bResourceUse1 = 0;
}
void read(char *user)
{
while(InterlockedIncrement(&g_bResourceUse)>0)
{
if(g_bResourceUse1==0)
{
printf("用戶%s 開始讀 ", user);
Sleep(100);
printf("用戶%s 正在讀..", user);
Sleep(100);
printf("用戶%s 讀完了 \r\n", user);
Sleep(100);
}
InterlockedDecrement(&g_bResourceUse);
Sleep(10);
//printf("%d \r\n", j);
}
}
void write(char *user)
{
mlock.lock();
while(InterlockedIncrement(&g_bResourceUse1)>0 )
{
if(g_bResourceUse==0)
{
printf("[用戶%s 開始寫..", user);
Sleep(100);
printf("用戶%s 正在寫..", user);
Sleep(100);
printf("用戶%s 寫完了]..\r\n", user);
Sleep(100);
}
InterlockedDecrement(&g_bResourceUse1);
Sleep(10);
}
mlock.unlock();
}
};
class ThreadTest
{
public:
char user[20];
ShareSource * share;
HANDLE hThread;
ThreadTest(char *_user, ShareSource * source)
{
strcpy(user, _user); //用戶名
hThread = 0;
share = source;
}
//工作線程的方法
static unsigned __stdcall WorkThreadCallBackMothodRead(LPVOID param) {
ThreadTest *th = (ThreadTest *) param;
while(1)
th->share->read(th->user);
return 0;
}
//工作線程的方法
static unsigned __stdcall WorkThreadCallBackMothodWrite(LPVOID param) {
ThreadTest *th = (ThreadTest *) param;
while(1)
th->share->write(th->user);
return 0;
}
//開始讀線程
void BeginReadThread()
{
hThread = (HANDLE) _beginthreadex(0, 0, ThreadTest::WorkThreadCallBackMothodRead, (LPVOID)this , 0, 0);
}
//開始寫線程
void BeginWriteThread()
{
hThread = (HANDLE) _beginthreadex(0, 0, ThreadTest::WorkThreadCallBackMothodWrite, (LPVOID)this , 0, 0);
}
};
int _tmain(int argc, _TCHAR* argv[])
{
ShareSource source;
ThreadTest T1("T1", &source);
ThreadTest T2("T2", &source);
ThreadTest T3("T3", &source);
//前面兩個讀
T1.BeginReadThread();
T2.BeginReadThread();
//最后一個線程寫
T3.BeginWriteThread();
getchar();
return 0;
}