生产者消费者练习

// Card.cpp : 定义控制台应用程序的入口点。
//
//1个生产者， 1个消费者， 1个缓冲区
//
#include "stdafx.h"
#include <windows.h>
#include <process.h>

//生产产品的个数
const int PRODUCE_NUMBER = 10;
// 两个事件， 前面表示缓冲区空， 后面表示缓冲区满，用于同步
HANDLE g_hEventBufferEmpty, g_hEventBufferFull;
// 临界区域， 用来实现互斥访问
CRITICAL_SECTION g_cs;

//产品
int g_buffer;

//设置控制台字体颜色
BOOL SetConsoleColor(WORD wAttributes)
{
HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
if (hConsole == INVALID_HANDLE_VALUE)
return FALSE;
return SetConsoleTextAttribute(hConsole, wAttributes);
}

//生产者线程函数
unsigned int __stdcall ProducerThreadFun(LPVOID pM)
{
printf("生产者开始生产！\n");
for (int i = 0; i < PRODUCE_NUMBER; i++)
{
//等待缓冲区为空的信号
WaitForSingleObject(g_hEventBufferEmpty, INFINITE);

//互斥访问缓冲区
EnterCriticalSection(&g_cs);
g_buffer = i;
printf("生产者将数据%d放入缓冲区\n", i);
LeaveCriticalSection(&g_cs);

//生产后设置缓冲区有数据的信号
SetEvent(g_hEventBufferFull);
}
return 0;
}

//消费者线程函数
unsigned int __stdcall ConsumerThreadFun(LPVOID pM)
{
printf("消费者开始消费！\n");
while(true)
{
//等待缓冲区中有数据
WaitForSingleObject(g_hEventBufferFull, INFINITE);

//互斥访问缓冲区
EnterCriticalSection(&g_cs);
//g_buffer = i;
SetConsoleColor(FOREGROUND_GREEN);
printf("  消费者消费数据%d\n", g_buffer);
//设置颜色
SetConsoleColor(FOREGROUND_RED|FOREGROUND_GREEN|FOREGROUND_BLUE);
if (g_buffer == PRODUCE_NUMBER-1)
break;
LeaveCriticalSection(&g_cs);

//通知缓冲区已为空
SetEvent(g_hEventBufferEmpty);
Sleep(500);
}
return 0;
}

int _tmain(int argc, _TCHAR* argv[])
{
const int THREADNUM = 2;
HANDLE hThread[THREADNUM];

InitializeCriticalSection(&g_cs);

//创建两个自动复位事件，一个表示缓冲区为空，一个表示缓冲区有数据
// CreateEvent函数，第二个参数(FALSE)表示是否手动复位，第三个参数(TRUE)表示初始值；
g_hEventBufferEmpty = CreateEvent(NULL, FALSE, TRUE, NULL);
g_hEventBufferFull = CreateEvent(NULL, FALSE, FALSE, NULL);

hThread[0] = (HANDLE)_beginthreadex(NULL, 0, ConsumerThreadFun, NULL, 0, NULL);
hThread[1] = (HANDLE)_beginthreadex(NULL, 0, ProducerThreadFun, NULL, 0, NULL);

WaitForMultipleObjects(THREADNUM, hThread, TRUE, INFINITE);

//销毁事件和临界区
CloseHandle(hThread[0]);
CloseHandle(hThread[1]);

CloseHandle(g_hEventBufferFull);
CloseHandle(g_hEventBufferEmpty);
DeleteCriticalSection(&g_cs);

return 0;
}