15,471
社区成员
发帖
与我相关
我的任务
分享
#include "SyncQueue.h"
#include <stdio.h>
DWORD WINAPI WriteThread(LPVOID lpParam)
{
SyncQueue<int> *pSyncQ = (SyncQueue<int>*) lpParam;
int i;
int upBound = (1 << 30);
for(i = 0; i < upBound; ++ i)
{
pSyncQ->Push(i);
}
return 0;
}
DWORD WINAPI ReadThread(LPVOID lpParam)
{
SyncQueue<int> *pSyncQ = (SyncQueue<int>*) lpParam;
int i;
int upBound = (1 << 30);
for(i = 0; i < upBound; ++ i)
{
int itemFromQ = pSyncQ->GetFront();
pSyncQ->Pop();
if (i != itemFromQ)
{
printf("Error!\n");
}
if (i % 10000 == 0)
{
printf("Processed : %d\n", i);
}
}
return 0;
}
void main()
{
SyncQueue<int> syncQ;
HANDLE hWriteThread = ::CreateThread(NULL, NULL, WriteThread, &syncQ, NULL, NULL);
HANDLE hReadThread = ::CreateThread(NULL, NULL, ReadThread, &syncQ, NULL, NULL);
::WaitForSingleObject(hWriteThread, INFINITE);
::WaitForSingleObject(hReadThread, INFINITE);
}
#include <windows.h>
#include <list>
using ::std::list;
template<class T>
class SyncQueue
{
public:
SyncQueue();
~SyncQueue();
void Push(T& item); // 入队
void Pop(); // 出队
T GetFront(); // 得到队头,队空则阻塞
BOOL TryGetFront(T *pRtnItem); // 得到队头,队空则直接返回FALSE,非阻塞。pRtnItem接收返回值
size_t GetSize(); // 得到队列当前长度
protected:
list<T> queue_;
CRITICAL_SECTION lock_;
HANDLE hEvent_;
};
template<class T>
SyncQueue<T>::SyncQueue()
{
::InitializeCriticalSection(&lock_);
hEvent_ = ::CreateEvent(NULL, FALSE, FALSE, NULL);
}
template<class T>
SyncQueue<T>::~SyncQueue()
{
queue_.clear();
::CloseHandle(hEvent_);
::DeleteCriticalSection(&lock_);
}
template<class T>
void SyncQueue<T>::Push(T& item) // 入队
{
::EnterCriticalSection(&lock_);
queue_.push_back(item);
::SetEvent(hEvent_);
::LeaveCriticalSection(&lock_);
}
template<class T>
void SyncQueue<T>::Pop() // 出队
{
::EnterCriticalSection(&lock_);
if (queue_.size() > 0)
{
queue_.pop_front();
}
::LeaveCriticalSection(&lock_);
}
template<class T>
T SyncQueue<T>::GetFront() // 得到队头,队空则阻塞
{
::EnterCriticalSection(&lock_);
while (queue_.size() == 0)
{
::LeaveCriticalSection(&lock_);
WaitForSingleObject(hEvent_, INFINITE);
::EnterCriticalSection(&lock_);
}
T headItem = queue_.front();
::LeaveCriticalSection(&lock_);
return headItem;
}
template<class T>
BOOL SyncQueue<T>::TryGetFront(T *pRtnItem) // 得到队头,队空则直接返回FALSE,非阻塞。pRtnItem接收返回值
{
assert(pRtnItem != NULL);
::EnterCriticalSection(&lock_);
if (queue_.size() == 0)
{
::LeaveCriticalSection(&lock_);
return FALSE;
}
T headItem = queue_.front();
::LeaveCriticalSection(&lock_);
return headItem;
}
template<class T>
size_t SyncQueue<T>::GetSize() // 得到队列当前长度
{
::EnterCriticalSection(&lock_);
size_t size = queue_.size();
::LeaveCriticalSection(&lock_);
return size;
}