每天一算法（三）生产着消费者问题

// multiThread.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include<windows.h>
#include<iostream>
using namespace std;

const unsigned short SIZE_OF_BUFFER=10; //the length of buffer
unsigned short buffer[SIZE_OF_BUFFER]={0}; //create space for buffer
unsigned short in=0;                   //the mark of position entering the space 
unsigned short out=0;                  //the mark of position leaving the space
unsigned short Product_ID=0;           //the ID of product,from 1 to 10,not for count
unsigned short Consume_ID=0;           //the ID of consume product_ID in the buffer
unsigned int produce_sum=0;            //the total produce number
unsigned int consume_sum=0;            //the total consume number
HANDLE mutex;                          //the mutex between threads
HANDLE Full_Semaphore;                 //the resource semaphore: buffer is full
HANDLE Empty_Semaphore;                //the resource semaphore: buffer is empty
const unsigned short p_count=20;        //the number of produce one time
const unsigned short c_count=6;        //the number of consumer one time
const unsigned short s_count=p_count+c_count;  //the sum number of threads
HANDLE threads[s_count];               //the handle of every thread
DWORD Producer_ID[p_count];            //the mark of producer thread
DWORD Consumer_ID[c_count];            //the mark of consumer thread
unsigned short control=1;              //control the program run or stop


DWORD WINAPI producer(LPVOID);         //the producer thread
DWORD WINAPI consumer(LPVOID);         //the consumer thread
void produce();                                            
void consume();
void Create_P_Threads();               //create producer thread
void Create_C_Threads();               //create consumer thread
void Product_Sum();                    //print the total of remain product number and print the buffer
void info();                           //info


void Product_Sum()
{
	int i,sum=0;
	for(i=0;i<SIZE_OF_BUFFER;i++)
	{
		if(buffer[i]!=0)
			sum++;
	}
	std::cout<<"  "<<sum<<"         ";
	for(i=0;i<SIZE_OF_BUFFER;i++)
	{
		std::cout<<buffer[i]<<" ";
	}
	printf("\n");
}


void produce()
{
	int i;
	std::cout<<"produce";
	if(Product_ID>=10)
		Product_ID=0;
	Product_ID++;
	produce_sum++;
	buffer[in]=Product_ID;
	printf(" buffer[%d]=%d    ",in,Product_ID);
	in=(in+1)%SIZE_OF_BUFFER;
	Product_Sum();
}

void consume()
{
	int i;
	std::cout<<"consume";
	consume_sum++;
	Consume_ID=buffer[out];
	printf(" buffer[%d]=%d    ",out,Consume_ID);
	buffer[out]=0;
	out=(out+1)%SIZE_OF_BUFFER;
	Product_Sum();
}

DWORD WINAPI producer(LPVOID)                               //producer thread
{
	while(control)
	{
		WaitForSingleObject(Full_Semaphore,INFINITE);       //resource semaphore P operation
		WaitForSingleObject(mutex,INFINITE);                //the mutex P operation
		produce();
		Sleep(1000);
		ReleaseMutex(mutex);                                //resource semaphore P operation
		ReleaseSemaphore(Empty_Semaphore,1,NULL);            //the mutex P operation
	}
	return 0;
}

DWORD WINAPI consumer(LPVOID)                              //consumer thread
{
	while(control)
	{
		WaitForSingleObject(Empty_Semaphore,INFINITE);
		WaitForSingleObject(mutex,INFINITE);
		consume();
		Sleep(1000);
		ReleaseMutex(mutex);
		ReleaseSemaphore(Full_Semaphore,1,NULL);
	}
	return 0;
}

void Create_P_Threads()                                  //create producer thread
{
	for(int i=0;i<p_count;i++)
	{
		threads[i]=CreateThread(NULL,0,producer,NULL,0,&Producer_ID[i]);
		if(threads[i]==NULL)
			control=0;
	}
}


void Create_C_Threads()
{
	for(int i=p_count;i<s_count;i++)
	{
		threads[i]=CreateThread(NULL,0,consumer,NULL,0,&Consumer_ID[i-p_count]);
		if(threads[i]==NULL)
			control=0;
	}
}



void info()
{
	std::cout<<"\n"<<std::endl;
	std::cout<<"**********I did refer to the program on the web."<<std::endl; 
	std::cout<<"*******and I simplify some things and make it more powful!"<<std::endl;
	std::cout<<"**But it is really a word a word knocked out by me based on understanding\n"<<std::endl;
	std::cout<<"produce/consume    remain_total  buffer_state(from 0 to 9)"<<std::endl;
}


int main()
{
	info();
	mutex=CreateMutex(NULL,FALSE,NULL);
	Full_Semaphore=CreateSemaphore(NULL,SIZE_OF_BUFFER,SIZE_OF_BUFFER,NULL);
	Empty_Semaphore=CreateSemaphore(NULL,0,SIZE_OF_BUFFER,NULL);
	Create_P_Threads();
	Create_C_Threads();
	while(control)
	{
		if(getchar())
		{
			std::cout<<std::endl;
			std::cout<<"the total produce product number is "<<produce_sum<<std::endl;
			std::cout<<"the total consume product number is "<<consume_sum<<std::endl;
			control=0;
		}
	}
	return 0;
}