生产者与消费者C语言版

痛定思痛。 2022-04-15 20:56 250阅读 0赞

#include <stdio.h>
    #include <stdlib.h>
    #include <process.h>
    #include <windows.h>
    
    #include "base.h"
    
    typedef struct
    {
    	HANDLE sem;//账号信号量
    	int count;//账号余额
    }Account;
    
    HANDLE threads[2];//第一个是inThread，第二个是outThread
    
    unsigned long inThrId;
    unsigned long outThrId;
    
    void *inThreadEntry(void *param);
    void *outThreadEntry(void *param);
    
    BOOL startInThread(Account *account);
    BOOL startOutThread(Account *account);
    
    BOOL instore(Account *account,int count);
    BOOL outstore(Account *account, int count);
    
    void initAccount(Account *account);
    void destroyAccount(Account *account);
    
    void main(void)
    {
    	Account *account = (Account *)malloc(sizeof(Account));
    	if (NULL == account)
    	{
    		printf("ERROR: Malloc for account failed!\n");
    		return;
    	}
    
    	initAccount(account);
    	
    	if (BOOL_TRUE == startInThread(account))
    	{
    		if (BOOL_TRUE != startOutThread(account))
    		{
    			destroyAccount(account);
    			CloseHandle(threads[0]);
    		}
    	}
    	else
    	{
    		destroyAccount(account);
    	}
    
    	WaitForMultipleObjects(2,threads,TRUE,INFINITE);
    
    	CloseHandle(threads[0]);
    	CloseHandle(threads[1]);
    	destroyAccount(account);
    }
    
    void *inThreadEntry(void *param)
    {
    	while (1)
    	{
    		instore((Account *)param,1);
    		Sleep(1000);
    	}
    
    	return NULL;
    }
    
    void *outThreadEntry(void *param)
    {
    	while (1)
    	{
    		outstore((Account *)param,5);
    		Sleep(1000);
    	}
    
    	return NULL;
    }
    
    BOOL startInThread(Account *account)
    {
    	threads[0] = (HANDLE)_beginthreadex(NULL,0,(unsigned int(_stdcall *)(void *))inThreadEntry,account,0,&inThrId);
    	if (NULL == threads[0])
    	{
    		printf("ERROR: Create in thread failed!\n");
    		return BOOL_FALSE;
    	}
    	else
    	{
    		printf("INFO: Create in thread succeed!\n");
    		return BOOL_TRUE;
    	}
    }
    
    BOOL startOutThread(Account *account)
    {
    	threads[1] = (HANDLE)_beginthreadex(NULL,0,(unsigned int(_stdcall *)(void *))outThreadEntry,account,0,&outThrId);
    	if (NULL == threads[1])
    	{
    		printf("ERROR: Create out thread failed!\n");
    		return BOOL_FALSE;
    	}
    	else
    	{
    		printf("INFO: Create out thread succeed!\n");
    		return BOOL_TRUE;
    	}
    }
    
    BOOL instore(Account *account,int count)
    {
    	if (account == NULL)
    	{
    		printf("ERROR: the account is not initated!\n");
    		return BOOL_FALSE;
    	}
    	
    	WaitForSingleObject(account->sem,INFINITE);
    	account->count = account->count + count;
    	printf("INFO: Store %d into this account!\n",count);
    	printf("INFO: After sotre, the count is %d!\n",account->count);
    	ReleaseSemaphore(account->sem,1,NULL);
    
    	return BOOL_TRUE;
    }
    
    BOOL outstore(Account *account, int count)
    {
    	int nowCount;
    
    	if (NULL == account)
    	{
    		printf("ERROR: Please first init the account!\n");
    		return BOOL_FALSE;
    	}
    
    	WaitForSingleObject(account->sem,INFINITE);
    	nowCount = account->count;
    	if (nowCount < count)
    	{
    		ReleaseSemaphore(account->sem,1,NULL);
    		printf("ERROR: The account is not enough,Waiting...!\n");
    		Sleep(500);
    		return outstore(account,count);
    	}
    	
    	account->count = account->count - count;
    	printf("INFO: Get %d from account succeed!\n",count);
    	printf("INFO: After outstore: %d!\n",account->count);
    	ReleaseSemaphore(account->sem,1,NULL);
    
    	return BOOL_TRUE;
    }
    
    void initAccount(Account *account)
    {
    	memset(account,0x00,sizeof(Account));
    	account->sem = CreateSemaphore(NULL,1,1,(LPCWSTR)"Semaphore for account!");
    	account->count = 0;
    }
    
    void destroyAccount(Account *account)
    {
    	if (NULL == account)
    	{
    		return;
    	}
    
    	CloseHandle(account->sem);
    
    	memset(account,0x00,sizeof(Account));
    	free(account);
    	account = NULL;
    }

执行结果：

![c150fb0a-69d8-31f7-a8a7-0203b72b5ed9.png][]

[c150fb0a-69d8-31f7-a8a7-0203b72b5ed9.png]: /images/20220416/1c74addb6abf43038e0db76c2de3325b.png