一个简单的C语言操作系统生产者消费者模型

#include <stdio.h>
#include <windows.h>
#include <conio.h>
#include <stdlib.h>
#include <time.h>
#define SIZE_OF_BUFFER 8
//一个生产者一个消费者
DWORD WINAPI Producer(LPVOID param);
DWORD WINAPI Consumer(LPVOID param);

void ErrorForHandle(HANDLE handle);
int getRandNum();
void init();
HANDLE  thread1,thread2;
DWORD threadID1,threadID2;

HANDLE   g_hFullSemaphore;              //当缓冲区满时迫使生产者等待
HANDLE   g_hEmptySemaphore;           //当缓冲区空时迫使消费者等待
HANDLE mutex;

//缓冲池大小
DWORD itemNum;

int endThreadFlag = 0;

void main()
{
    printf("演示过程中请按回车键结束/n按回车开始演示/n");
    getch();
    itemNum = 0;
    g_hFullSemaphore = CreateSemaphore(NULL,0,SIZE_OF_BUFFER-1,NULL);
    ErrorForHandle(g_hFullSemaphore);
    g_hEmptySemaphore = CreateSemaphore(NULL,SIZE_OF_BUFFER-1,SIZE_OF_BUFFER-1,NULL);
    ErrorForHandle(g_hEmptySemaphore);
    mutex  = ::CreateMutex(NULL,FALSE,NULL);
    ErrorForHandle(mutex);
    thread1 = ::CreateThread(NULL,0,Producer,NULL,0,&threadID1);
    ErrorForHandle(thread1);
    thread2 = ::CreateThread(NULL,0,Consumer,NULL,0,&threadID2);
    ErrorForHandle(thread2);
    getch();
    endThreadFlag = 1;
    //关闭句柄
    CloseHandle(g_hFullSemaphore);
    CloseHandle(g_hEmptySemaphore);
    CloseHandle(mutex);
    CloseHandle(thread1);
    CloseHandle(thread2);

}

DWORD WINAPI Producer(LPVOID param)
{
    while(!endThreadFlag)
    {
        ::WaitForSingleObject(g_hEmptySemaphore,INFINITE);
        ::WaitForSingleObject(mutex,INFINITE);
        itemNum ++;
        printf("生产者生产产品,当前产品总数为 %d/n",itemNum);
        ::ReleaseMutex(mutex);
        ::ReleaseSemaphore(g_hFullSemaphore,1,NULL);
        Sleep(getRandNum());
    }
    return 0;
}

DWORD WINAPI Consumer(LPVOID param)
{
    while(!endThreadFlag)
    {
        ::WaitForSingleObject(g_hFullSemaphore,INFINITE);
        ::WaitForSingleObject(mutex,INFINITE);
        itemNum --;
        printf("消费者消费产品 ,当前产品总数为%d/n",itemNum);
        ::ReleaseMutex(mutex);
        ::ReleaseSemaphore(g_hEmptySemaphore,1,NULL);
        Sleep(getRandNum());
    }
    return 0;
}

//定义由于句柄分配引起的错误处理
void ErrorForHandle(HANDLE handle)
{
    if(handle == INVALID_HANDLE_VALUE)
    {
        printf("句柄分配错误/n");
        ::exit(0);
    }
}

//初始化操作
void init()
{
    //得到随机种子
    time_t t;
    srand(unsigned(time(&t)));
}

//得到一个随机数
int getRandNum()
{
    int num;
    num = rand() % 2000 +1;
    return num;
}