生产者消费者问题是一个著名的线程同步问题,该问题描述如下:有一个生产者在生产产品,这些产品将提供给若干个消费者去消费,为了使生产者和消费者能并发执行,在两者之间设置一个具有多个缓冲区的缓冲池,生产者将它生产的产品放入一个缓冲区中,消费者可以从缓冲区中取走产品进行消费,显然生产者和消费者之间必须保持同步,即不允许消费者到一个空的缓冲区中取产品,也不允许生产者向一个已经放入产品的缓冲区中再次投放产品。
先假设生产者和消费者都只有一个,且缓冲区也只有一个。这样情况就简便多了。
第一 从缓冲区取出产品和向缓冲区投放产品必须是互斥进行的。可以用关键段和互斥量来完成。
第二 生产者要等待缓冲区为空(或者有空位),这样才可以投放产品,消费者要等待缓冲区不为空,这样才可以取出产品进行消费。并且由于有二个等待过程,所以要用二个事件或信号量来控制。
#include <iostream> #include <windows.h> #include <process.h> using namespace std; const int NUM = 2, BUFFER_SIZE = 10; int buffer; HANDLE buffer_full, buffer_empty; HANDLE thread[NUM]; CRITICAL_SECTION sc; unsigned int __stdcall produce(PVOID pm) { int i = 0; for(;i < BUFFER_SIZE; i++) { WaitForSingleObject(buffer_empty, INFINITE); EnterCriticalSection(&sc); buffer = i; cout << "生产者生产了" << buffer << endl; LeaveCriticalSection(&sc); SetEvent(buffer_full); } return 0; } unsigned int __stdcall consume(PVOID pm) { bool flag = true; while(flag){ WaitForSingleObject(buffer_full, INFINITE); EnterCriticalSection(&sc); cout << " 消费者消费了" << buffer << endl; if(buffer == BUFFER_SIZE) { flag = false; } LeaveCriticalSection(&sc); SetEvent(buffer_empty); } return 0; } int main() { buffer_full = CreateEvent(NULL, FALSE, FALSE, NULL); buffer_empty = CreateEvent(NULL, FALSE, TRUE, NULL); InitializeCriticalSection(&sc); thread[0] = (HANDLE) _beginthreadex(NULL, 0, produce, NULL, 0, NULL); thread[1] = (HANDLE) _beginthreadex(NULL, 0, consume, NULL, 0, NULL); WaitForMultipleObjects(NUM, thread, TRUE, INFINITE); DeleteCriticalSection(&sc); getchar(); return 0; }
采用事件进行同步控制,运行效果如下:
接下来,我们加大缓冲区,增加消费者的个数。
需要注意:
1. 在生产者中,信号量个数的释放限制了消费者的并发数;
2. 缓冲区的大小和信号量的个数相等。 在生产者的实际操作中就不用判断缓冲区是否已满,用信号量进行限制。(在消费者中释放了多少个信号量,说明在生产者中就可以生产多少个数据)
代码如下:
#include <iostream> #include <windows.h> #include <process.h> using namespace std; const int NUM = 4, BUFFER_SIZE = 5, END_FLAG_NUM = 20; int buffer_p, buffer_c, p, c; int buffer[BUFFER_SIZE]; HANDLE buffer_full, buffer_empty; HANDLE thread[NUM]; CRITICAL_SECTION sc, oc; unsigned int __stdcall produce(PVOID pm) { int i = 0; for(; i<END_FLAG_NUM; i++) { WaitForSingleObject(buffer_empty, INFINITE); EnterCriticalSection(&sc); buffer[buffer_p] = i; cout << "生产者生产了" << buffer[buffer_p] << endl; buffer_p = (i + 1) % BUFFER_SIZE; LeaveCriticalSection(&sc); ReleaseSemaphore(buffer_full, 1, NULL); } return 0; } unsigned int __stdcall consume(PVOID pm) { bool flag = true; while(flag) { if(c == END_FLAG_NUM) { flag = false; } else { WaitForSingleObject(buffer_full, 5000); EnterCriticalSection(&sc); // 再次检测 if(c == END_FLAG_NUM) { cout << " 超时或者生产者结束生产" << endl; flag = false; } else { cout << " 消费者消费了" << buffer[buffer_c] << endl; buffer_c = (++c) % BUFFER_SIZE; } LeaveCriticalSection(&sc); ReleaseSemaphore(buffer_empty, 1, NULL); } } EnterCriticalSection(&oc); cout << "消费者线程结束运行,线程号为:" << GetCurrentThreadId() << endl; LeaveCriticalSection(&oc); return 0; } int main() { buffer_full = CreateSemaphore(NULL, 0, BUFFER_SIZE, NULL); buffer_empty = CreateSemaphore(NULL, BUFFER_SIZE, BUFFER_SIZE, NULL); InitializeCriticalSection(&sc); InitializeCriticalSection(&oc); thread[0] = (HANDLE) _beginthreadex(NULL, 0, produce, NULL, 0, NULL); thread[1] = (HANDLE) _beginthreadex(NULL, 0, consume, NULL, 0, NULL); thread[2] = (HANDLE) _beginthreadex(NULL, 0, consume, NULL, 0, NULL); thread[3] = (HANDLE) _beginthreadex(NULL, 0, consume, NULL, 0, NULL); thread[4] = (HANDLE) _beginthreadex(NULL, 0, consume, NULL, 0, NULL); WaitForMultipleObjects(NUM, thread, TRUE, INFINITE); DeleteCriticalSection(&sc); DeleteCriticalSection(&oc); CloseHandle(buffer_full); CloseHandle(buffer_empty); cout << "运行完毕" << endl; getchar(); return 0; }
我们假设了四个消费者,在消费者中,我们用到了双重检验法来进行互斥操作,并且其中的等待函数必须设置超时时间,否则如果设置成无限等待,可能出现有的消费线程结束,有的消费线程无限等待的情况。
对数组下角标采取求余的方式,就可以将缓冲区看成是一个循环的模型,并且一定要保证的是消费者跟在生产者的屁股后面取数据,释放该信号量--->生产者捕获该信号量生产一个数据。
运行效果如下: