wzdque.cpp

来自「MFC扩展编程实例」· C++ 代码 · 共 58 行

// WzdQue.cpp: implementation of the CWzdQueue class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "WzdQue.h"

//////////////////////////////////////////////////////////////////////
// Construct Message
//////////////////////////////////////////////////////////////////////

CWzdMsg::CWzdMsg(int id,LPSTR pHdr,LPSTR pBody,int len,int error)
{
	m_nID=id;
	m_pHdr=pHdr;
	m_pBody=pBody;
	m_len=len;
	m_error=error;
}


CWzdMsg::~CWzdMsg()
{
	delete []m_pHdr;
	delete []m_pBody;
}


//////////////////////////////////////////////////////////////////////
// Add Message
//////////////////////////////////////////////////////////////////////

void CWzdQueue::Add(CWzdMsg *pMsg)
{
	CSingleLock slock(&m_mutex);

	if (slock.Lock(1000))  // timeout in milliseconds, default= INFINITE
	{
		AddTail(pMsg); // fifo
	}
}

//////////////////////////////////////////////////////////////////////
// Remove Message
//////////////////////////////////////////////////////////////////////

CWzdMsg *CWzdQueue::Remove()
{
	CSingleLock slock(&m_mutex);

	if (slock.Lock(1000))  // timeout in milliseconds, default= INFINITE
	{
		if (!IsEmpty())
			return (CWzdMsg*)RemoveHead();
	}
	return NULL;
}