waiting.cpp

来自「带文件传输功能的聊天室,可多人聊天,也可私聊,任意用户间可互传文件」· C++ 代码 · 共 86 行

// Waiting.cpp: implementation of the CWaiting class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Chatter.h"
#include "Waiting.h"
#include "connectdlg.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CWaiting::CWaiting()
{
	m_pDlg = new CConnectDlg;
	m_pDlg->Create(IDD_CONNECTING, NULL);
	m_pDlg->CenterWindow(NULL);
	m_pDlg->ShowWindow(SW_SHOW);
}

CWaiting::~CWaiting()
{
	m_pDlg->ShowWindow(SW_HIDE);
	m_pDlg->DestroyWindow();
	delete m_pDlg;
}

void CWaiting::BeginWaiting()
{
	m_pDlg->ShowWindow(SW_SHOW);
}
void CWaiting::EndWaiting()
{
	m_pDlg->ShowWindow(SW_HIDE);
		
}

void CWaiting::SetPrompt(LPCSTR sPrompt)
{
	m_pDlg->GetDlgItem(IDC_HINT)->SetWindowText(sPrompt);
}

BOOL CWaiting::DoWaiting(HANDLE hEvent, DWORD nDelay)
{
	DWORD sStart = GetTickCount();
	do 
	{
		MSG msg ; 
		
		//从消息队列中读取数据
		if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) 
		{ 
			DispatchMessage(&msg); 
		}		
		
		//检查是否有事件激活
		int result = MsgWaitForMultipleObjects(1, 
			&hEvent, 
			TRUE,
			0,
			QS_ALLEVENTS);
		
		if(result == WAIT_OBJECT_0)
		{ 
			//事件来了,返回TRUE
			return TRUE;
		}
		else
		{
			//windows消息来了或者超时了:继续循环
			continue;
		} 

	} while(GetTickCount() - sStart < nDelay);
	
	AfxMessageBox("超时");
	//超时
	return FALSE;
}